mlsanityData to 7 Oct 2026

Act First, Reason Later: Accelerating On-Policy Distillation for Multi-Turn Agents via Reference-Conditioned Inverse Dynamics

arXiv 2609.36608 ↗First posted 29 Sep 2026cs.LG cs.CLcited by 0

Not on the map. Why?

Abstract (arXiv)

On-policy distillation (OPD) trains multi-turn language agents with dense teacher supervision on student-generated responses. However, standard think-then-act rollouts require lengthy reasoning before each short action, delaying environment transitions and experience collection. Generating actions directly reduces this delay but can degrade rollout quality. To address this, we propose ActFirst-OPD, an act-first, reason-later training framework that decouples environment interaction from full-response generation. The student infers and executes actions through reference-conditioned inverse dynamics using its current interaction context and a reference next observation, and switches to autonomous next-action prediction when the resulting transition deviates from the reference trajectory. From the collected interaction contexts, the student asynchronously generates full think-then-act responses for token-level teacher supervision. Experiments across 0.6B-, 1.7B-, and 4B-parameter Qwen3 students show that ActFirst-OPD achieves average wall-clock training speedups of 2.3× on ALFWorld, 1.8× on WebShop, and 4.9× on ScienceWorld over Vanilla OPD. It matches or exceeds all compared OPD baselines in mean task success rate across eight of nine benchmark-model settings. These results demonstrate that reasoning need not block acting during multi-turn agent distillation.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 32 of its 36 references that are in the map. First posted on arXiv 29 Sep 2026.

Builds on

32 of 36 references.

References · 32

  1. 2017Proximal Policy Optimization Algorithmscited by 14,441
  2. 2022ReAct: Synergizing Reasoning and Acting in Language Modelscited by 8,355
  3. 2023Efficient Memory Management for Large Language Model Serving with PagedAttentioncited by 5,767
  4. 2024HybridFlow: A Flexible and Efficient RLHF Frameworkcited by 1,980
  5. 2022WebShop: Towards Scalable Real-World Web Interaction with Grounded Language Agentscited by 1,197
  6. 2017Curiosity-driven Exploration by Self-supervised Predictioncited by 1,173
  7. 2020ALFWorld: Aligning Text and Embodied Environments for Interactive Learningcited by 1,143
  8. 2026DeepSeek-V4: Towards Highly Efficient Million-Token Context Intelligencecited by 877
  9. 2023On-Policy Distillation of Language Models: Learning from Self-Generated Mistakescited by 840
  10. 2018Behavioral Cloning from Observationcited by 482
  11. 2025Group-in-Group Policy Optimization for LLM Agent Trainingcited by 442
  12. 2018TextWorld: A Learning Environment for Text-based Gamescited by 304
20 more
  1. 2022ScienceWorld: Is your Agent Smarter than a 5th Grader?cited by 298
  2. 2024Video Prediction Policy: A Generalist Robot Policy with Predictive Visual Representationscited by 279
  3. 2026Rethinking On-Policy Distillation of Large Language Models: Phenomenology, Mechanism, and Recipecited by 262
  4. 2023DeepSpeed Ulysses: System Optimizations for Enabling Training of Extreme Long Sequence Transformer Modelscited by 239
  5. 2019Mastering Complex Control in MOBA Games with Deep Reinforcement Learningcited by 150
  6. 2026A Survey of On-Policy Distillation for Large Language Modelscited by 148
  7. 2025SimpleTIR: End-to-End Reinforcement Learning for Multi-Turn Tool-Integrated Reasoningcited by 145
  8. 2026Code as Agent Harnesscited by 40
  9. 2026TCOD: Exploring Temporal Curriculum in On-Policy Distillation for Multi-turn Autonomous Agentscited by 38
  10. 2026Lightning OPD: Efficient Post-Training for Large Reasoning Models with Offline On-Policy Distillationcited by 28
  11. 2026On-Policy Distillation with Curriculum Turn-level Guidance for Multi-turn Agentscited by 21
  12. 2019RIDM: Reinforced Inverse Dynamics Modeling for Learning from a Single Observed Demonstrationcited by 19
  13. 2025Trinity-RFT: A General-Purpose and Unified Framework for Reinforcement Fine-Tuning of Large Language Modelscited by 19
  14. 2026TurnOPD: Making On-Policy Distillation Turn-Aware for Efficient Long-Horizon Agent Trainingcited by 14
  15. 2026Multi-Turn On-Policy Distillation with Prefix Replaycited by 12
  16. 2025Structured Agent Distillation for Large Language Modelcited by 11
  17. 2026SAGE-OPD: Selective Agent-Guided Intervention for Multi-Turn On-Policy Distillationcited by 11
  18. 2026Turning Video Models into Generalist Robot Policiescited by 10
  19. 2026Multi-Rollout On-Policy Distillation via Peer Successes and Failurescited by 8
  20. 2026Near-Policy: Accelerating On-Policy Distillation via Asynchronous Generation and Selective Packingcited by 3

Built on by

Not cited yet.