OneRec: Unifying Retrieve and Rank with Generative Recommender and Iterative Preference Alignment

Authors: Jiaxin Deng, Shiyao Wang, Kuo Cai, Lejian Ren, Qigen Hu, Weifeng Ding, Qiang Luo, Guorui Zhou

License: CC BY 4.0
AI-generated keywords: OneRec

AI-generated Key Points

  • OneRec is a unified end-to-end generative framework for single-stage recommendation
  • Incorporates encoder-decoder architecture and scales up model parameters based on sparse Mixture-of-Experts (MoE) structure
  • Adopts session-wise list generation approach considering content and order of items within each session
  • Explores preference learning through direct preference optimization (DPO) using self-hard rejected samples from beam search results
  • Implements Iterative Preference Alignment (IPA) strategy to rank sampled responses based on pre-trained reward model scores
  • Extensive experiments demonstrate superiority in generating high-quality recommendations
  • Successfully deployed in Kuaishou, resulting in a significant 1.6% increase in watch-time
Also access our AI generated: Comprehensive summary, Lay summary, Blog-like article; or ask questions about this paper to our AI assistant.

Abstract: Recently, generative retrieval-based recommendation systems have emerged as a promising paradigm. However, most modern recommender systems adopt a retrieve-and-rank strategy, where the generative model functions only as a selector during the retrieval stage. In this paper, we propose OneRec, which replaces the cascaded learning framework with a unified generative model. To the best of our knowledge, this is the first end-to-end generative model that significantly surpasses current complex and well-designed recommender systems in real-world scenarios. Specifically, OneRec includes: 1) an encoder-decoder structure, which encodes the user's historical behavior sequences and gradually decodes the videos that the user may be interested in. We adopt sparse Mixture-of-Experts (MoE) to scale model capacity without proportionally increasing computational FLOPs. 2) a session-wise generation approach. In contrast to traditional next-item prediction, we propose a session-wise generation, which is more elegant and contextually coherent than point-by-point generation that relies on hand-crafted rules to properly combine the generated results. 3) an Iterative Preference Alignment module combined with Direct Preference Optimization (DPO) to enhance the quality of the generated results. Unlike DPO in NLP, a recommendation system typically has only one opportunity to display results for each user's browsing request, making it impossible to obtain positive and negative samples simultaneously. To address this limitation, We design a reward model to simulate user generation and customize the sampling strategy. Extensive experiments have demonstrated that a limited number of DPO samples can align user interest preferences and significantly improve the quality of generated results. We deployed OneRec in the main scene of Kuaishou, achieving a 1.6\% increase in watch-time, which is a substantial improvement.

Submitted to arXiv on 26 Feb. 2025

Ask questions about this paper to our AI assistant

You can also chat with multiple papers at once here.

AI assistant instructions?

Results of the summarizing process for the arXiv paper: 2502.18965v1

  • Comprehensive Summary
  • Key points
  • Layman's Summary
  • Blog article
, , , , This paper introduces OneRec, a unified end-to-end generative framework for single-stage recommendation, designed to address the challenges faced by traditional recommendation systems. Inspired by the scaling laws observed in training large language models, OneRec incorporates an encoder-decoder architecture and scales up its model parameters based on a sparse Mixture-of-Experts (MoE) structure. This allows OneRec to effectively capture user interests. Unlike conventional point-by-point prediction methods, OneRec adopts a session-wise list generation approach that considers both the relative content and order of items within each session. This eliminates the need for hand-crafted strategies and allows the model to autonomously learn optimal session structures from input data. To further improve the quality of generated recommendations, OneRec explores preference learning through direct preference optimization (DPO). To construct preference pairs, self-hard rejected samples are created from beam search results instead of random sampling. An Iterative Preference Alignment (IPA) strategy ranks sampled responses based on scores provided by a pre-trained reward model (RM), identifying the best-chosen and worst-rejected samples. Extensive experiments conducted on large-scale industry datasets demonstrate the superiority of OneRec in generating high-quality recommendations. The framework was successfully deployed in Kuaishou, a popular short video recommendation platform with millions of daily active users, resulting in a significant 1.6% increase in watch-time. Overall, OneRec represents a novel approach to recommendation systems that combines advanced generative modeling techniques with innovative preference alignment strategies to deliver superior performance in real-world scenarios.

Assess the quality of the AI-generated content by voting

Score: 0

Why do we need votes?

Votes are used to determine whether we need to re-run our summarizing tools. If the count reaches -10, our tools can be restarted.

The previous summary was created more than a year ago and can be re-run (if necessary) by clicking on the Run button below.

Similar papers summarized with our AI tools

Navigate through even more similar papers through a

tree representation

Disclaimer: The AI-based summarization tool and virtual assistant provided on this website may not always provide accurate and complete summaries or responses. We encourage you to carefully review and evaluate the generated content to ensure its quality and relevance to your needs.