Heterogeneous neural metric learning for spatio-temporal modeling of infectious diseases with incomplete data

Qi Tan,Yang Liu,Jiming Liu,Benyun Shi,Shang Xia,Xiao-Nong Zhou
DOI: https://doi.org/10.1016/j.neucom.2019.12.145
IF: 6
2021-10-01
Neurocomputing
Abstract:<p>Infectious disease data, recording the numbers of infection cases in different locations and time, is one of the most typical categories of spatio-temporal data and plays an important role in the infectious disease control and prevention. However, due to the insufficient resources and manpower, the observations and records of infection cases are inevitably missing in some locations and time, which brings difficulties to the accurate risk assessment and timely disease control. Moreover, imputing the missing infectious disease data is challenging as the infectious disease diffusion can be potentially caused and affected by many risk factors. To address the above-mentioned challenges, a novel machine learning method, Heterogeneous Neural Metric Learning (HNML), is developed to restore the integrity of case reporting data using both the incomplete reported cases and the underlying disease-related risk factors from heterogeneous data sources. We empirically validate the effectiveness of our developed method on a representative infectious disease, malaria. We test the developed method under three common real-life data missing patterns with different levels of missing rates. By incorporating the disease-related risk factors as external resources through a machine learning method, we demonstrate significant accuracy improvement over baseline and state-of-the-art inference methods for predicting unobserved malaria cases based on the incomplete reporting data. The results suggest that the disease-related risk factors can provide valuable information about the transmission patterns of infectious diseases and should be taken into account when implementing the surveillance.</p>
computer science, artificial intelligence
What problem does this paper attempt to address?