DRIFT: Decoupled Rollouts and Importance-Weighted Fine-Tuning for Efficient Multi-Turn Optimization
Not on the map. Why?
Abstract (arXiv)
Large language models are increasingly deployed in multi-turn interactive settings where users or environments can iteratively provide lightweight feedback. Unfortunately, optimizing such behavior presents a sharp dilemma in practice: online reinforcement learning is able to effectively address multi-turn dynamics but is prohibitively expensive due to the cost of generating full correction trajectories at every update, whereas offline supervised fine-tuning (SFT) is efficient but suffers from distribution shift and behavioral collapse. To this end, we novelly propose DRIFT (Decoupled Rollouts and Importance-Weighted Fine-Tuning), a framework that operationalizes the theoretical insight that the KL-regularized RL objective is equivalent to importance-weighted supervised learning. DRIFT decouples rollout from optimization by sampling offline interaction trajectories from a fixed reference policy, deriving return-based importance weights, and optimizing the policy via weighted SFT on the resulting dataset. Empirically, we demonstrate that DRIFT matches or exceeds the performance of multi-turn reinforcement learning baselines while maintaining the training efficiency and simplicity of standard supervised fine-tuning. Code is available at https://github.com/2020-qqtcg/DRIFT.
Key facts. Cited by 1 paper in the map (data through 14 Oct 2026). Builds on 32 of its 37 references that are in the map. First posted on arXiv 29 May 2026.
Builds on
32 of 37 references.
References · 32
- 2022Training language models to follow instructions with human feedbackcited by 15,971
- 2017Proximal Policy Optimization Algorithmscited by 15,215
- 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 9,379
- 2023Direct Preference Optimization: Your Language Model is Secretly a Reward Modelcited by 8,065
- 2025DeepSeek-R1: Incentivizing Reasoning Capability in LLMs via Reinforcement Learningcited by 6,330
- 2021Measuring Mathematical Problem Solving With the MATH Datasetcited by 5,371
- 2023Reflexion: Language Agents with Verbal Reinforcement Learningcited by 4,276
- 2025Gemini 2.5: Pushing the Frontier with Advanced Reasoning, Multimodality, Long Context, and Next Generation Agentic Capabilitiescited by 3,796
- 2023Self-Refine: Iterative Refinement with Self-Feedbackcited by 3,442
- 2023GPQA: A Graduate-Level Google-Proof Q&A Benchmarkcited by 2,879
- 2024MMLU-Pro: A More Robust and Challenging Multi-Task Language Understanding Benchmarkcited by 1,848
- 2025Group Sequence Policy Optimizationcited by 835
20 more
- 2019Advantage-Weighted Regression: Simple and Scalable Off-Policy Reinforcement Learningcited by 782
- 2023MetaMath: Bootstrap Your Own Mathematical Questions for Large Language Modelscited by 735
- 2018Reinforcement Learning and Control as Probabilistic Inference: Tutorial and Reviewcited by 482
- 2025LLMs Get Lost In Multi-Turn Conversationcited by 429
- 2024Training Language Models to Self-Correct via Reinforcement Learningcited by 426
- 2023Beyond Human Data: Scaling Self-Training for Problem-Solving with Language Modelscited by 349
- 2025Reinforcement Learning with Verifiable Rewards Implicitly Incentivizes Correct Reasoning in Base LLMscited by 289
- 2022Generating Sequences by Learning to Self-Correctcited by 277
- 2024Are We Done with MMLU?cited by 194
- 2023TheoremQA: A Theorem-driven Question Answering datasetcited by 192
- 2016Reward Augmented Maximum Likelihood for Neural Structured Predictioncited by 81
- 2025Beyond Single-Turn: A Survey on Multi-Turn Interactions with Large Language Modelscited by 62
- 2024Confidence Matters: Revisiting Intrinsic Self-Correction Capabilities of Large Language Modelscited by 47
- 2024Understanding the Dark Side of LLMs' Intrinsic Self-Correctioncited by 42
- 2025Supervised Fine Tuning on Curated Data is Reinforcement Learning (and can be improved)cited by 38
- 2024Regressing the Relative Future: Efficient Policy Optimization for Multi-turn RLHFcited by 30
- 2025Online Preference Alignment for Language Models via Count-based Explorationcited by 26
- 2025RLHF in an SFT Way: From Optimal Solution to Reward-Weighted Alignmentcited by 14
- 2025A Simple "Try Again" Can Elicit Multi-Turn LLM Reasoningcited by 11
- 2025Offline RL by Reward-Weighted Fine-Tuning for Conversation Optimizationcited by 10
Built on by
Citing papers per year: 2026: 1.
1 paper.
Similar, not cited
Close in wording, no citation either way
- 2025Stabilizing Off-Policy Training for Long-Horizon LLM Agent via Turn-Level Importance Sampling and Clipping-Triggered Normalization
- 2026RTPO: Reverse-Turn Policy Optimization for Stabilizing Agentic RL Training
- 2026TSR: Trajectory-Search Rollouts for Multi-Turn RL of LLM Agents
- 2025Information Gain-based Policy Optimization: A Simple and Effective Approach for Multi-Turn Search Agents
- 2026RSPO: Reward-Swap Policy Optimization for Multi-Turn LLM Agents
- 2024Q-SFT: Q-Learning for Language Models via Supervised Fine-Tuning
- 2025One-Token Rollout: Guiding Supervised Fine-Tuning of LLMs with Policy Gradient
- 2026Active Offline-to-Online Reinforcement Learning