To Reason or Not to: Selective Chain-of-Thought in Medical Question Answering
Organizations: Department of Electrical and Computer Engineering, University of Minnesota, 200 Union St SE, 55455, Minneapolis, MN, United States · Division of Computational Health Sciences, Department of Surgery, University of Minnesota, 420 Delaware St SE, 55455, Minneapolis, MN, United States · Department of Computer Science and Engineering, University of Minnesota, 200 Union St SE, 55455, Minneapolis, MN, United States · Division of Biostatistics and Health Data Science, School of Public Health, University of Minnesota, 2221 University Ave SE, 55414, Minneapolis, MN, United States
Abstract
Objective: To improve the efficiency of medical question answering (MedQA) with large language models (LLMs) by avoiding unnecessary reasoning while maintaining accuracy. Methods: We propose Selective Chain-of-Thought (Selective CoT), an inference-time strategy that first predicts whether a question requires reasoning and generates a rationale only when needed. Two open-source LLMs (Llama-3.1-8B and Qwen-2.5-7B) were evaluated on four biomedical QA benchmarks-HeadQA, MedQA-USMLE, MedMCQA, and PubMedQA. Metrics included accuracy, total generated tokens, and inference time. Results: Selective CoT reduced inference time by 13-45% and token usage by 8-47% with minimal accuracy loss (4%). In some model-task pairs, it achieved both higher accuracy and greater efficiency than standard CoT. Compared with fixed-length CoT, Selective CoT reached similar or superior accuracy at substantially lower computational cost. Discussion: Selective CoT dynamically balances reasoning depth and efficiency by invoking explicit reasoning only when beneficial, reducing redundancy on recall-type questions while preserving interpretability. Conclusion: Selective CoT provides a simple, model-agnostic, and cost-effective approach for medical QA, aligning reasoning effort with question complexity to enhance real-world deployability of LLM-based clinical systems.
Figures & tables
| Dataset | Testset | Source |
| HeadQA | 244 | Spanish medical exams |
| MedQA-USMLE | 1,273 | US Medical exam |
| MedMCQA | 4,183 | Indian Medical exam |
| PubMedQA | 500 | Biomedical literature |
| Acc | # Tokens | Time (s) | ||||||
| Dataset | Model | Method | Score | Drop (%) | Value | Drop (%) | Value | Drop (%) |
| \multirow 4*HeadQA | \multirow 2*Llama-3.1-8B | CoT | 0.7090 | 135504 | 2695.1428 | |||
| Selective CoT | 0.6803 | 4.0462 | 136880 | -1.0155 | 2178.3438 | 19.1752 | ||
| \multirow 2*Qwen2.5-7B | CoT | 0.6598 | 93352 | 2574.1979 | ||||
| Selective CoT | 0.7172 | -8.6957 | 75600 | 19.0162 | 2120.5520 | 17.6228 | ||
| \multirow 4*MedMCQA | \multirow 2*Llama-3.1-8B | CoT | 0.5826 | 2184424 | 45140.2206 | |||
CoT Reasoning
Chain-of-thought (CoT) reasoning studies how language models use intermediate steps, visible or latent, to solve multi-step tasks.
See all 228 papers