DS-1000: A Natural and Reliable Benchmark for Data Science Code Generation. - Yuhang Lai; Chengxi Li 011; Yiming Wang; Tianyi Zhang; Ruiqi Zhong; Luke Zettlemoyer; Wen-Tau Yih; Daniel Fried; Sida I. Wang; Tao Yu 009 (ICML, 2023)

ALaRM: Align Language Models via Hierarchical Rewards Modeling. - Yuhang Lai; Siyuan Wang; Shujun Liu; Xuanjing Huang 001; Zhongyu Wei (ACL (Findings), 2024)

EvoR: Evolving Retrieval for Code Generation. - Hongjin Su; Shuyang Jiang; Yuhang Lai; Haoyuan Wu; Boao Shi; Che Liu; Qian Liu; Tao Yu 009 (EMNLP (Findings), 2024)

ARKS: Active Retrieval in Knowledge Soup for Code Generation. - Hongjin Su; Shuyang Jiang; Yuhang Lai; Haoyuan Wu; Boao Shi; Che Liu; Qian Liu; Tao Yu 009 (CoRR, 2024)

HAF-RM: A Hybrid Alignment Framework for Reward Model Training. - Shujun Liu; Xiaoyu Shen; Yuhang Lai; Siyuan Wang; Shengbin Yue; Zengfeng Huang; Xuanjing Huang 001; Zhongyu Wei (ACL (1), 2025)

How Jailbreak Defenses Work and Ensemble? A Mechanistic Investigation. - Zhuohan Long; Siyuan Wang; Shujun Liu; Yuhang Lai; Xuanjing Huang 001; Zhongyu Wei (CoRR, 2025)