HieNet: Bidirectional Hierarchy Framework for Automated ICD Coding

Shi Wang, Daniel Tang*, Luchen Zhang, Huilin Li, Ding Han

*Corresponding author for this work

Research output: Chapter in Book/Report/Conference proceedingArticle in proceedingsResearchpeer-review


International Classification of Diseases (ICD) is a set of classification codes for medical records. Automated ICD coding, which assigns unique International Classification of Diseases codes with each medical record, is widely used recently for its efficiency and error-prone avoidance. However, there are challenges that remain such as heterogeneity, label unbalance, and complex relationships between ICD codes. In this work, we proposed a novel Bidirectional Hierarchy Framework(HieNet) to address the challenges. Specifically, a personalized PageRank routine is developed to capture the co-relation of codes, a bidirectional hierarchy passage encoder to capture the codes’ hierarchical representations, and a progressive predicting method is then proposed to narrow down the semantic searching space of prediction. We validate our method on two widely used datasets. Experimental results on two authoritative public datasets demonstrate that our proposed method boosts the state-of-the-art performance by a large margin.
Original languageEnglish
Title of host publicationProceedings of International Conference on Database Systems for Advanced Applications, part II
EditorsArnab Bhattacharya, Janice Lee Mong Li, Divyakant Agrawal, P. Krishna Reddy, Mukesh Mohania, Anirban Mondal, Vikram Goyal, Rage Uday Kiran
Number of pages17
Publication date2022
ISBN (Print)978-3-031-00125-3
ISBN (Electronic)978-3-031-00126-0
Publication statusPublished - 2022
EventInternational Conference on Database Systems for Advanced Applications - Hyderabad, India
Duration: 11 Apr 202214 Apr 2022
Conference number: 27


ConferenceInternational Conference on Database Systems for Advanced Applications
SeriesLecture Notes in Computer Science (including subseries Lecture Notes in Artificial Intelligence and Lecture Notes in Bioinformatics)


  • Structural encoder
  • ICD coding
  • Bidirectional passage retriever
  • Hierarchical embedding
  • Healthcare
  • Co-occurrence encoder


Dive into the research topics of 'HieNet: Bidirectional Hierarchy Framework for Automated ICD Coding'. Together they form a unique fingerprint.

Cite this