Skip to content

CoLa-ICD: A Knowledge-Enhanced Framework for Long-Tail Automated Medical Coding

Aug 2026 · 0 citations · 40 references
Computer Science

TL;DR

Experiments show that CoLa-ICD improves long-tail prediction with larger gains in larger and sparser label spaces and achieves state-of-the-art performance in AUC, F1, and P@k.

Abstract

Automatic medical coding assigns ICD codes to clinical notes, but it remains challenging due to long documents, imbalanced label distributions, and diverse terms. These challenges are especially severe for rare codes, which have limited training instances and are easily confused with semantically similar labels. We introduce CoLa-ICD, a knowledge-enhanced framework for long-tail prediction. CoLa-ICD enriches ICD labels with external terms, models dependencies among related codes, and learns stronger alignment between label semantics and clinical evidence for long-tail prediction. Experiments show that CoLa-ICD improves long-tail prediction with larger gains in larger and sparser label spaces and achieves state-of-the-art performance in AUC, F1, and P@k. Our code is available at https://github.com/youwillbethebest/Cola-ICD.

View source

Similar papers

#natural language process... Preprint Aug 2026

KREL: Automatic Medical Coding via Knowledge-Guided Reasoning over Clinical Evidence with LLMs

This work proposes Knowledge-Guided Reasoning over Clinical Evidence with LLMs (KREL), a framework that leverages LLMs for clinical text understanding and reasoning while integrating external ICD coding guidelines as structured knowledge, and enables tight coupling between domain knowledge and LLM reasoning.

Xubin Chen, Yi-Peng Zhou, Wenxin Sun et al. · 0 citations
Open access Sep 2026

MMAC-Net: A Multi-Modal Multi-Label Attention-Based Deep Learning Approach for Automated ICD-9 Coding of Rare Disease Admissions from Electronic Health Records

A multi-modal deep learning framework designed to enhance the retrospective assignment of ICD-9 codes to admissions involving rare pathologies, and underscores that incorporating structured clinical context partially mitigates the limitations of purely natural language processing approaches.

A. F. Ashrafi, Reda Alhajj, J. Rokne · 0 citations
Conference Aug 2026

Large Language Models for Automatic ICD-10 Coding in Cancer Treatment and Billing: An Academic-Industry Partnership

The increasing complexity of medical diagnosis and billing necessitates improving International Classification of Diseases (ICD) coding practices. It also requires partnerships between academia and industry to support rigor and implementation in real-world settings. We conducted an academic-industry collaboration invol...

M. Zitu, Dana Rollison, Kedar S. Kulkarni et al. · 0 citations
Preprint Aug 2026

xMICD: Explainable Representation of Multiple ICD Codes

xMICD provides a practical way to integrate embedding-based semantic relationships into interpretable clinical feature spaces for machine learning models, and achieves predictive performance comparable to embedding-based representations such as ICD2Vec across multiple clinical prediction tasks.

P. Vatiwutipong, Kumkup Keeratisiwakul, Albert Phuoc Kien Van Truong et al. · 0 citations

Related blog posts

We use cookies to run the site and, with your consent, for analytics and to show ads. See our Cookie Policy.