arXiv:2605.02943cs.LGcs.AI2026-05被引 5

构建医疗智能体训练环境,解决多轮决策中工具使用下降问题

Healthcare AI GYM for Medical Agents

论文配图:Healthcare AI GYM for Medical Agents
图 1 · 摘自论文原文
  • 设计10个临床领域、3600+任务的强化学习训练环境
  • 发现多轮对话易退化为冗长单轮,工具调用频率显著下降
  • 提出新方法提升训练稳定性,实现更快收敛与持续多步交互

临床推理需要多轮交互——采集病史、安排检查、解读结果并做出安全治疗决策——但目前仍缺乏统一的训练环境来覆盖广泛临床领域并支持通用医疗AI智能体通过强化学习进行训练。我们构建了一个基于 exttt{gym}的综合性多轮智能体强化学习实验平台,涵盖10个临床领域,包含3600多个任务、135种领域专用工具和82.8万条医学文本知识库。分析发现,多轮结构会退化为冗长的单轮独白,表现为响应长度持续膨胀,同时工具使用频率显著降低。我们揭示这种退化及蒸馏不稳定性源于稀疏的最终奖励与连续临床轨迹之间的错位。发现基础GRPO方法在部分基准上虽能达到高准确率,但存在训练震荡,响应长度波动大,收敛周期长。为此,我们提出转级截断在线蒸馏(TT-OPD),一种无需梯度的自我蒸馏框架,由一个基于经验移动平均的教师模型,在每一轮对话中利用结果导向信息提供密集的、面向结果的KL正则化。该方法在18个基准中的10个上表现最优,平均优于非强化学习基线3.9个百分点,且实现更早收敛、响应长度可控、多轮工具调用持续稳定。

原文摘要 · Abstract (English)

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.

医疗AI强化学习多轮对话智能体

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。