• Home
  • Search
  • Learning Through Mistakes: Explainable Medical Question-Answering using Reinforcement Learning in Large Language Models
  • https://doi.org/10.36227/techrxiv.175695836.60215814/v1Copy DOI Icon

Learning Through Mistakes: Explainable Medical Question-Answering using Reinforcement Learning in Large Language Models

  • Sep 4, 2025
  • Rasoul Abaryan
Show More
  • Abstract
  • Literature Map
  • Similar Papers
Abstract

Large Language Models (LLMs) have demonstrated strong performance in many natural language tasks, but robust and interpretable reasoning remains a challenge, especially in medical domains. This research applies a two-stage fine-tuning framework to improve the reasoning and explainability of LLMs on the MedMCQA (Pal, A., et al, 2022, MedMCQA) medical question-answering benchmark. In the first stage, standard Supervised Fine-Tuning (SFT) (Raffel et al., 2020), is used with structured prompts to guide the model to select the correct option (A, B, C, or D) and provide an accompanying explanation. In the second stage, the model is further refined using Group Relative Policy Optimization (GRPO) (Shao, Z., 2024, DeepSeekMath), with custom-designed reward functions targeting answer correctness, XML-format compliance using explicit and tags, and reasoning coherence. This reward-driven polishing step strengthens both factual accuracy and the structured Chain-of-Thought (CoT) reasoning ability. The base model used is Qwen-v2.5-0.5B (Yang, A., et al, 2024, Qwen), which after fine-tuning is referred to as BioXP-0.5B in this research. Using this method, our model achieved 64.58% Question-Answering accuracy a substantial improvement over the zero-shot baseline (~23%) and a 12.5-point gain over the supervised fine-tuning stage (~52%). Despite its small size, its performance approaches that of much larger models (e.g., PaLM-540B, Codex 5-shot), while also producing richer, more interpretable rationales. This work highlights the practicality of combining classical SFT with GRPO & structured reward optimization for reasoning enhancement and interpretability in domain-specific tasks. It offers novel insights into aligning LLM outputs with expert-level expectations under memory-efficient quantization settings and a practical recipe for structured reasoning alignment in low-resource medical NLP applications.

Similar Papers
  • Conference Article
  • Citations5

A Causal Explainable Guardrails for Large Language Models

  • Dec 02, 2024
  • Zhixuan Chu +5
  • Research Article
  • Citations22

Large Language Models in Medical Diagnostics: Scoping Review With Bibliometric Analysis.

  • Jun 09, 2025
  • Journal of medical Internet research
  • Hankun Su +14
  • Research Article

Evaluating large language models for clinical note processing: local fine-tuning and internal-external validation using electronic health records from South Asia.

  • Feb 25, 2026
  • BMC medical informatics and decision making
  • Seyed Alireza Hasheminasab +18
  • Research Article
  • Citations2

LEAF: Learning and Evaluation Augmented by Fact-Checking to Improve Factualness in Large Language Models.

  • Jan 01, 2025
  • Proceedings of the Conference on Empirical Methods in Natural Language Processing. Conference on Empirical Methods in Natural Language Processing
  • Hieu Tran +5
  • Research Article

Correctness Coverage Evaluation for Medical Multiple-Choice Question Answering Based on the Enhanced Conformal Prediction Framework

  • May 07, 2025
  • Mathematics
  • Yusong Ke +4
  • Research Article
  • Citations39

MedExpQA: Multilingual benchmarking of Large Language Models for Medical Question Answering

  • Jul 31, 2024
  • Artificial Intelligence In Medicine
  • Iñigo Alonso +2
  • Research Article

TencentLLMEval: A Hierarchical Evaluation of Real-World Capabilities for Human-Aligned LLMs

  • Apr 29, 2025
  • ACM Transactions on Intelligent Systems and Technology
  • Shuyi Xie +15
  • Research Article

Large language models in healthcare and biomedical informatics: A comprehensive review

  • Jan 01, 2026
  • Innovation and Emerging Technologies
  • Andrew Hornback +8
  • PDF
  • Research Article
  • Citations104

Benchmarking large language models for biomedical natural language processing applications and recommendations

  • Apr 06, 2025
  • Nature Communications
  • Qingyu Chen +20
  • Research Article

Retrieval-augmented generation in healthcare: A narrative review of methods, contributions, and future directions

  • Feb 24, 2026
  • Digital Medicine
  • Hao Kechun +1
  • Research Article

BEnchmarking Large Language Models for Ophthalmology (BELO): An Expert-Curated Data Set and Evaluation Framework for Knowledge and Reasoning

  • Dec 26, 2025
  • Ophthalmology Science
  • Sahana Srinivasan +31
  • Research Article
  • Citations3

TWOSOME: An Efficient Online Framework to Align LLMs with Embodied Environments via Reinforcement Learning

  • Jun 01, 2024
  • International Journal of Artificial Intelligence and Robotics Research
  • Weihao Tan +5
  • Research Article

Evaluating cultural knowledge processing in large language models: a cognitive benchmarking framework integrating retrieval-augmented generation

  • Nov 26, 2025
  • The Electronic Library
  • Hung-Shin Lee +3
  • Research Article
  • Citations9

A survey for large language models in biomedicine.

  • Dec 01, 2025
  • Artificial intelligence in medicine
  • Chong Wang +16
  • Research Article
  • Citations3

MedSumGraph: enhancing GraphRAG for medical QA with summarization and optimized prompts.

  • Feb 01, 2026
  • Artificial intelligence in medicine
  • Daeho Kim +2
Cactus Communications logo

Copyright 2026 Cactus Communications. All rights reserved.