mlsanityData to 14 Oct 2026

Safe, Efficient, and Robust Reinforcement Learning for Ranking and Diffusion Models

arXiv 2510.15429 ↗First posted 17 Oct 2025cs.LGcited by 0

On the map: Diffusion models → Diffusion and flow policies for robotics

also in 1 more
Abstract (arXiv)

This dissertation investigates how reinforcement learning (RL) methods can be designed to be safe, sample-efficient, and robust. Framed through the unifying perspective of contextual-bandit RL, the work addresses two major application domains - ranking and recommendation, and text-to-image diffusion models. The first part of the thesis develops theory and algorithms for safe deployment in ranking systems. An exposure-based generalisation bound is derived, leading to a counterfactual risk-minimisation objective whose solution is guaranteed not to underperform the logging policy, even with sparse feedback. This guarantee is extended to doubly robust estimators, enabling safety even under adversarial or misspecified user models and offering practitioners explicit control over permissible utility loss. The second part turns to single-action bandits, where various off-policy estimators are unified within a baseline-correction framework. A closed-form optimal baseline is proposed and shown to minimise both evaluation and policy-gradient variance, thereby improving off-policy learning reliability. The final part examines the trade-offs between efficiency and effectiveness in generative RL. A systematic study of PPO and REINFORCE motivates the Leave-One-Out PPO (LOOP) algorithm, which combines multiple diffusion trajectories with a REINFORCE-style baseline inside PPO's clipped objective. LOOP achieves PPO-level sample efficiency while producing generations that align more faithfully with textual attributes.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Placed in Diffusion and flow policies for robotics, one of 14 directions in Diffusion models. Builds on 4 of its 5 references that are in the map. First posted on arXiv 17 Oct 2025.

Builds on

4 of 5 references.

Closest ancestors · 1

Same direction, newest first

  1. 2025A Simple and Effective Reinforcement Learning Method for Text-to-Image Diffusion Fine-tuning2 Mar

Other references · 3

  1. 2024Optimal Baseline Corrections for Off-Policy Contextual Banditscited by 13
  2. 2023Safe Deployment for Counterfactual Learning to Rank with Exposure-Based Risk Minimizationcited by 11
  3. 2024Practical and Robust Safety Guarantees for Advanced Counterfactual Learning to Rankcited by 7

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2024Safe and Robust Reinforcement Learning: Principles and Practice
  2. 2023Robust Reinforcement Learning Objectives for Sequential Recommender Systems
  3. 2023Bridging Distributionally Robust Learning and Offline RL: An Approach to Mitigate Distribution Shift and Partial Data Coverage
  4. 2026Adaptive Loss Balancing for Noise-Robust GRPO in Generative Recommendation
  5. 2024Online Bandit Learning with Offline Preference Data for Improved RLHF
  6. 2025Adversarial Diffusion for Robust Reinforcement Learning
  7. 2023Unified Off-Policy Learning to Rank: a Reinforcement Learning Perspective
  8. 2023Iterative Preference Learning from Human Feedback: Bridging Theory and Practice for RLHF under KL-Constraint