mlsanityData to 14 Oct 2026

Model selection meets clinical semantics: Optimizing ICD-10-CM prediction via LLM-as-Judge evaluation, redundancy-aware sampling, and section-aware fine-tuning

arXiv 2509.18846 ↗First posted 23 Sep 2025cs.AIcited by 0

Not on the map. Why?

Abstract (arXiv)

Accurate International Classification of Diseases (ICD) coding is critical for clinical documentation, billing, and healthcare analytics, yet it remains a labour-intensive and error-prone task. Although large language models (LLMs) show promise in automating ICD coding, their challenges in base model selection, input contextualization, and training data redundancy limit their effectiveness. We propose a modular framework for ICD-10 Clinical Modification (ICD-10-CM) code prediction that addresses these challenges through principled model selection, redundancy-aware data sampling, and structured input design. The framework integrates an LLM-as-judge evaluation protocol with Plackett-Luce aggregation to assess and rank open-source LLMs based on their intrinsic comprehension of ICD-10-CM code definitions. We introduced embedding-based similarity measures, a redundancy-aware sampling strategy to remove semantically duplicated discharge summaries. We leverage structured discharge summaries from Taiwanese hospitals to evaluate contextual effects and examine section-wise content inclusion under universal and section-specific modelling paradigms. Experiments across two institutional datasets demonstrate that the selected base model after fine-tuning consistently outperforms baseline LLMs in internal and external evaluations. Incorporating more clinical sections consistently improves prediction performance. This study uses open-source LLMs to establish a practical and principled approach to ICD-10-CM code prediction. The proposed framework provides a scalable, institution-ready solution for real-world deployment of automated medical coding systems by combining informed model selection, efficient data refinement, and context-aware prompting.

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

Builds on

21 of 44 references.

References · 21

  1. 2023Llama 2: Open Foundation and Fine-Tuned Chat Modelscited by 13,028
  2. 2023Judging LLM-as-a-Judge with MT-Bench and Chatbot Arenacited by 8,626
  3. 2023QLoRA: Efficient Finetuning of Quantized LLMscited by 3,418
  4. 2023Mistral 7Bcited by 3,164
  5. 2020Approximate Nearest Neighbor Negative Contrastive Learning for Dense Text Retrievalcited by 849
  6. 2021Deduplicating Training Data Makes Language Models Bettercited by 714
  7. 2024The Faiss librarycited by 376
  8. 2024BioMistral: A Collection of Open-Source Pretrained Large Language Models for Medical Domainscited by 301
  9. 2020MiniLMv2: Multi-Head Self-Attention Relation Distillation for Compressing Pretrained Transformerscited by 245
  10. 2018Characterizing and Avoiding Negative Transfercited by 229
  11. 2023D4: Improving LLM Pretraining via Document De-Duplication and Diversificationcited by 150
  12. 2020A Label Attention Model for ICD Coding from Clinical Textcited by 48
9 more
  1. 2023Principled Instructions Are All You Need for Questioning LLaMA-1/2, GPT-3.5/4cited by 48
  2. 2021Does the Magic of BERT Apply to Medical Code Assignment? A Quantitative Studycited by 39
  3. 2020Explainable Automated Coding of Clinical Notes using Hierarchical Label-wise Attention Networks and Label Embedding Initialisationcited by 38
  4. 2025Atla Selene Mini: A General Purpose Evaluation Modelcited by 27
  5. 2021Towards BERT-based Automatic ICD Coding: Limitations and Opportunitiescited by 25
  6. 2023Automated clinical coding using off-the-shelf large language modelscited by 14
  7. 2025Confidence and Stability of Global and Pairwise Scores in NLP Evaluationcited by 4
  8. 2023Clinical Text Deduplication Practices for Efficient Pretraining and Improved Clinical Taskscited by 3
  9. 2024Pre-Trained Model Recommendation for Downstream Fine-tuningcited by 1

Built on by

Not cited yet.