Yimeng Bai

Yimeng Bai

Auto-generated profile, based on publicly available information

Yimeng Bai

University of Science and Technology of China, Hefei, China

Research Focus & Metrics

Academic Career Length

4Years (2023-Present)

(Calculated once Scilit publication data is synced: current year − year of first publication + 1)

Representative Publications

View All Publications
23 December 2024 in ACM Transactions on Recommender Systems
Verified
Journal Article
Open access
YZYang Zhang
Yang Zhang
ZHZhiyu Hu
Zhiyu Hu
YBYimeng Bai
Yimeng Bai
JWJiancan Wu
Jiancan Wu
QWQifan Wang
Qifan Wang
FFFuli Feng
Fuli Feng
Recommendation unlearning is an emerging task to serve users for erasing unusable data (e.g., some historical behaviors) from a well-trained recommender model. Existing methods process unlearning requests by fully or partially retraining the model after removing the unusable data. However, these methods are impractical due to the high computation cost of full retraining and the highly possible performance damage of partial training. In this light, a desired recommendation unlearning method should obtain a similar model as full retraining in a more efficient manner, i.e., achieving complete, efficient and harmless unlearning. In this work, we propose a new Influence Function-based Recommendation Unlearning (IFRU) framework, which efficiently updates the model without retraining by estimating the influence of the unusable data on the model via the influence function . In the light that recent recommender models use historical data for both the constructions of the optimization loss and the computational graph (e.g., neighborhood aggregation), IFRU jointly estimates the direct influence of unusable data on optimization loss and the spillover influence on the computational graph to pursue complete unlearning. Furthermore, we propose an importance-based pruning algorithm to reduce the cost of the influence function. IFRU is harmless and applicable to mainstream differentiable models. Extensive experiments demonstrate that IFRU achieves more than 250 times acceleration compared to retraining-based methods with recommendation performance comparable to full retraining. Codes are available at https://github.com/baiyimeng/IFRU.

21 October 2023
Verified
Conference Paper
Open access
YZYang Zhang
Yang Zhang
YBYimeng Bai
Yimeng Bai
JCJianxin Chang
Jianxin Chang
XZXiaoxue Zang
Xiaoxue Zang
SLSong Lu
Song Lu
YSYang Song
Yang Song
With the proliferation of short video applications, the significance of short video recommendations has vastly increased. Unlike other recommendation scenarios, short video recommendation systems heavily rely on feedback from watch time. Existing approaches simply treat watch time as a direct label, failing to effectively harness its extensive semantics and introduce bias, thereby limiting the potential for modeling user interests based on watch time. To overcome this challenge, we propose a framework named Debiased Multiple-semantics-extracting Labeling (DML). DML constructs labels that encompass various semantics by utilizing quantiles derived from the distribution of watch time, prioritizing relative order rather than absolute label values. This approach facilitates easier model learning while aligning with the ranking objective of recommendations. Furthermore, we introduce a method inspired by causal adjustment to refine label definitions, thereby directly mitigating bias at the label level. We substantiate the effectiveness of our DML framework through both online and offline experiments. Extensive results demonstrate that our DML could effectively leverage watch time to discover users' real interests, enhancing their engagement in our application.

4 March 2024
Verified
Conference Paper
Open access
YBYimeng Bai
Yimeng Bai
YZYang Zhang
Yang Zhang
JLJing Lu
Jing Lu
JCJianxin Chang
Jianxin Chang
XZXiaoxue Zang
Xiaoxue Zang
FFFuli Feng
Fuli Feng

24 August 2024
Verified
Conference Paper
Open access
YBYimeng Bai
Yimeng Bai
YZYang Zhang
Yang Zhang
FFFuli Feng
Fuli Feng
JLJing Lu
Jing Lu
XZXiaoxue Zang
Xiaoxue Zang
YSYang Song
Yang Song
Recommender systems require the simultaneous optimization of multiple objectives to accurately model user interests, necessitating the application of multi-task learning methods. However, existing multi-task learning methods in recommendations overlook the specific characteristics of recommendation scenarios, falling short in achieving proper gradient balance. To address this challenge, we set the target of multi-task learning as attaining the appropriate magnitude balance and the global direction balance, and propose an innovative methodology named GradCraft in response. GradCraft dynamically adjusts gradient magnitudes to align with the maximum gradient norm, mitigating interference from gradient magnitudes for subsequent manipulation. It then employs projections to eliminate gradient conflicts in directions while considering all conflicting tasks simultaneously, theoretically guaranteeing the global resolution of direction conflicts. GradCraft ensures the concurrent achievement of appropriate magnitude balance and global direction balance, aligning with the inherent characteristics of recommendation scenarios. Both offline and online experiments attest to the efficacy of GradCraft in enhancing multi-task performance in recommendations. The source code for GradCraft can be accessed at https://github.com/baiyimeng/GradCraft.

13 July 2025
Verified
Conference Paper
Open access
YBYimeng Bai
Yimeng Bai
SZShunyu Zhang
Shunyu Zhang
YZYang Zhang
Yang Zhang
HLHu Liu
Hu Liu
WBWentian Bao
Wentian Bao
WOWenwu Ou
Wenwu Ou

19 July 2026
Verified
Conference Paper
Open access
YBYimeng Bai
Yimeng Bai
CLChang Liu
Chang Liu
YZYang Zhang
Yang Zhang
DWDingxian Wang
Dingxian Wang
FYFrank Yang
Frank Yang
FFFuli Feng
Fuli Feng