Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

AIPO: Learning to Reason from Active Interaction

About

Recent advances in large language models (LLMs) have demonstrated remarkable reasoning capabilities, largely stimulated by Reinforcement Learning with Verifiable Rewards (RLVR). However, existing RL algorithms face a fundamental limitation: their exploration remains largely constrained by the inherent capability boundary of the policy model. Although recent methods introduce external expert demonstrations to extend this boundary, they typically rely on complete trajectory-level guidance, which is sample-inefficient, information-sparse, and may confine exploration to a static guidance space. Inspired by the potential of multi-agent systems, we propose $\textbf{AIPO}$, an enhanced reinforcement learning framework that improves LLM reasoning through active multi-agent interaction during exploration. Specifically, AIPO enables the policy model to proactively consult three functional collaborative agents, $\textit{Verify Agent}$, $\textit{Knowledge Agent}$, and $\textit{Reasoning Agent}$, when encountering reasoning bottlenecks, thereby receiving fine-grained and targeted guidance to actively expand its capability boundary during training. We further introduce a tailored importance sampling coefficient together with a clipping strategy to mitigate the off-policy bias and gradient vanishing issues that arise when learning from agent-provided feedback. After training, the policy model performs reasoning independently without relying on collaborative agents. Extensive experiments on diverse reasoning benchmarks, including AIME, MATH500, GPQA-Diamond, and LiveCodeBench, show that AIPO consistently improves reasoning performance, generalizes robustly across different policy models and RLVR algorithms, and effectively expands the reasoning capability boundary of the policy model.

Junnan Liu, Linhao Luo, Thuy-Trang Vu, Gholamreza Haffari• 2026

Related benchmarks

TaskDatasetResultRank
Mathematical ReasoningMATH500 (test)--
895
Code GenerationMBPP (test)--
405
Mathematical ReasoningAIME 2024 (test)--
209
Mathematical ReasoningAIME 2025 (test)--
148
MathMATH500
Score99.32
39
Mathematical ReasoningAIME 25--
33
Code GenerationLiveCodeBench v6 (test)
Avg@421.1
20
Mathematical ReasoningLiveMathBench v202505 (test)
Avg@417.5
20
Puzzle ReasoningReasoning-Gym (test)
Avg@417.8
20
CodeMBPP
Average Performance67.19
16
Showing 10 of 25 rows

Other info

Follow for update