Can Reasoning LLMs Enhance Clinical Document Classification?
Akram Mustafa, Usman Naseem, Mostafa Rahimi Azghadi
TL;DR
The paper investigates whether reasoning LLMs can improve ICD-10 coding for clinical discharge summaries by comparing four reasoning and four non-reasoning models on a balanced MIMIC-IV-derived dataset preprocessed with cTAKES. It finds that reasoning models generally achieve higher accuracy and F1, with Gemini 2.0 Flash Thinking reaching 75% accuracy and 76% F1, while non-reasoning models show greater consistency across repeated runs. The results reveal a trade-off between accuracy and reliability, with well-defined codes showing strong performance and abstract codes exhibiting instability, suggesting a hybrid ensemble approach. The work highlights the need for domain-specific fine-tuning, multi-label extensions, and ensemble strategies to translate AI-assisted coding into robust clinical practice.
Abstract
Clinical document classification is essential for converting unstructured medical texts into standardised ICD-10 diagnoses, yet it faces challenges due to complex medical language, privacy constraints, and limited annotated datasets. Large Language Models (LLMs) offer promising improvements in accuracy and efficiency for this task. This study evaluates the performance and consistency of eight LLMs; four reasoning (Qwen QWQ, Deepseek Reasoner, GPT o3 Mini, Gemini 2.0 Flash Thinking) and four non-reasoning (Llama 3.3, GPT 4o Mini, Gemini 2.0 Flash, Deepseek Chat); in classifying clinical discharge summaries using the MIMIC-IV dataset. Using cTAKES to structure clinical narratives, models were assessed across three experimental runs, with majority voting determining final predictions. Results showed that reasoning models outperformed non-reasoning models in accuracy (71% vs 68%) and F1 score (67% vs 60%), with Gemini 2.0 Flash Thinking achieving the highest accuracy (75%) and F1 score (76%). However, non-reasoning models demonstrated greater stability (91% vs 84% consistency). Performance varied across ICD-10 codes, with reasoning models excelling in complex cases but struggling with abstract categories. Findings indicate a trade-off between accuracy and consistency, suggesting that a hybrid approach could optimise clinical coding. Future research should explore multi-label classification, domain-specific fine-tuning, and ensemble methods to enhance model reliability in real-world applications.
