The predictive skill of convolutional neural networks models for disease forecasting

Kookjin Lee, Jaideep Ray, Cosmin Safta

Research output: Contribution to journalArticlepeer-review

Abstract

In this paper we investigate the utility of one-dimensional convolutional neural network (CNN) models in epidemiological forecasting. Deep learning models, in particular variants of recurrent neural networks (RNNs) have been studied for ILI (Influenza-Like Illness) forecasting, and have achieved a higher forecasting skill compared to conventional models such as ARIMA. In this study, we adapt two neural networks that employ one-dimensional temporal convolutional layers as a primary building block—temporal convolutional networks and simple neural attentive meta-learners—for epidemiological forecasting. We then test them with influenza data from the US collected over 2010-2019. We find that epidemiological forecasting with CNNs is feasible, and their forecasting skill is comparable to, and at times, superior to, plain RNNs. Thus CNNs and RNNs bring the power of nonlinear transformations to purely data-driven epidemiological models, a capability that heretofore has been limited to more elaborate mechanistic/compartmental disease models.

Original languageEnglish (US)
Article numbere0254319
JournalPloS one
Volume16
Issue number7 July
DOIs
StatePublished - Jul 2021
Externally publishedYes

ASJC Scopus subject areas

  • General

Fingerprint

Dive into the research topics of 'The predictive skill of convolutional neural networks models for disease forecasting'. Together they form a unique fingerprint.

Cite this