ICDXML: enhancing ICD coding with probabilistic label trees and dynamic semantic representations.

Journal: Scientific reports
Published Date:

Abstract

Accurately assigning standardized diagnosis and procedure codes from clinical text is crucial for healthcare applications. However, this remains challenging due to the complexity of medical language. This paper proposes a novel model that incorporates extreme multi-label classification tasks to enhance International Classification of Diseases (ICD) coding. The model utilizes deformable convolutional neural networks to fuse representations from hidden layer outputs of pre-trained language models and external medical knowledge embeddings fused using a multimodal approach to provide rich semantic encodings for each code. A probabilistic label tree is constructed based on the hierarchical structure existing in ICD labels to incorporate ontological relationships between ICD codes and enable structured output prediction. Experiments on medical code prediction on the MIMIC-III database demonstrate competitive performance, highlighting the benefits of this technique for robust clinical code assignment.

Authors

  • Zeqiang Wang
    Department of Computing, Xi'an Jiaotong Liverpool University, Suzhou, 21500, China.
  • Yuqi Wang
    Graduate School at Shenzhen, Tsinghua University, Shenzhen, Guangdong, P.R. China.
  • Haiyang Zhang
    Department of Computer Science, University of Sheffield, UK.
  • Wei Wang
    State Key Laboratory of Quality Research in Chinese Medicine, Institute of Chinese Medical Sciences, University of Macau, Macau 999078, China.
  • Jun Qi
    Plastic Surgery Hospital, Chinese Academy of Medical Science, Peking Union Medical College, Beijing, China.
  • Jianjun Chen
    Department of Computing, Xi'an Jiaotong Liverpool University, Suzhou, 21500, China.
  • Nishanth Sastry
    School of Computer Science and Electronic Engineering, University of Surrey, Surrey, GU2 7XH, UK.
  • Jon Johnson
    UCL Social Research Institute, University College London, London, WC1E 6BT, UK.
  • Suparna De
    School of Computer Science and Electronic Engineering, University of Surrey, Surrey, GU2 7XH, UK. s.de@surrey.ac.uk.