Healthcare AI GYM for Medical Agents
On the map: Knowledge distillation → On-policy and self-distillation for LLMs
Abstract (arXiv)
Clinical reasoning demands multi-step interactions -- gathering patient history, ordering tests, interpreting results, and making safe treatment decisions -- yet a unified training environment provides the breadth of clinical domains and specialized tools to train generalizable medical AI agents through reinforcement learning remains elusive. We present a comprehensive empirical study of multi-turn agentic RL for medical AI, built on gym, a gymnasium-compatible environment spanning 10 clinical domains with 3.6K+ tasks, 135 domain-specific tools, and a knowledge base of 828K medical passages. Our analysis reveals that agentic multi-turn structure degrades into verbose single-turn monologues, characterized by monotonic length explosion and a simultaneous erosion of tool-use frequency. We characterize how this collapse, alongside distillation instability, stems from the misalignment of sparse terminal rewards with sequential clinical trajectories. We find that vanilla GRPO achieves strong final accuracy on some benchmarks but suffers from training instability, evidenced by significant oscillations in response length and prolonged convergence periods. To improve training efficiency and stability, we propose Turn-level Truncated On-Policy Distillation (TT-OPD), a self-distillation framework where a gradient-free EMA teacher leverages outcome-privileged information to provide dense, outcome-aware KL regularization at every conversation turn. TT-OPD achieves the best performance on 10 of 18 benchmarks with an average +3.9~pp improvement over the non-RL baseline with faster early convergence, controlled response length, and sustained multi-turn tool use.
Key facts. Cited by 4 papers in the map (data through 14 Oct 2026). Placed in On-policy and self-distillation for LLMs, one of 12 directions in Knowledge distillation. Builds on 48 of its 69 references that are in the map. First posted on arXiv 1 May 2026.
Builds on
48 of 69 references.
Closest ancestors · 4
Same direction, newest first
- 2026Self-Distilled RLVR3 Apr
- 2026Unifying Group-Relative and Self-Distillation Policy Optimization via Sample Routing2 Apr
- 2026A Survey of On-Policy Distillation for Large Language Models1 Apr
- 2026Self-Distilled Reasoner: On-Policy Self-Distillation for Large Language Models26 Jan
Other references · 44
- 2022Training language models to follow instructions with human feedbackcited by 15,971
- 2017Proximal Policy Optimization Algorithmscited by 15,215
- 2022Chain-of-Thought Prompting Elicits Reasoning in Large Language Modelscited by 12,459
- 2015Distilling the Knowledge in a Neural Networkcited by 10,324
- 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 9,379
- 2020Retrieval-Augmented Generation for Knowledge-Intensive NLP Taskscited by 9,297
- 2022ReAct: Synergizing Reasoning and Acting in Language Modelscited by 8,695
- 2023Direct Preference Optimization: Your Language Model is Secretly a Reward Modelcited by 8,065
- 2020Measuring Massive Multitask Language Understandingcited by 7,963
- 2022Self-Consistency Improves Chain of Thought Reasoning in Language Modelscited by 5,708
- 2024Qwen2.5 Technical Reportcited by 4,942
- 2023Toolformer: Language Models Can Teach Themselves to Use Toolscited by 4,182
32 more
- 2023Let's Verify Step by Stepcited by 3,789
- 2025DAPO: An Open-Source LLM Reinforcement Learning System at Scalecited by 2,810
- 2023ToolLLM: Facilitating Large Language Models to Master 16000+ Real-world APIscited by 1,947
- 2023Self-RAG: Learning to Retrieve, Generate, and Critique through Self-Reflectioncited by 1,839
- 2020What Disease does this Patient Have? A Large-scale Open Domain Question Answering Dataset from Medical Examscited by 1,356
- 2025Understanding R1-Zero-Like Training: A Critical Perspectivecited by 1,345
- 2022STaR: Bootstrapping Reasoning With Reasoningcited by 940
- 2022Solving math word problems with process- and outcome-based feedbackcited by 901
- 2023Capabilities of GPT-4 on Medical Challenge Problemscited by 638
- 2022MedMCQA : A Large-scale Multi-Subject Multi-Choice Dataset for Medical domain Question Answeringcited by 543
- 2020PathVQA: 30000+ Questions for Medical Visual Question Answeringcited by 475
- 2021SLAKE: A Semantically-Labeled Knowledge-Enhanced Dataset for Medical Visual Question Answeringcited by 449
- 2023Towards Expert-Level Medical Question Answering with Large Language Modelscited by 384
- 2025Demystifying Long Chain-of-Thought Reasoning in LLMscited by 326
- 2024HuatuoGPT-o1, Towards Medical Complex Reasoning with LLMscited by 265
- 2024AgentClinic: a multimodal agent benchmark to evaluate AI in simulated clinical environmentscited by 231
- 2024Agent Hospital: A Simulacrum of Hospital with Evolvable Medical Agentscited by 218
- 2023Rethinking Benchmark and Contamination for Language Models with Rephrased Samplescited by 195
- 2023MedCPT: Contrastive Pre-trained Transformers with Large-scale PubMed Search Logs for Zero-shot Biomedical Information Retrievalcited by 185
- 2025Agent-R: Training Language Model Agents to Reflect via Iterative Self-Trainingcited by 82
- 2024Direct Multi-Turn Preference Optimization for Language Agentscited by 64
- 2026CRISP: Compressed Reasoning via Iterative Self-Policy Distillationcited by 58
- 2024Improving Medical Reasoning through Retrieval and Self-Reflection with Retrieval-Augmented Large Language Modelscited by 34
- 2024K-QA: A Real-World Medical Q&A Benchmarkcited by 31
- 2025MedAgentGym: A Scalable Agentic Training Environment for Code-Centric Reasoning in Biomedical Data Sciencecited by 23
- 2025Iterative Tool Usage Exploration for Multimodal Agents via Step-wise Preference Tuningcited by 19
- 2026Learning to Hint for Reinforcement Learningcited by 12
- 2025DiaTool-DPO: Multi-Turn Direct Preference Optimization for Tool-Augmented Large Language Modelscited by 8
- 2025PGPO: Enhancing Agent Reasoning via Pseudocode-style Planning Guided Preference Optimizationcited by 8
- 2026MediX-R1: Open Ended Medical Reinforcement Learningcited by 7
- 2026MedOpenClaw and MedFlowBench: Auditing Medical Agents in Full-Study Workflowscited by 5
- 2025DEPO: Dual-Efficiency Preference Optimization for LLM Agentscited by 4
Built on by
Citing papers per year: 2026: 4.
4 papers.
- 2026A Formula-Driven Survey and Research Agenda for On-Policy Distillationcited by 6
- 2026OPDSearch+: On-Policy Distillation with RL Refinement for Search-Augmented Reasoningcited by 3
- 2026The Path to Self-Evolving Clinical Systems: Scaling Medical Agents from Assistance to Autonomycited by 2
- 2026OpenBioRQ: Unsolved Biomedical Research Questions for Agentscited by 0
- 2026OPDSearch+: On-Policy Distillation with RL Refinement for Search-Augmented Reasoning25 Aug 2026
- 2026The Path to Self-Evolving Clinical Systems: Scaling Medical Agents from Assistance to Autonomy13 Jul 2026
- 2026A Formula-Driven Survey and Research Agenda for On-Policy Distillation22 Jun 2026
- 2026OpenBioRQ: Unsolved Biomedical Research Questions for Agents20 Jun 2026
Similar, not cited
Close in wording, no citation either way
- 2025Doctor-R1: Mastering Clinical Inquiry with Experiential Agentic Reinforcement Learning
- 2025MMedAgent-RL: Optimizing Multi-Agent Collaboration for Multimodal Medical Reasoning
- 2026MAPLE: Elevating Medical Reasoning from Statistical Consensus to Process-Led Alignment
- 2026Scaling Medical Reasoning Verification via Tool-Integrated Reinforcement Learning
- 2026ResidencyRL: Reinforcement Learning in Simulated Clinical Environments
- 2025Real-World Doctor Agent with Proactive Consultation through Multi-Agent Reinforcement Learning
- 2025Beyond Prediction: Reinforcement Learning as the Defining Leap in Healthcare AI
- 2026Agentic AI in medicine: architectures, applications, evaluation, and challenges for clinical translation