mlsanityData to 14 Oct 2026

Rewarding Reasoning, Not Answers: Fixing and Bounding Test-Time Reinforcement Learning on Medical QA

arXiv 2609.16660 ↗First posted 15 Sep 2026cs.CLcited by 0

Not on the map. Why?

Abstract (arXiv)

Test-time reinforcement learning adapts a model on its own unlabeled test set using majority-vote pseudo-labels and has shown strong results in mathematics. We show that this recipe collapses on medical multiple-choice QA: accuracy stagnates while output diversity rapidly declines. Through a controlled experiment that keeps the questions, model, and optimizer fixed while changing only the answer space, we trace this failure to answer-space structure rather than domain difficulty. In small answer spaces, incorrect rollouts often collide on the same wrong pseudo-label and reinforce it; in large answer spaces, they disperse and receive little reward. This diagnosis motivates PROSE, Process Reward Guided Self-Training, which rewards reasoning quality instead of answer agreement. PROSE scores each reasoning step with a medical process reward model, assigns the trajectory reward as the minimum score across steps, and enforces answer-format constraints. Without labels, PROSE substantially improves a general Llama model, surpassing purpose-built medical models and matching much larger systems. Because the process signal is internalized into the policy, the adapted model requires no reward model at inference and transfers its gains to unseen datasets. We further show that the minimum aggregation is essential: mean aggregation can be exploited, saturating the proxy reward while degrading accuracy.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 21 of its 30 references that are in the map. First posted on arXiv 15 Sep 2026.

Builds on

21 of 30 references.

References · 21

  1. 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 9,379
  2. 2025Qwen3 Technical Reportcited by 8,820
  3. 2023Let's Verify Step by Stepcited by 3,789
  4. 2024Gemini 1.5: Unlocking multimodal understanding across millions of tokens of contextcited by 3,224
  5. 2024Scaling LLM Test-Time Compute Optimally can be More Effective than Scaling Model Parameterscited by 1,986
  6. 2016Concrete Problems in AI Safetycited by 1,642
  7. 2020What Disease does this Patient Have? A Large-scale Open Domain Question Answering Dataset from Medical Examscited by 1,356
  8. 2022Scaling Laws for Reward Model Overoptimizationcited by 1,095
  9. 2025Does Reinforcement Learning Really Incentivize Reasoning Capacity in LLMs Beyond the Base Model?cited by 982
  10. 2022Solving math word problems with process- and outcome-based feedbackcited by 901
  11. 2018Confounding variables can degrade generalization performance of radiological deep learning modelscited by 379
  12. 2024HuatuoGPT-o1, Towards Medical Complex Reasoning with LLMscited by 265
9 more
  1. 2022DDXPlus: A New Dataset For Automatic Medical Diagnosiscited by 88
  2. 2020Annotation-efficient deep learning for automatic medical image segmentationcited by 79
  3. 2024UltraMedical: Building Specialized Generalists in Biomedicinecited by 67
  4. 2025ETTRL: Balancing Exploration and Exploitation in LLM Test-Time Reinforcement Learning Via Entropy Mechanismcited by 36
  5. 2025Med-PRM: Medical Reasoning Models with Stepwise, Guideline-verified Process Rewardscited by 24
  6. 2026TTCS: Test-Time Curriculum Synthesis for Self-Evolvingcited by 19
  7. 2025Continuous Self-Improvement of Large Language Models by Test-time Training with Verifier-Driven Sample Selectioncited by 13
  8. 2026CoVerRL: Breaking the Consensus Trap in Label-Free Reasoning via Generator-Verifier Co-Evolutioncited by 12
  9. 2025Rethinking Reward Models for Multi-Domain Test-Time Scalingcited by 10

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2025Med-U1: Incentivizing Unified Medical Reasoning in LLMs via Large-scale Reinforcement Learning
  2. 2026MAPLE: Elevating Medical Reasoning from Statistical Consensus to Process-Led Alignment
  3. 2025Rewarding the Journey, Not Just the Destination: A Composite Path and Answer Self-Scoring Reward Mechanism for Test-Time Reinforcement Learning
  4. 2026Reinforcement Learning for Evidence-Seeking Diagnostic Reasoning with Large Language Models
  5. 2026Scaling Medical Reasoning Verification via Tool-Integrated Reinforcement Learning
  6. 2025Adaptive Reinforcement for Open-ended Medical Reasoning via Semantic-Guided Reward Collapse Mitigation
  7. 2025MedS3: Towards Medical Slow Thinking with Self-Evolved Soft Dual-sided Process Supervision
  8. 2025Med-RLVR: Emerging Medical Reasoning from a 3B base model via reinforcement Learning