Fine-Tuning Diffusion-Based Recommender Systems via Reinforcement Learning with Reward Function Optimization
On the map: Diffusion models → Diffusion recommenders
also in 2 more
Abstract (arXiv)
Diffusion models recently emerged as a powerful paradigm for recommender systems, offering state-of-the-art performance by modeling the generative process of user-item interactions. However, training such models from scratch is both computationally expensive and yields diminishing returns once convergence is reached. To remedy these challenges, we propose ReFiT, a new framework that integrates Reinforcement learning (RL)-based Fine-Tuning into diffusion-based recommender systems. In contrast to prior RL approaches for diffusion models depending on external reward models, ReFiT adopts a task-aligned design: it formulates the denoising trajectory as a Markov decision process (MDP) and incorporates a collaborative signal-aware reward function that directly reflects recommendation quality. By tightly coupling the MDP structure with this reward signal, ReFiT empowers the RL agent to exploit high-order connectivity for fine-grained optimization, while avoiding the noisy or uninformative feedback common in naive reward designs. Leveraging policy gradient optimization, ReFiT maximizes exact log-likelihood of observed interactions, thereby enabling effective post hoc fine-tuning of diffusion recommenders. Comprehensive experiments on wide-ranging real-world datasets demonstrate that the proposed ReFiT framework (a) exhibits substantial performance gains over strong competitors (up to 36.3% on sequential recommendation), (b) demonstrates strong efficiency with linear complexity in the number of users or items, and (c) generalizes well across multiple diffusion-based recommendation scenarios. The source code and datasets are publicly available at https://anonymous.4open.science/r/ReFiT-4C60.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Placed in Diffusion recommenders, one of 14 directions in Diffusion models. Builds on 41 of its 59 references that are in the map. First posted on arXiv 10 Nov 2025.
Builds on
41 of 59 references.
Closest ancestors · 9
Same direction, newest first
- 2024S-Diff: An Anisotropic Diffusion Model for Collaborative Filtering in Spectral Domain31 Dec
- 2024Score-based Generative Diffusion Models for Social Recommendations20 Dec
- 2024RecDiff: Diffusion Model for Social Recommendation1 Jun
- 2024Collaborative Filtering Based on Diffusion Models: Unveiling the Potential of High-Order Connectivity22 Apr
- 2024Denoising Diffusion Recommender Model13 Jan
- 2023Graph Signal Diffusion Model for Collaborative Filtering15 Nov
- 2023Generate What You Prefer: Reshaping Sequential Recommendation via Guided Diffusion31 Oct
- 2023A Diffusion model for POI recommendation14 Apr
- 2023DiffuRec: A Diffusion Model for Sequential Recommendation3 Apr
Other references · 32
- 2014Adam: A Method for Stochastic Optimizationcited by 37,240
- 2020Denoising Diffusion Probabilistic Modelscited by 21,076
- 2021High-Resolution Image Synthesis with Latent Diffusion Modelscited by 18,863
- 2015Deep Unsupervised Learning using Nonequilibrium Thermodynamicscited by 7,058
- 2020LightGCN: Simplifying and Powering Graph Convolution Network for Recommendationcited by 1,139
- 2022Planning with Diffusion for Flexible Behavior Synthesiscited by 1,061
- 2023Training Diffusion Models with Reinforcement Learningcited by 982
- 2019Graph Neural Networks for Social Recommendationcited by 714
- 2019Neural Graph Collaborative Filteringcited by 661
- 2022Diffusion Policies as an Expressive Policy Class for Offline Reinforcement Learningcited by 562
- 2023Aligning Text-to-Image Models using Human Feedbackcited by 422
- 2018Variational Autoencoders for Collaborative Filteringcited by 310
20 more
- 2020Self-supervised Graph Learning for Recommendationcited by 305
- 2019Monte Carlo Gradient Estimation in Machine Learningcited by 283
- 2022A Survey on Generative Diffusion Modelcited by 240
- 2018An Introduction to Deep Reinforcement Learningcited by 206
- 2023Efficient Diffusion Policies for Offline Reinforcement Learningcited by 157
- 2023Contrastive Energy Prediction for Exact Energy-Guided Diffusion Sampling in Offline Reinforcement Learningcited by 149
- 2019Sequential Recommender Systems: Challenges, Progress and Prospectscited by 115
- 2023Diffusion Model is an Effective Planner and Data Synthesizer for Multi-Task Reinforcement Learningcited by 115
- 2023Diffusion Recommender Modelcited by 88
- 2023Optimizing DDPM Sampling with Shortcut Fine-Tuningcited by 81
- 2021Towards Long-term Fairness in Recommendationcited by 77
- 2019RecVAE: a New Variational Autoencoder for Top-N Recommendations with Implicit Feedbackcited by 63
- 2019Exploiting Cognitive Structure for Adaptive Learningcited by 37
- 2021Improved Fine-Tuning by Better Leveraging Pre-Training Datacited by 36
- 2018Large-scale Interactive Recommendation with Tree-structured Policy Gradientcited by 30
- 2024Diffusion Models for Generative Outfit Recommendationcited by 29
- 2020Neural Interactive Collaborative Filteringcited by 24
- 2023A General Offline Reinforcement Learning Framework for Interactive Recommendationcited by 18
- 2025Hyperbolic Diffusion Recommender Modelcited by 9
- 2021Reinforcement Learning with a Disentangled Universal Value Function for Item Recommendationcited by 4
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2025Energy-Guided Diffusion Sampling for Long-Term User Behavior Prediction in Reinforcement Learning-based Recommendation
- 2026ProRL: Effective Reinforcement Learning for Proactive Recommendation via Rectified Policy Gradient Estimation
- 2025On Efficiency-Effectiveness Trade-off of Diffusion-based Recommenders
- 2026LLM-Enhanced Reinforcement Learning for Long-Term User Satisfaction in Interactive Recommendation
- 2024A Survey on Diffusion Models for Recommender Systems
- 2026FlexRec: Adapting LLM-based Recommenders for Flexible Needs via Reinforcement Learning
- 2026ReCAST: Reward Credit Assignment across Timesteps for Online Diffusion Reinforcement
- 2023Model-enhanced Contrastive Reinforcement Learning for Sequential Recommendation