Breaking barriers in ICD classification with a robust graph neural network for hierarchical coding.

Journal: Scientific reports
Published Date:

Abstract

The accurate classification of International Classification of Diseases (ICD) codes is a complex and critical multi-label task in clinical documentation, involving the assignment of diagnostic codes to medical discharge summaries. Existing automated methods face challenges due to the sparsity and nuanced nature of medical text, while traditional backpropagation-based models often lack flexibility and robustness. To address these issues, we propose Labeled Graph Generation with Node Representation Grasp (LGG-NRGrasp), an advanced adversarial learning framework that models ICD coding as a labeled graph generation problem. By leveraging a hierarchical structure to refine feature learning, our approach addresses the issue of over-smoothing in deep graph neural networks. A key innovation of LGG-NRGrasp is the integration of adversarial reinforcement learning and domain adaptation techniques, which enhance its ability to generalize across heterogeneous datasets. Extensive evaluations on benchmark datasets indicate that LGG-NRGrasp markedly surpasses leading models, exhibiting enhanced performance and dependability in automated ICD coding.

Authors

  • Suyang Xi
    School of Artificial Intelligence and Robotics, Xiamen University Malaysia, Sepang, Malaysia.
  • Jiesen Shi
    School of Artificial Intelligence and Robotics, Xiamen University Malaysia, Sepang, Malaysia.
  • Jiachen Yan
    School of Communication, Xiamen University Malaysia, Sepang, Malaysia.
  • MingJing Lin
    School of Artificial Intelligence and Robotics, Xiamen University Malaysia, Sepang, Malaysia.
  • Xinyi Zhou
    State Key Laboratory of Bioreactor Engineering, East China University of Science and Technology, Shanghai 200237, China.
  • Yuan Cheng
    Science Island Branch, University of Science and Technology of China, Hefei, Anhui, China.
  • Hong Ding
    Department of Ultrasound, Zhongshan Hospital, Fudan University, No. 180 Fenglin Road, Xuhui District, Shanghai, 200032, China. ding.hong@zs-hospital.sh.cn.
  • Chia Chao Kang
    School of Electrical Engineering and Artificial Intelligence, Xiamen University Malaysia, Sepang, Selangor, Malaysia.

Keywords

No keywords available for this article.