English

A Study on Graph-Structured Recurrent Neural Networks and Sparsification with Application to Epidemic Forecasting

Machine Learning 2019-02-15 v1 Optimization and Control Machine Learning

Abstract

We study epidemic forecasting on real-world health data by a graph-structured recurrent neural network (GSRNN). We achieve state-of-the-art forecasting accuracy on the benchmark CDC dataset. To improve model efficiency, we sparsify the network weights via transformed-1\ell_1 penalty and maintain prediction accuracy at the same level with 70% of the network weights being zero.

Keywords

Cite

@article{arxiv.1902.05113,
  title  = {A Study on Graph-Structured Recurrent Neural Networks and Sparsification with Application to Epidemic Forecasting},
  author = {Zhijian Li and Xiyang Luo and Bao Wang and Andrea L. Bertozzi and Jack Xin},
  journal= {arXiv preprint arXiv:1902.05113},
  year   = {2019}
}
R2 v1 2026-06-23T07:40:23.042Z