paper-with-me

홈 › Papers

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

2025-10-17 · Shashank Gupta 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.

📄 PDF Abstract BibTeX arXiv:2510.15429

Code (0)

등록된 구현이 없습니다.

Tasks

Reinforcement Learning

Similar Papers 제목 키워드 기반

Individual Control Barrier Functions-Guided Diffusion Model for Safe Offline Multi-Agent Reinforcement Learning

2026-06-10 · Qingyun Guo, Junyi Shi, Jianuo Huang, Tianyu Shi arxiv

Offline reinforcement learning allows control policies to be learned directly from data without online interaction, making it suitable for safety-critical tasks. Recent studies have applied diffusion models to offline re…

Multi-agent Reinforcement Learning

Alignment and Safety of Diffusion Models via Reinforcement Learning and Reward Modeling: A Survey

2025-05-23 · Preeti Lamba, Kiran Ravish, Ankita Kushwaha, Pawan Kumar

Diffusion models have emerged as leading generative models for images and other modalities, but aligning their outputs with human preferences and safety constraints remains a critical challenge. This thesis proposal inve…

Active LearningReinforcement Learning (RL)Safety Alignment

SafeDiffusion-R1: Online Reward Steering for Safe Diffusion Post-Training

2026-05-18 · Komal Kumar, Ankan Deria, Abhishek Basu, Fahad Shamshad 외 arxiv

Diffusion models have been widely studied for removing unsafe content learned during pre-training. Existing methods require expensive supervised data, either unsafe-text paired with safe-image groundtruth or negative/pos…

Reinforcement Learning

How Does the Lagrangian Guide Safe Reinforcement Learning through Diffusion Models?

2026-02-02 · Xiaoyuan Cheng, Wenxuan Yuan, Boyang Li, Yuanchao Xu 외 arxiv

Diffusion policy sampling enables reinforcement learning (RL) to represent multimodal action distributions beyond suboptimal unimodal Gaussian policies. However, existing diffusion-based RL methods primarily focus on off…

Reinforcement Learning

Diffusion Models for Offline Multi-agent Reinforcement Learning with Safety Constraints

2024-06-30 · Jianuo Huang

In recent advancements in Multi-agent Reinforcement Learning (MARL), its application has extended to various safety-critical scenarios. However, most methods focus on online learning, which presents substantial risks whe…

Multi-agent Reinforcement Learning