HealthNet: A Health Progression Network via Heterogeneous Medical Information Fusion

Published in IEEE TNNLS, CCF B, 2023

Recommended citation: Fuqiang Yu, Lizhen Cui, Huanhuan Chen, Yiming Cao, Ning Liu, Weiming Huang, Yonghui Xu, Hua Lu. "HealthNet: A Health Progression Network via Heterogeneous Medical Information Fusion," in IEEE Transactions on Neural Networks and Learning Systems, vol. 34, no. 10, pp. 6940-6954, Oct. 2023, doi: 10.1109/TNNLS.2022.3202305. https://ieeexplore.ieee.org/abstract/document/9887968

Numerous electronic health records (EHRs) offer valuable opportunities for understanding patients’ health status at different stages, namely health progression. Extracting the health progression patterns allows researchers to perform accurate predictive analysis of patient outcomes. However, most existing works on this task suffer from the following two limitations: 1) the diverse dependencies among heterogeneous medical entities are overlooked, which leads to the one-sided modeling of patients’ status and 2) the extraction granularity of patient’s health progression patterns is coarse, limiting the model’s ability to accurately infer the patient’s future status. To address these challenges, a pretrained Health progression network via heterogeneous medical information fusion, HealthNet, is proposed in this article. Specifically, a global heterogeneous graph in HealthNet is built to integrate heterogeneous medical entities and the dependencies among them. In addition, the proposed health progression network is designed to model hierarchical medical event sequences. By this method, the fine-grained health progression patterns of patients’ health can be captured. The experimental results on real disease datasets demonstrate that HealthNet outperforms the state-of-the-art models for both diagnosis prediction task and mortality prediction task. Download paper here