mlsanityData to 7 Oct 2026

From Documents to Spans: Scalable Supervision for Evidence-Based ICD Coding with LLMs

arXiv 2603.15270 ↗First posted 16 Mar 2026cs.CL cs.AIcited by 0

Not on the map. Why?

Abstract (arXiv)

International Classification of Diseases (ICD) coding assigns diagnosis codes to clinical documents and is essential for healthcare billing and clinical analysis. Reliable coding requires that each predicted code be supported by explicit textual evidence. However, existing public datasets provide only code labels, without evidence annotations, limiting models' ability to learn evidence-grounded predictions. In this work, we argue that dense, document-level evidence annotation is not always necessary for learning evidence-based coding. Instead, models can learn code-specific evidence patterns from local spans and use these patterns to support document-level evidence-based coding. Based on this insight, we propose Span-Centric Learning (SCL), a training framework that strengthens LLMs' coding ability at the span level and transfers this capability to full clinical documents. Specifically, we use a small set of annotated documents to supervise evidence recognition, aggregation, and code assignment, while leveraging a large collection of lightweight evidence spans to reinforce span-level reasoning. Due to their compactness, span annotations are scalable and can be further augmented through synthesis. Under the same Llama3.1-8B backbone, our approach achieves an 8.2-point improvement in macro-F1 at only 20% of the training cost of standard SFT, and provides explicit supporting evidence for each predicted code, enabling human auditing and revision.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 22 of its 29 references that are in the map. First posted on arXiv 16 Mar 2026.

Builds on

22 of 29 references.

References · 22

  1. 2022Chain-of-Thought Prompting Elicits Reasoning in Large Language Modelscited by 12,302
  2. 2023Efficient Memory Management for Large Language Model Serving with PagedAttentioncited by 5,767
  3. 2022Self-Consistency Improves Chain of Thought Reasoning in Language Modelscited by 5,601
  4. 2024LlamaFactory: Unified Efficient Fine-Tuning of 100+ Language Modelscited by 1,442
  5. 2018Explainable Prediction of Medical Codes from Clinical Textcited by 231
  6. 2023ACI-BENCH: a Novel Ambient Clinical Intelligence Dataset for Benchmarking Automatic Visit Note Generationcited by 116
  7. 2022Code Synonyms Do Matter: Multiple Synonyms Matching Network for Automatic ICD Codingcited by 52
  8. 2022PLM-ICD: Automatic ICD Coding with Pretrained Language Modelscited by 49
  9. 2023Automated Medical Coding on MIMIC-III and MIMIC-IV: A Critical Review and Replicability Studycited by 48
  10. 2023MDACE: MIMIC Documents Annotated with Code Evidencecited by 19
  11. 2024CoRelation: Boosting Automatic ICD Coding Through Contextualized Code Relation Learningcited by 17
  12. 2023Automated clinical coding using off-the-shelf large language modelscited by 14
10 more
  1. 2024An Unsupervised Approach to Achieve Supervised-Level Explainability in Healthcare Recordscited by 12
  2. 2024MedCodER: A Generative AI Assistant for Medical Codingcited by 10
  3. 2025Code Like Humans: A Multi-Agent Solution for Medical Codingcited by 10
  4. 2024Accurate and Well-Calibrated ICD Code Assignment Through Attention Over Diverse Label Embeddingscited by 9
  5. 2023DKEC: Domain Knowledge Enhanced Multi-Label Classification for Diagnosis Predictioncited by 8
  6. 2025A General Knowledge Injection Framework for ICD Codingcited by 7
  7. 2024Beyond Label Attention: Transparency in Language Models for Automated Medical Coding via Dictionary Learningcited by 5
  8. 2024Aligning AI Research with the Needs of Clinical Coding Workflows: Eight Recommendations Based on US Data Analysis and Critical Reviewcited by 5
  9. 2025Toward Reliable Clinical Coding with Language Models: Verification and Lightweight Adaptationcited by 4
  10. 2025RuCCoD: Towards Automated ICD Coding in Russiancited by 1

Built on by

Not cited yet.