Su, Yi

14 publications

WACV 2025 CUNSB-RFIE: Context-Aware Unpaired Neural Schrodinger Bridge in Retinal Fundus Image Enhancement Xuanzhao Dong, Vamsi Krishna Vasa, Wenhui Zhu, Peijie Qiu, Xiwen Chen, Yi Su, Yujian Xiong, Zhangsihao Yang, Yanxi Chen, Yalin Wang

ICML 2025 EVOLvE: Evaluating and Optimizing LLMs for In-Context Exploration Allen Nie, Yi Su, Bo Chang, Jonathan Lee, Ed H. Chi, Quoc V Le, Minmin Chen

ICLR 2025 Training Language Models to Self-Correct via Reinforcement Learning Aviral Kumar, Vincent Zhuang, Rishabh Agarwal, Yi Su, John D Co-Reyes, Avi Singh, Kate Baumli, Shariq Iqbal, Colton Bishop, Rebecca Roelofs, Lei M Zhang, Kay McKinney, Disha Shrivastava, Cosmin Paduraru, George Tucker, Doina Precup, Feryal Behbahani, Aleksandra Faust

NeurIPSW 2024 Device-Directed Speech Detection for Follow-up Conversations Using Large Language Models Ognjen Rudovic, Pranay Dighe, Yi Su, Vineet Garg, Sameer Dharur, Xiaochuan Niu, Ahmed Hussen Abdelaziz, Saurabh Adya, Ahmed Tewfik

NeurIPS 2024 Online Feature Updates Improve Online (Generalized) Label Shift Adaptation Ruihan Wu, Siddhartha Datta, Yi Su, Dheeraj Baby, Yu-Xiang Wang, Kilian Q. Weinberger

WACV 2024 Ordinal Classification with Distance Regularization for Robust Brain Age Prediction Jay Shah, Md Mahfuzur Rahman Siddiquee, Yi Su, Teresa Wu, Baoxin Li

ICLR 2023 Offline RL for Natural Language Generation with Implicit Language Q Learning Charlie Victor Snell, Ilya Kostrikov, Yi Su, Sherry Yang, Sergey Levine

NeurIPS 2023 Unified Off-Policy Learning to Rank: A Reinforcement Learning Perspective Zeyu Zhang, Yi Su, Hui Yuan, Yiran Wu, Rishab Balasubramanian, Qingyun Wu, Huazheng Wang, Mengdi Wang

NeurIPS 2022 Data-Driven Offline Decision-Making via Invariant Representation Learning Han Qi, Yi Su, Aviral Kumar, Sergey Levine

MLOSS 2022 Tianshou: A Highly Modularized Deep Reinforcement Learning Library Jiayi Weng, Huayu Chen, Dong Yan, Kaichao You, Alexis Duburcq, Minghao Zhang, Yi Su, Hang Su, Jun Zhu

NeurIPS 2021 Online Adaptation to Label Distribution Shift Ruihan Wu, Chuan Guo, Yi Su, Kilian Q. Weinberger

ICML 2020 Adaptive Estimator Selection for Off-Policy Evaluation Yi Su, Pavithra Srinath, Akshay Krishnamurthy

ICML 2020 Doubly Robust Off-Policy Evaluation with Shrinkage Yi Su, Maria Dimakopoulou, Akshay Krishnamurthy, Miroslav Dudik

ICML 2019 CAB: Continuous Adaptive Blending for Policy Evaluation and Learning Yi Su, Lequn Wang, Michele Santacatterina, Thorsten Joachims