Healing with hierarchy: Hierarchical attention empowered graph neural networks for predictive analysis in medical data.

Journal: Artificial intelligence in medicine
PMID:

Abstract

In healthcare, predictive analysis using unstructured medical data is crucial for gaining insights into patient conditions and outcomes. However, unstructured data, which contains valuable patient information such as symptoms and medical histories, often presents challenges, including lengthy text sequences and incomplete data. To address these issues, we introduce a new framework named Hierarchical Attention-based Integrated Learning (HAIL), designed to predict in-hospital mortality and the duration of stay in the intensive care unit. HAIL combines hierarchical attention mechanisms with graph neural networks to effectively manage missing data and enhance outcome predictions. Our model iteratively refines embeddings, resulting in a more thorough analysis of electronic health record data. Experimental findings demonstrate a notable performance improvement of 2%-3% across various metrics when compared to existing benchmarks on standard datasets, highlighting HAIL's effectiveness in time-sensitive clinical decision-making. Additionally, our analysis underscores the significance of patient networks in maintaining the robustness and consistent performance of the HAIL framework.

Authors

  • Shivani Gupta
    Indian Institute of Technology Patna, Department of Computer Science and Engineering, Patna, 801103, Bihar, India. Electronic address: shivani_2221cs18@iitp.ac.in.
  • Saurabh Sharma
    Department of Surgery, Stanford School of Medicine, Stanford University Medical Center, Stanford, CA, United States.
  • Rajesh Sharma
    University of Tartu, Institute of Computer Science, Ülikooli 18-133, Tartu, 50090, Estonia. Electronic address: rajesh.sharma@ut.ee.
  • Joydeep Chandra
    Indian Institute of Technology Patna, Department of Computer Science and Engineering, Patna, 801103, Bihar, India. Electronic address: joydeep@iitp.ac.in.