Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

SPRec: Self-Play to Debias LLM-based Recommendation

About

Large language models (LLMs) have attracted significant attention in recommendation systems. Current work primarily applies supervised fine-tuning (SFT) to adapt the model for recommendation tasks. However, SFT on positive examples only limits the model's ability to align with user preference. To address this, researchers recently introduced Direct Preference Optimization (DPO), which explicitly aligns LLMs with user preferences using offline preference ranking data. However, we found that DPO inherently biases the model towards a few items, exacerbating the filter bubble issue and ultimately degrading user experience. In this paper, we propose SPRec, a novel self-play framework designed to mitigate over-recommendation and improve fairness without requiring additional data or manual intervention. In each self-play iteration, the model undergoes an SFT step followed by a DPO step, treating offline interaction data as positive samples and the predicted outputs from the previous iteration as negative samples. This effectively re-weights the DPO loss function using the model's logits, adaptively suppressing biased items. Extensive experiments on multiple real-world datasets demonstrate SPRec's effectiveness in enhancing recommendation accuracy and fairness. The implementation is available via https://github.com/RegionCh/SPRec

Chongming Gao, Ruijun Chen, Shuai Yuan, Kexin Huang, Yuanqing Yu, Xiangnan He• 2024

Related benchmarks

TaskDatasetResultRank
RecommendationMovieLens 1M (test)--
116
Sequential RecommendationMicroLens (test)
Recall@567.98
31
RecommendationNetflix (test)--
30
Sequential RecommendationAmazon Toy
N@51.48
24
Generative RecommendationYelp (test)
NG@51.24
24
Generative RecommendationToys
Recall@100.0359
23
RecommendationGoodreads
Hit Rate @ 51.94
20
Generative RecommendationBeauty (test)
Recall@53.7
16
Generative RecommendationInstruments (test)
R@57.73
16
Sequential RecommendationAmazon Office
N@52.49
15
Showing 10 of 33 rows

Other info

Follow for update