mlsanityData to 14 Oct 2026

ProMed: Shapley Information Gain Guided Reinforcement Learning for Proactive Medical LLMs

arXiv 2508.13514 ↗First posted 19 Aug 2025cs.CL cs.AIcited by 13

Not on the map. Why?

Abstract (arXiv)

Interactive medical questioning is essential in clinical consultations, where physicians must actively gather necessary patient information. Yet existing medical Large Language Models (LLMs) predominantly follow a reactive paradigm, risking diagnostic errors by answering before seeking sufficient details. To bridge this gap, we propose ProMed, a reinforcement learning framework that transitions LLMs toward a proactive paradigm, enabling them to ask clinically valuable questions before decision-making. Central to ProMed is the Shapley Information Gain (SIG) reward, which quantifies a question's clinical utility as the amount of newly acquired information, while considering its contextual importance via Shapley values. We integrate SIG into a two-stage training pipeline: (1) SIG-Guided Model Initialization uses Monte Carlo Tree Search to construct high-reward interaction trajectories for supervision, and (2) SIG-Augmented Policy Optimization, with a novel SIG-guided Reward Distribution Mechanism that prioritizes informative questions for fine-grained optimization. Experiments on partial-information medical benchmarks show that ProMed significantly outperforms state-of-the-art methods by 6.29% on average and delivers a 54.45% gain over the reactive paradigm, and generalizes robustly to out-of-domain cases. Our codes are available at https://github.com/hxxding/ProMed.

Key facts. Cited by 13 papers in the map (data through 14 Oct 2026). Builds on 24 of its 56 references that are in the map. First posted on arXiv 19 Aug 2025.

Builds on

24 of 56 references.

References · 24

  1. 2017Proximal Policy Optimization Algorithmscited by 15,215
  2. 2024The Llama 3 Herd of Modelscited by 14,715
  3. 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 9,379
  4. 2025Qwen3 Technical Reportcited by 8,820
  5. 2023Direct Preference Optimization: Your Language Model is Secretly a Reward Modelcited by 8,065
  6. 2023Efficient Memory Management for Large Language Model Serving with PagedAttentioncited by 5,944
  7. 2025DAPO: An Open-Source LLM Reinforcement Learning System at Scalecited by 2,810
  8. 2024LlamaFactory: Unified Efficient Fine-Tuning of 100+ Language Modelscited by 1,464
  9. 2020What Disease does this Patient Have? A Large-scale Open Domain Question Answering Dataset from Medical Examscited by 1,356
  10. 2024ORPO: Monolithic Preference Optimization without Reference Modelcited by 562
  11. 2025Beyond the 80/20 Rule: High-Entropy Minority Tokens Drive Effective Reinforcement Learning for LLM Reasoningcited by 559
  12. 2025ReTool: Reinforcement Learning for Strategic Tool Use in LLMscited by 392
12 more
  1. 2024HuatuoGPT-o1, Towards Medical Complex Reasoning with LLMscited by 265
  2. 2024WebRL: Training LLM Web Agents via Self-Evolving Online Curriculum Reinforcement Learningcited by 256
  3. 2024Reinforcement learningcited by 247
  4. 2023HuatuoGPT, towards Taming Language Model to Be a Doctorcited by 214
  5. 2024MediQ: Question-Asking LLMs and a Benchmark for Reliable Interactive Clinical Reasoningcited by 154
  6. 2023CMB: A Comprehensive Medical Benchmark in Chinesecited by 115
  7. 2024HealthQ: Unveiling Questioning Capabilities of LLM Chains in Healthcare Conversationscited by 31
  8. 2024Learning to Ask Informative Questions: Enhancing LLMs with Preference Optimization and Expected Information Gaincited by 21
  9. 2024TPO: Aligning Large Language Models with Multi-branch & Multi-step Preference Treescited by 19
  10. 2025Ask Patients with Patience: Enabling LLMs for Human-Centric Medical Dialogue with Grounded Reasoningcited by 19
  11. 2025Magical: Medical Lay Language Generation via Semantic Invariance and Layperson-tailored Adaptationcited by 5
  12. 2025Dialogue is Better Than Monologue: Instructing Medical LLMs via Strategical Conversationscited by 4

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2025: 6; 2026: 7.

13 papers.

Sort:

By area:

Similar, not cited

Close in wording, no citation either way

  1. 2026Aligning Medical Conversational AI through Online Reinforcement Learning with Information-Theoretic Rewards
  2. 2025Med-U1: Incentivizing Unified Medical Reasoning in LLMs via Large-scale Reinforcement Learning
  3. 2025Beyond Distillation: Pushing the Limits of Medical LLM Reasoning with Minimalist Rule-Based RL
  4. 2026ConRub-Med: Reinforcement Learning with Consensus Rubrics for Open-Ended Medical Question Answering
  5. 2026Reinforcement Learning In Partially Verifiable Environments for Colon Cancer Treatment Planning
  6. 2025Adaptive Reinforcement for Open-ended Medical Reasoning via Semantic-Guided Reward Collapse Mitigation
  7. 2025Med-R3: Enhancing Medical Retrieval-Augmented Reasoning of LLMs via Progressive Reinforcement Learning
  8. 2026Eliciting Medical Reasoning with Knowledge-enhanced Data Synthesis: A Semi-Supervised Reinforcement Learning Approach