News
16 May 2025
One paper is accepted by ACL 2025 on LLM personalization.
5 Apr 2025
One paper is accepted by SIGIR 2025 on calibration.
23 Sep 2024
One paper is accepted by ACM TORS on recommendation unlearning.
17 May 2024
One paper is accepted by KDD 2024 on multi-task learning for recommendation.
20 Oct 2023
One paper is accepted by WSDM 2024 about label learning for short video recommendation.
5 Aug 2023
One paper is accepted by CIKM 2023 about label debiasing for short video recommendation.
|
Yimeng Bai
Email: baiyimeng77@gmail.com • GitHub • Google Scholar • DBLP • ORCID
|
I am currently in the second year of my PhD program at Lab for Data Science, USTC, with a research focus on generative recommendation, and LLM personalization.
In the Year of 2024:
In the Year of 2023:
Education
| University of Science and Technology of China (USTC) PhD student in Cyber Science and Technology Sep. 2022 - Now, Hefei, China Advisor: Prof. Fuli Feng Mentor: Dr. Yang Zhang |
| Beijing Jiaotong University (BJTU) Bachelor in Science Sep. 2018 - Jun. 2022, Beijing, China |
Experiences
| Research Intern, Kuaishou, Dec. 2022 - Aug. 2024 Mentor: Dr. Jing Lu |
| Research Intern, ByteDance (TikTok-Data-Rec Core), Sep. 2024 - Now Mentor: Shaohui Ruan |
Preprints
 |
MERGE: Next-Generation Item Indexing Paradigm for Large-Scale Streaming Recommendation
Jing Yan, Yimeng Bai, Zongyu Liu, Yahui Liu, Junwei Wang, Jingze Huang, Haoda Li, Sihao Ding, Shaohui Ruan, Yang Zhang ArXiv 2026. |
|
UniGRec: Unified Generative Recommendation with Soft Identifiers for End-to-End Optimization
Jialei Li, Yang Zhang, Yimeng Bai, Shuai Zhu, Ziqi Xue, Xiaoyan Zhao, Dingxian Wang, Frank Yang, Andrew Rabinovich, Xiangnan He ArXiv 2026. Codes |
|
Unveiling Inference Scaling for Difference-Aware User Modeling in LLM Personalization
Suyu Chen, Yimeng Bai, Yulong Huang, Xiaoyan Zhao, Yang Zhang ArXiv 2025. Codes |
|
Bi-Level Optimization for Generative Recommendation: Bridging Tokenization and Generation
Yimeng Bai, Chang Liu, Yang Zhang, Dingxian Wang, Frank Yang, Andrew Rabinovich, Wenge Rong, Fuli Feng ArXiv 2025. |
|
Unconditional Diffusion for Generative Sequential Recommendation
Yimeng Bai, Yang Zhang, Sihao Ding, Shaohui Ruan, Han Yao, Fuli Feng, Tat-Seng Chua ArXiv 2025. Codes |
|
DiscRec: Disentangled Semantic-Collaborative Modeling for Generative Recommendation
Chang Liu, Yimeng Bai, Xiaoyan Zhao, Yang Zhang, Fuli Feng, Wenge Rong ArXiv 2025. Codes |
|
Causality-Enhanced Behavior Sequence Modeling in LLMs for Personalized Recommendation
Yang Zhang, Juntao You, Yimeng Bai, Jizhi Zhang, Keqin Bao, Wenjie Wang, Tat-Seng Chua ArXiv 2024. Codes |
Publications
In the Year of 2025: |
DuST: Dual-Stage Tuning of LLMs for Next-Post-Topic Prediction in Social Media Bots
Yulong Huang, Yimeng Bai, Yang Zhang, Fuli Feng, Haoyan Liu, Chongming Gao The 1st Workshop on LLM Agents for Social Simulation (CIKM 2025). |
|
Measuring What Makes You Unique: Difference-Aware User Modeling for Enhancing LLM Personalization
Yilun Qiu, Xiaoyan Zhao, Yang Zhang, Yimeng Bai, Wenjie Wang, Hong Cheng, Fuli Feng, Tat-Seng Chua ACL 2025 (Findings). Codes |
|
Unconstrained Monotonic Calibration of Predictions in Deep Ranking Systems
Yimeng Bai, Shunyu Zhang, Yang Zhang, Hu Liu, Wentian Bao, Enyun Yu, Fuli Feng, Wenwu Ou SIGIR 2025 (Full Paper, Acceptance Rate: 21.5%). Codes |
|
Recommendation Unlearning via Influence Function
Yang Zhang, Zhiyu Hu, Yimeng Bai, Jiancan Wu, Qifan Wang, Fuli Feng ACM TORS (Original Research). Codes |
|
GradCraft: Elevating Multi-task Recommendations through Holistic Gradient Crafting
Yimeng Bai, Yang Zhang, Fuli Feng, Jing Lu, Xiaoxue Zang, Chenyi Lei, Yang Song KDD 2024 (Applied Data Science, Acceptance Rate: 20%) Codes |
|
LabelCraft: Empowering Short Video Recommendations with Automated Label Crafting
Yimeng Bai, Yang Zhang, Jing Lu, Jianxin Chang, Xiaoxue Zang, Yanan Niu, Yang Song, Fuli Feng WSDM 2024 (Poster, Acceptance Rate: 18%). Codes |
|
Leveraging Watch-time Feedback for Short-Video Recommendations: A Causal Labeling Framework
Yang Zhang, Yimeng Bai, Jianxin Chang, Xiaoxue Zang, Song Lu, Jing Lu, Fuli Feng, Yanan Niu, Yang Song CIKM 2023 (Applied Research Paper, Acceptance Rate: 32%). Codes |
Awards
|
National Scholarship, 2024 - USTC, China |
Professional Services
| AC/SPC: KDD 2026 (ADS, Cycle 1) |
| PC member: KDD 2025 (ADS, Cycle 2); WWW 2026 (Industry); KDD 2026 (ADS, Cycle 1–2); SIGIR 2026; |