Learning Entangled Single-Sample Distributions via Iterative Trimming. - Hui Yuan 002; Yingyu Liang (AISTATS, 2020)

Learning Entangled Single-Sample Gaussians in the Subset-of-Signals Model. - Yingyu Liang; Hui Yuan 002 (COLT, 2020)

Bandit Theory and Thompson Sampling-Guided Directed Evolution for Sequence Optimization. - Hui Yuan 002; Chengzhuo Ni; Huazheng Wang; Xuezhou Zhang; Le Cong; Csaba Szepesvári; Mengdi Wang (NeurIPS, 2022)

Reward-Directed Conditional Diffusion: Provable Distribution Estimation and Reward Improvement. - Hui Yuan 002; Kaixuan Huang; Chengzhuo Ni; Minshuo Chen; Mengdi Wang (NeurIPS, 2023)

Unified Off-Policy Learning to Rank: a Reinforcement Learning Perspective. - Zeyu Zhang; Yi Su; Hui Yuan 002; Yiran Wu; Rishab Balasubramanian; Qingyun Wu; Huazheng Wang; Mengdi Wang (NeurIPS, 2023)

Tree Search-Based Evolutionary Bandits for Protein Sequence Optimization. - Jiahao Qiu; Hui Yuan 002; Jinghong Zhang; Wentao Chen; Huazheng Wang; Mengdi Wang (AAAI, 2024)

MaxMin-RLHF: Alignment with Diverse Human Preferences. - Souradip Chakraborty; Jiahao Qiu; Hui Yuan 002; Alec Koppel; Dinesh Manocha; Furong Huang; Amrit S. Bedi; Mengdi Wang (ICML, 2024)

Conversational Dueling Bandits in Generalized Linear Models. - Shuhua Yang; Hui Yuan 002; Xiaoying Zhang; Mengdi Wang; Hong Zhang 037; Huazheng Wang (KDD, 2024)

Gradient Guidance for Diffusion Models: An Optimization Perspective. - Yingqing Guo; Hui Yuan 002; Yukang Yang; Minshuo Chen; Mengdi Wang (NeurIPS, 2024)

MaxMin-RLHF: Towards Equitable Alignment of Large Language Models with Diverse Human Preferences. - Souradip Chakraborty; Jiahao Qiu; Hui Yuan 002; Alec Koppel; Furong Huang; Dinesh Manocha; Amrit Singh Bedi; Mengdi Wang (CoRR, 2024)

Diffusion Model for Data-Driven Black-Box Optimization. - Zihao Li; Hui Yuan 002; Kaixuan Huang; Chengzhuo Ni; Yinyu Ye 001; Minshuo Chen; Mengdi Wang (CoRR, 2024)

Adversarial Attacks on Online Learning to Rank with Stochastic Click Models. - Zichen Wang; Rishab Balasubramanian; Hui Yuan 002; Chenyu Song; Mengdi Wang; Huazheng Wang (Trans. Mach. Learn. Res., 2024)

A Common Pitfall of Margin-based Language Model Alignment: Gradient Entanglement. - Hui Yuan 002; Yifan Zeng; Yue Wu; Huazheng Wang; Mengdi Wang; Liu Leqi (ICLR, 2025)

MATH-Perturb: Benchmarking LLMs' Math Reasoning Abilities against Hard Perturbations. - Kaixuan Huang; Jiacheng Guo; Zihao Li; Xiang Ji; Jiawei Ge 003; Wenzhe Li; Yingqing Guo; Tianle Cai; Hui Yuan 002; Runzhe Wang; Yue Wu; Ming Yin; et al. (CoRR, 2025)