ProMed: Shapley Information Gain Guided Reinforcement Learning for Proactive Medical LLMs
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
- 2017Proximal Policy Optimization Algorithmscited by 15,215
- 2024The Llama 3 Herd of Modelscited by 14,715
- 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 9,379
- 2025Qwen3 Technical Reportcited by 8,820
- 2023Direct Preference Optimization: Your Language Model is Secretly a Reward Modelcited by 8,065
- 2023Efficient Memory Management for Large Language Model Serving with PagedAttentioncited by 5,944
- 2025DAPO: An Open-Source LLM Reinforcement Learning System at Scalecited by 2,810
- 2024LlamaFactory: Unified Efficient Fine-Tuning of 100+ Language Modelscited by 1,464
- 2020What Disease does this Patient Have? A Large-scale Open Domain Question Answering Dataset from Medical Examscited by 1,356
- 2024ORPO: Monolithic Preference Optimization without Reference Modelcited by 562
- 2025Beyond the 80/20 Rule: High-Entropy Minority Tokens Drive Effective Reinforcement Learning for LLM Reasoningcited by 559
- 2025ReTool: Reinforcement Learning for Strategic Tool Use in LLMscited by 392
12 more
- 2024HuatuoGPT-o1, Towards Medical Complex Reasoning with LLMscited by 265
- 2024WebRL: Training LLM Web Agents via Self-Evolving Online Curriculum Reinforcement Learningcited by 256
- 2024Reinforcement learningcited by 247
- 2023HuatuoGPT, towards Taming Language Model to Be a Doctorcited by 214
- 2024MediQ: Question-Asking LLMs and a Benchmark for Reliable Interactive Clinical Reasoningcited by 154
- 2023CMB: A Comprehensive Medical Benchmark in Chinesecited by 115
- 2024HealthQ: Unveiling Questioning Capabilities of LLM Chains in Healthcare Conversationscited by 31
- 2024Learning to Ask Informative Questions: Enhancing LLMs with Preference Optimization and Expected Information Gaincited by 21
- 2024TPO: Aligning Large Language Models with Multi-branch & Multi-step Preference Treescited by 19
- 2025Ask Patients with Patience: Enabling LLMs for Human-Centric Medical Dialogue with Grounded Reasoningcited by 19
- 2025Magical: Medical Lay Language Generation via Semantic Invariance and Layperson-tailored Adaptationcited by 5
- 2025Dialogue is Better Than Monologue: Instructing Medical LLMs via Strategical Conversationscited by 4
Built on by
Citing papers per year: 2025: 6; 2026: 7.
13 papers.
Sort:
By area:
- 2025A Survey of Reinforcement Learning for Large Reasoning Modelscited by 180
- 2026ATPO: Adaptive Tree Policy Optimization for Multi-Turn Medical Dialoguecited by 8
- 2025MedKGI: Iterative Differential Diagnosis with Medical Knowledge Graphs and Information-Guided Inquiringcited by 7
- 2025ADEPT: Continual Pretraining via Adaptive Expansion and Dynamic Decoupled Tuningcited by 5
- 2025Proactive Reasoning-with-Retrieval Framework for Medical Multimodal Large Language Modelscited by 2
- 2026GraphWalker: Patient Analogy Meets Information Gain for Clinical Reasoning with Large Language Modelscited by 2
- 2026The Tell-Tale Norm: ℓ2 Magnitude as a Signal for Reasoning Dynamics in Large Language Modelscited by 2
- 2025Beyond MedQA: Towards Real-world Clinical Decision Making in the Era of LLMscited by 1
- 2026Search for Truth from Reasoning: A Dynamic Representation Editing Framework for Steering LLM Trajectoriescited by 1
- 2025GraphMed-LT: Patient-Specific Graph Memory with Latent Clinical Thought Refinement for Multi-Turn Medical Conversationscited by 0
- 2026Beyond Information Seeking: Severity-Aware Question Supervision for Proactive Medical Dialoguecited by 0
- 2026When LLM Meets Tree Search: A Systematic View of Inference as Search in Large Language Modelscited by 0
1 more
- 2026PCQC: Privileged Counterfactual Question Credit for Multi-Turn Medical Dialogue23 Sep 2026
- 2026When LLM Meets Tree Search: A Systematic View of Inference as Search in Large Language Models31 Aug 2026
- 2026Beyond Information Seeking: Severity-Aware Question Supervision for Proactive Medical Dialogue25 Aug 2026
- 2026Search for Truth from Reasoning: A Dynamic Representation Editing Framework for Steering LLM Trajectories26 Jun 2026
- 2026The Tell-Tale Norm: ℓ2 Magnitude as a Signal for Reasoning Dynamics in Large Language Models4 Jun 2026
- 2026GraphWalker: Patient Analogy Meets Information Gain for Clinical Reasoning with Large Language Models8 Apr 2026
- 2026ATPO: Adaptive Tree Policy Optimization for Multi-Turn Medical Dialogue10 Feb 2026
- 2025MedKGI: Iterative Differential Diagnosis with Medical Knowledge Graphs and Information-Guided Inquiring30 Dec 2025
- 2025Beyond MedQA: Towards Real-world Clinical Decision Making in the Era of LLMs22 Oct 2025
- 2025Proactive Reasoning-with-Retrieval Framework for Medical Multimodal Large Language Models21 Oct 2025
- 2025ADEPT: Continual Pretraining via Adaptive Expansion and Dynamic Decoupled Tuning11 Oct 2025
- 2025GraphMed-LT: Patient-Specific Graph Memory with Latent Clinical Thought Refinement for Multi-Turn Medical Conversations3 Oct 2025
1 more
- 2025A Survey of Reinforcement Learning for Large Reasoning Modelscited by 180
- 2025Proactive Reasoning-with-Retrieval Framework for Medical Multimodal Large Language Modelscited by 2
- 2026GraphWalker: Patient Analogy Meets Information Gain for Clinical Reasoning with Large Language Modelscited by 2
- 2026The Tell-Tale Norm: ℓ2 Magnitude as a Signal for Reasoning Dynamics in Large Language Modelscited by 2
- 2026Search for Truth from Reasoning: A Dynamic Representation Editing Framework for Steering LLM Trajectoriescited by 1
- 2025GraphMed-LT: Patient-Specific Graph Memory with Latent Clinical Thought Refinement for Multi-Turn Medical Conversationscited by 0
- 2026When LLM Meets Tree Search: A Systematic View of Inference as Search in Large Language Modelscited by 0
- 2026When LLM Meets Tree Search: A Systematic View of Inference as Search in Large Language Models31 Aug 2026
- 2026Search for Truth from Reasoning: A Dynamic Representation Editing Framework for Steering LLM Trajectories26 Jun 2026
- 2026The Tell-Tale Norm: ℓ2 Magnitude as a Signal for Reasoning Dynamics in Large Language Models4 Jun 2026
- 2026GraphWalker: Patient Analogy Meets Information Gain for Clinical Reasoning with Large Language Models8 Apr 2026
- 2025Proactive Reasoning-with-Retrieval Framework for Medical Multimodal Large Language Models21 Oct 2025
- 2025GraphMed-LT: Patient-Specific Graph Memory with Latent Clinical Thought Refinement for Multi-Turn Medical Conversations3 Oct 2025
- 2025A Survey of Reinforcement Learning for Large Reasoning Models10 Sep 2025
- 2026ATPO: Adaptive Tree Policy Optimization for Multi-Turn Medical Dialoguecited by 8
- 2025MedKGI: Iterative Differential Diagnosis with Medical Knowledge Graphs and Information-Guided Inquiringcited by 7
- 2025ADEPT: Continual Pretraining via Adaptive Expansion and Dynamic Decoupled Tuningcited by 5
- 2025Beyond MedQA: Towards Real-world Clinical Decision Making in the Era of LLMscited by 1
- 2026Beyond Information Seeking: Severity-Aware Question Supervision for Proactive Medical Dialoguecited by 0
- 2026PCQC: Privileged Counterfactual Question Credit for Multi-Turn Medical Dialoguecited by 0
- 2026PCQC: Privileged Counterfactual Question Credit for Multi-Turn Medical Dialogue23 Sep 2026
- 2026Beyond Information Seeking: Severity-Aware Question Supervision for Proactive Medical Dialogue25 Aug 2026
- 2026ATPO: Adaptive Tree Policy Optimization for Multi-Turn Medical Dialogue10 Feb 2026
- 2025MedKGI: Iterative Differential Diagnosis with Medical Knowledge Graphs and Information-Guided Inquiring30 Dec 2025
- 2025Beyond MedQA: Towards Real-world Clinical Decision Making in the Era of LLMs22 Oct 2025
- 2025ADEPT: Continual Pretraining via Adaptive Expansion and Dynamic Decoupled Tuning11 Oct 2025
Similar, not cited
Close in wording, no citation either way
- 2026Aligning Medical Conversational AI through Online Reinforcement Learning with Information-Theoretic Rewards
- 2025Med-U1: Incentivizing Unified Medical Reasoning in LLMs via Large-scale Reinforcement Learning
- 2025Beyond Distillation: Pushing the Limits of Medical LLM Reasoning with Minimalist Rule-Based RL
- 2026ConRub-Med: Reinforcement Learning with Consensus Rubrics for Open-Ended Medical Question Answering
- 2026Reinforcement Learning In Partially Verifiable Environments for Colon Cancer Treatment Planning
- 2025Adaptive Reinforcement for Open-ended Medical Reasoning via Semantic-Guided Reward Collapse Mitigation
- 2025Med-R3: Enhancing Medical Retrieval-Augmented Reasoning of LLMs via Progressive Reinforcement Learning
- 2026Eliciting Medical Reasoning with Knowledge-enhanced Data Synthesis: A Semi-Supervised Reinforcement Learning Approach