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

Yihang Cheng, Veronica Liesaputra, Andrew Trotman


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.
Anthology ID:
2026.findings-emnlp.731
Volume:
Findings of the Association for Computational Linguistics: EMNLP 2026
Month:
October
Year:
2026
Address:
Budapest, Hungary
Editors:
Hung-Yi Lee, Sunipa Dev, Jessy Li, Desmond Elliott
Venue:
Findings
SIG:
SIGDAT
Publisher:
Association for Computational Linguistics
Note:
Pages:
15368–15380
Language:
URL:
https://aclanthology.org/2026.findings-emnlp.731/
DOI:
Bibkey:
Cite (ACL):
Yihang Cheng, Veronica Liesaputra, and Andrew Trotman. 2026. CoLa-ICD: A Knowledge-Enhanced Framework for Long-Tail Automated Medical Coding. In Findings of the Association for Computational Linguistics: EMNLP 2026, pages 15368–15380, Budapest, Hungary. Association for Computational Linguistics.
Cite (Informal):
CoLa-ICD: A Knowledge-Enhanced Framework for Long-Tail Automated Medical Coding (Cheng et al., Findings 2026)
Copy Citation:
PDF:
https://aclanthology.org/2026.findings-emnlp.731.pdf