Single Model for Influenza Forecasting of Multiple Countries by Multi-task Learning

by   Taichi Murayama, et al.

The accurate forecasting of infectious epidemic diseases such as influenza is a crucial task undertaken by medical institutions. Although numerous flu forecasting methods and models based mainly on historical flu activity data and online user-generated contents have been proposed in previous studies, no flu forecasting model targeting multiple countries using two types of data exists at present. Our paper leverages multi-task learning to tackle the challenge of building one flu forecasting model targeting multiple countries; each country as each task. Also, to develop the flu prediction model with higher performance, we solved two issues; finding suitable search queries, which are part of the user-generated contents, and how to leverage search queries efficiently in the model creation. For the first issue, we propose the transfer approaches from English to other languages. For the second issue, we propose a novel flu forecasting model that takes advantage of search queries using an attention mechanism and extend the model to a multi-task model for multiple countries' flu forecasts. Experiments on forecasting flu epidemics in five countries demonstrate that our model significantly improved the performance by leveraging the search queries and multi-task learning compared to the baselines.



There are no comments yet.


page 1

page 2

page 3

page 4


Multi-task Learning for Financial Forecasting

Financial forecasting is challenging and attractive in machine learning....

Learning to Transfer for Traffic Forecasting via Multi-task Learning

Deep neural networks have demonstrated superior performance in short-ter...

DiaNet: BERT and Hierarchical Attention Multi-Task Learning of Fine-Grained Dialect

Prediction of language varieties and dialects is an important language p...

Multi-task Prediction of Patient Workload

Developing reliable workload predictive models can affect many aspects o...

Deep Latent Emotion Network for Multi-Task Learning

Feed recommendation models are widely adopted by numerous feed platforms...

A parameter-free population-dynamical approach to health workforce supply forecasting of EU countries

Many countries face challenges like impending retirement waves, negative...
This week in AI

Get the week's most popular data science and artificial intelligence research sent straight to your inbox every Saturday.