AIPO: Learning to Reason from Active Interaction
On the map: LLM reasoning → Reinforcement learning for reasoning
Abstract (arXiv)
Recent advances in LLMs have demonstrated strong reasoning capabilities, largely stimulated by RLVR. However, the exploration of existing RLVR algorithms remains largely constrained by the knowledge and reasoning strategies already accessible to the policy model. Although recent methods introduce external expert demonstrations to broaden exploration, they typically rely on complete trajectory-level guidance, which can be sample-inefficient, information-sparse, and insufficiently adaptive to intermediate reasoning bottlenecks. Inspired by collaborative multi-agent systems, we propose AIPO, an enhanced reinforcement learning framework that improves LLM reasoning through active multi-agent interaction during exploration. Specifically, when encountering reasoning bottlenecks, AIPO enables the policy model to proactively consult three functional collaborative agents, namely the verify agent, knowledge agent, and reasoning agent, thereby obtaining fine-grained and state-dependent guidance during rollout. The resulting mixed-policy trajectories expose the policy to reasoning directions that may be difficult to discover through isolated on-policy exploration. To learn effectively from collaborator-provided tokens, we further introduce a corrected importance sampling coefficient together with a lower-bound clipping strategy to mitigate off-policy discrepancy and vanishing gradients. After training, the policy model reasons independently without relying on collaborative agents. Extensive experiments across mathematical, scientific, coding, and puzzle reasoning benchmarks show that AIPO consistently improves reasoning performance and generalizes across different policy models, collaborator backbones, and RLVR algorithms.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Placed in Reinforcement learning for reasoning, one of 17 directions in LLM reasoning. Builds on 67 of its 73 references that are in the map. First posted on arXiv 8 May 2026.
Builds on
67 of 73 references.
Closest ancestors · 22
Same direction, newest first
- 2025Supervised Reinforcement Learning: From Expert Trajectories to Step-wise Reasoning29 Oct
- 2025Nudging the Boundaries of LLM Reasoning30 Sep
- 2025Beyond Two-Stage Training: Cooperative SFT and RL for LLM Reasoning8 Sep
- 2025SRFT: A Single-Stage Method with Supervised and Reinforcement Fine-Tuning for Reasoning24 Jun
- 2025Reasoning with Exploration: An Entropy Perspective17 Jun
- 2025Learning What Reinforcement Learning Can't: Interleaved Online Fine-Tuning for Hardest Questions9 Jun
- 2025Improving Data Efficiency for LLM Reinforcement Fine-tuning Through Difficulty-targeted Online Data Selection and Rollout Replay5 Jun
- 2025Rewarding the Unlikely: Lifting GRPO Beyond Distribution Sharpening3 Jun
- 2025KDRL: Post-Training Reasoning LLMs via Unified Knowledge Distillation and Reinforcement Learning2 Jun
- 2025Beyond the 80/20 Rule: High-Entropy Minority Tokens Drive Effective Reinforcement Learning for LLM Reasoning2 Jun
- 2025Enhancing Efficiency and Exploration in Reinforcement Learning for LLMs24 May
- 2025Learning to Reason under Off-Policy Guidance21 Apr
10 more
- 2025Does Reinforcement Learning Really Incentivize Reasoning Capacity in LLMs Beyond the Base Model?18 Apr
- 2025DeepMath-103K: A Large-Scale, Challenging, Decontaminated, and Verifiable Mathematical Dataset for Advancing Reasoning15 Apr
- 2025Echo Chamber: RL Post-training Amplifies Behaviors Learned in Pretraining10 Apr
- 2025VAPO: Efficient and Reliable Reinforcement Learning for Advanced Reasoning Tasks7 Apr
- 2025Open-Reasoner-Zero: An Open Source Approach to Scaling Up Reinforcement Learning on the Base Model31 Mar
- 2025Understanding R1-Zero-Like Training: A Critical Perspective26 Mar
- 2025DAPO: An Open-Source LLM Reinforcement Learning System at Scale18 Mar
- 2025Process Reinforcement through Implicit Rewards3 Feb
- 2023Let's Verify Step by Step31 May
- 2017Proximal Policy Optimization Algorithms20 Jul
Other references · 45
- 2022Training language models to follow instructions with human feedbackcited by 15,712
- 2022Chain-of-Thought Prompting Elicits Reasoning in Large Language Modelscited by 12,302
- 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 8,851
- 2025Qwen3 Technical Reportcited by 8,195
- 2023Efficient Memory Management for Large Language Model Serving with PagedAttentioncited by 5,767
- 2021Measuring Mathematical Problem Solving With the MATH Datasetcited by 5,203
- 2024Qwen2.5 Technical Reportcited by 4,713
- 2023Reflexion: Language Agents with Verbal Reinforcement Learningcited by 4,064
- 2021Program Synthesis with Large Language Modelscited by 3,509
- 2023GPQA: A Graduate-Level Google-Proof Q&A Benchmarkcited by 2,748
- 2024HybridFlow: A Flexible and Efficient RLHF Frameworkcited by 1,980
- 2024Scaling LLM Test-Time Compute Optimally can be More Effective than Scaling Model Parameterscited by 1,928
33 more
- 2024LiveCodeBench: Holistic and Contamination Free Evaluation of Large Language Models for Codecited by 1,906
- 2023A Survey on Hallucination in Large Language Models: Principles, Taxonomy, Challenges, and Open Questionscited by 1,582
- 2025Search-R1: Training LLMs to Reason and Leverage Search Engines with Reinforcement Learningcited by 1,371
- 2025s1: Simple test-time scalingcited by 1,311
- 2025Group Sequence Policy Optimizationcited by 788
- 2017Deep Learning Scaling is Predictable, Empiricallycited by 556
- 2020Scaling Laws for Autoregressive Generative Modelingcited by 510
- 2024Training Language Models to Self-Correct via Reinforcement Learningcited by 405
- 2026Self-Distilled Reasoner: On-Policy Self-Distillation for Large Language Modelscited by 386
- 2025Towards Reasoning Era: A Survey of Long Chain-of-Thought for Reasoning Large Language Modelscited by 381
- 2025R1-Searcher: Incentivizing the Search Capability in LLMs via Reinforcement Learningcited by 322
- 2025ZeroSearch: Incentivize the Search Capability of LLMs without Searchingcited by 203
- 2023ReMax: A Simple, Effective, and Efficient Reinforcement Learning Method for Aligning Large Language Modelscited by 184
- 2024O1 Replication Journey: A Strategic Progress Report -- Part 1cited by 140
- 2025REASONING GYM: Reasoning Environments for Reinforcement Learning with Verifiable Rewardscited by 121
- 2024O1 Replication Journey -- Part 2: Surpassing O1-preview through Simple Distillation, Big Progress or Bitter Lesson?cited by 90
- 2025On-Policy RL Meets Off-Policy Experts: Harmonizing Supervised Fine-Tuning and Reinforcement Learning via Dynamic Weightingcited by 88
- 2025ReMA: Learning to Meta-think for LLMs with Multi-Agent Reinforcement Learningcited by 78
- 2025Learn to Reason Efficiently with Adaptive Length-based Reward Shapingcited by 74
- 2024Are Your LLMs Capable of Stable Reasoning?cited by 65
- 2025Multi-Agent Evolve: LLM Self-Improve through Co-evolutioncited by 58
- 2025FlowReasoner: Reinforcing Query-Level Meta-Agentscited by 56
- 2025MARFT: Multi-Agent Reinforcement Fine-Tuningcited by 51
- 2025Weight Ensembling Improves Reasoning in Language Modelscited by 36
- 2025Towards a Unified View of Large Language Model Post-Trainingcited by 36
- 2025Reflect, Retry, Reward: Self-Improving LLMs via Reinforcement Learningcited by 32
- 2024Re-ReST: Reflection-Reinforced Self-Training for Language Agentscited by 29
- 2025CoMAS: Co-Evolving Multi-Agent Systems via Interaction Rewardscited by 27
- 2025TruthRL: Incentivizing Truthful LLMs via Reinforcement Learningcited by 20
- 2026Exploratory Memory-Augmented LLM Agent via Hybrid On- and Off-Policy Optimizationcited by 17
- 2025Incentivizing LLMs to Self-Verify Their Answerscited by 16
- 2025Interactive Learning for LLM Reasoningcited by 4
- 2025SituatedThinker: Grounding LLM Reasoning with Real-World through Situated Thinkingcited by 2
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2025Agentic Reinforced Policy Optimization
- 2025EAPO: Enhancing Policy Optimization with On-Demand Expert Assistance
- 2026BAPO: Boundary-Aware Policy Optimization for Reliable Agentic Search
- 2025Selective Expert Guidance for Effective and Diverse Exploration in Reinforcement Learning of LLMs
- 2025Navigate the Unknown: Enhancing LLM Reasoning with Intrinsic Motivation Guided Exploration
- 2025RAGEN: Understanding Self-Evolution in LLM Agents via Multi-Turn Reinforcement Learning
- 2025Agentic Policy Optimization via Instruction-Policy Co-Evolution
- 2025Reasoning through Exploration: A Reinforcement Learning Framework for Robust Function Calling