Our new X account is live! Follow @wizwand_team for updates
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
Generative RecommendationToys
Recall@100.0359
23
Sequential RecommendationAmazon Office
N@52.49
15
Sequential RecommendationAmazon Toy
N@51.48
15
Sequential RecommendationAmazon Clothing
N@50.0047
15
Generative RecommendationSports
Recall@53.53
15
Sequential RecommendationAmazon-Book
N@50.72
15
Sequential RecommendationArT (test)
Hit@50.0875
13
Sequential RecommendationGame (test)
Hit@56.81
13
Sequential RecommendationInstrument (test)
Hit Rate@58.88
13
Generative RecommendationYelp (test)
HR@10.66
8
Showing 10 of 14 rows

Other info

Follow for update