Safety
CoLa-ICD: A Knowledge-Enhanced Framework for Long-Tail Automated Medical Coding
arXiv:2608.30234v1 Announce Type: new Abstract: Automatic medical coding assigns ICD codes to clinical notes, but it remains challenging due to long documents, imbalanced label distributions, and dive
arXiv:2608.30234v1 Announce Type: new 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.
Related
- MultiDx: A Multi-Source Knowledge Integration Framework towards Diagnostic Reasoning
- DAMEL: Dual-Axis Multi-Expert Learning for Class-Imbalanced Learning
- Know More, Know Clearer: A Meta-Cognitive Framework for Knowledge Augmentation in Large Language Models
- From Safety Documentation to Safety Knowledge Support: An Evidence-Grounded LLM Framework for Medical Devices
Source: arXiv cs.AI | 2026-09-01