Similarity-Aware Collaborative Learning for Patient Outcome Prediction
Published in DASFAA, CCF B, 2022
Recommended citation: Fuqiang Yu, Lizhen Cui*, Yiming Cao, Ning Liu*, Weiming Huang, Yonghui Xu. (2022). Similarity-Aware Collaborative Learning for Patient Outcome Prediction. In: Bhattacharya, A., et al. Database Systems for Advanced Applications. DASFAA 2022. Lecture Notes in Computer Science, vol 13246. Springer, Cham. https://doi.org/10.1007/978-3-031-00126-0_31 https://link.springer.com/chapter/10.1007/978-3-031-00126-0_31
The rapid growth in the use of electronic health records (EHR) offers promises for predicting patient outcomes. Previous works on this task focus on exploiting temporal patterns from sequential EHR data. Nevertheless, such approaches model patients independently, missing out on the similarities between patients, which are crucial for patients’ health risk assessment. Moreover, they fail to capture the fine-grained progression of patients’ status, which assist in inferring the patients’ future status. In this work, we propose a similarity-aware collaborative learning model SiaCo for patient outcome prediction. In particular, we design two similarity measurers and two global knowledge matrices to separately calculate the similarity of patients with different information levels and support collaborative learning between patients. To capture the more fine-grained progression of patients’ status, we design a parallelized LSTM to model the temporal-dependent patterns of patient status. Finally, SiaCo integrates the information learned from two measures and the parallelized LSTM to predict patient outcomes. Extensive experiments are conducted on two real disease datasets. The experimental results demonstrate that SiaCo outperforms the state-of-the-art models for two typical tasks of patient outcome prediction.