STAN: spatio-temporal attention network for pandemic prediction using real-world evidence
Genre
Journal articleDate
2021-01-22Author
Gao, JunyiSharma, Rakshith
Qian, Cheng
Glass, Lucas M.
Spaeder, Jeffrey
Romberg, Justin
Sun, Jimeng
Xiao, Cao
Department
Business and ManagementPermanent link to this record
http://hdl.handle.net/20.500.12613/6169
Metadata
Show full item recordDOI
https://doi.org/10.1093/jamia/ocaa322Abstract
OBJECTIVE: We aim to develop a hybrid model for earlier and more accurate predictions for the number of infected cases in pandemics by (1) using patients' claims data from different counties and states that capture local disease status and medical resource utilization; (2) utilizing demographic similarity and geographical proximity between locations; and (3) integrating pandemic transmission dynamics into a deep learning model. MATERIALS AND METHODS: We proposed a spatio-temporal attention network (STAN) for pandemic prediction. It uses a graph attention network to capture spatio-temporal trends of disease dynamics and to predict the number of cases for a fixed number of days into the future. We also designed a dynamics-based loss term for enhancing long-term predictions. STAN was tested using both real-world patient claims data and COVID-19 statistics over time across US counties. RESULTS: STAN outperforms traditional epidemiological models such as susceptible-infectious-recovered (SIR), susceptible-exposed-infectious-recovered (SEIR), and deep learning models on both long-term and short-term predictions, achieving up to 87% reduction in mean squared error compared to the best baseline prediction model. CONCLUSIONS: By combining information from real-world claims data and disease case counts data, STAN can better predict disease status and medical resource utilization.Citation
Junyi Gao, Rakshith Sharma, Cheng Qian, Lucas M Glass, Jeffrey Spaeder, Justin Romberg, Jimeng Sun, Cao Xiao, STAN: spatio-temporal attention network for pandemic prediction using real-world evidence, Journal of the American Medical Informatics Association, 2021;, ocaa322, https://doi.org/10.1093/jamia/ocaa322Citation to related work
Oxford University PressHas part
Journal of the American Medical Informatics AssociationADA compliance
For Americans with Disabilities Act (ADA) accommodation, including help with reading this content, please contact scholarshare@temple.eduae974a485f413a2113503eed53cd6c53
http://dx.doi.org/10.34944/dspace/6151