Learning Functional Distributions with Private Labels. - Changlong Wu; Yifan Wang 035; Ananth Grama; Wojciech Szpankowski (ICML, 2023)

A Theory of Fault-Tolerant Learning. - Changlong Wu; Yifan Wang 035; Ananth Grama (ICML, 2024)

Cascade Reward Sampling for Efficient Decoding-Time Alignment. - Bolian Li; Yifan Wang 035; Ananth Grama; Ruqi Zhang (CoRR, 2024)

Deconvolving Complex Neuronal Networks into Interpretable Task-Specific Connectomes. - Yifan Wang 035; Ananth Grama; Vikram Ravindra (BCB, 2025)

From Low Rank Gradient Subspace Stabilization to Low-Rank Weights: Observations, Theories, and Applications. - Ajay Kumar Jaiswal; Yifan Wang 035; Lu Yin 006; Shiwei Liu 003; Runjin Chen; Jiawei Zhao; Ananth Grama; Yuandong Tian; Zhangyang Wang (ICML, 2025)

More is Less: The Pitfalls of Multi-Model Synthetic Preference Data in DPO Safety Alignment. - Yifan Wang 035; Runjin Chen; Bolian Li; David Cho; Yihe Deng; Ruqi Zhang; Tianlong Chen 001; Zhangyang Wang; Ananth Grama; Junyuan Hong (CoRR, 2025)

DRIFT: Learning from Abundant User Dissatisfaction in Real-World Preference Learning. - Yifan Wang 035; Bolian Li; Junlin Wu; Zhaoxuan Tan; Zheli Liu; Ruqi Zhang; Ananth Grama; Qingkai Zeng 001 (CoRR, 2025)

LLMs Can Get "Brain Rot"! - Shuo Xing; Junyuan Hong; Yifan Wang 035; Runjin Chen; Zhenyu Zhang 015; Ananth Grama; Zhengzhong Tu; Zhangyang Wang (CoRR, 2025)

SARL: Label-Free Reinforcement Learning by Rewarding Reasoning Topology. - Yifan Wang 035; Bolian Li; David Cho; Ruqi Zhang; Fanping Sui; Ananth Grama (CoRR, 2026)

Inference-Time Code Selection via Symbolic Equivalence Partitioning. - David Cho; Yifan Wang 035; Fanping Sui; Ananth Grama (CoRR, 2026)

Addressing Performance Saturation for LLM RL via Precise Entropy Curve Control. - Bolian Li; Yifan Wang 035; Yi Ding; Anamika Lochab; Ananth Grama; Ruqi Zhang (CoRR, 2026)