Score: 1

Graph Inference Towards ICD Coding

Published: January 12, 2026 | arXiv ID: 2601.07496v1

By: Xiaoxiao Deng

Potential Business Impact:

Helps doctors quickly give correct patient codes.

Business Areas:
Machine Learning Artificial Intelligence, Data and Analytics, Software

Automated ICD coding involves assigning standardized diagnostic codes to clinical narratives. The vast label space and extreme class imbalance continue to challenge precise prediction. To address these issues, LabGraph is introduced -- a unified framework that reformulates ICD coding as a graph generation task. By combining adversarial domain adaptation, graph-based reinforcement learning, and perturbation regularization, LabGraph effectively enhances model robustness and generalization. In addition, a label graph discriminator dynamically evaluates each generated code, providing adaptive reward feedback during training. Experiments on benchmark datasets demonstrate that LabGraph consistently outperforms previous approaches on micro-F1, micro-AUC, and P@K.

Page Count
4 pages

Category
Computer Science:
Machine Learning (CS)