Attention-based recurrent neural network for influenza epidemic prediction

BMC Bioinformatics. 2019 Nov 25;20(Suppl 18):575. doi: 10.1186/s12859-019-3131-8.

Abstract

Background: Influenza is an infectious respiratory disease that can cause serious public health hazard. Due to its huge threat to the society, precise real-time forecasting of influenza outbreaks is of great value to our public.

Results: In this paper, we propose a new deep neural network structure that forecasts a real-time influenza-like illness rate (ILI%) in Guangzhou, China. Long short-term memory (LSTM) neural networks is applied to precisely forecast accurateness due to the long-term attribute and diversity of influenza epidemic data. We devise a multi-channel LSTM neural network that can draw multiple information from different types of inputs. We also add attention mechanism to improve forecasting accuracy. By using this structure, we are able to deal with relationships between multiple inputs more appropriately. Our model fully consider the information in the data set, targetedly solving practical problems of the Guangzhou influenza epidemic forecasting.

Conclusion: We assess the performance of our model by comparing it with different neural network structures and other state-of-the-art methods. The experimental results indicate that our model has strong competitiveness and can provide effective real-time influenza epidemic forecasting.

Keywords: Attention mechanism; Influenza epidemic prediction; Multi-channel LSTM neural network.

Publication types

  • Evaluation Study

MeSH terms

  • China / epidemiology
  • Disease Outbreaks
  • Forecasting / methods*
  • Humans
  • Influenza, Human / epidemiology*
  • Neural Networks, Computer*
  • Public Health / statistics & numerical data