论文标题
共同词典学习和在线NMF的COVID-19时间序列预测
COVID-19 Time-series Prediction by Joint Dictionary Learning and Online NMF
论文作者
论文摘要
预测COVID-19的传播和遏制是广泛的科学界目前面临的最重要的挑战。困难的主要来源之一是,每天的Covid-19案例数据都非常有限,除少数例外,大多数国家目前处于“指数分阶段”,因此可用的信息稀缺可以预测一个可以预测传播和遏制之间的相位过渡。 在本文中,我们提出了一种新的方法,可以根据词典学习和在线非负矩阵分解(在线NMF)预测Covid-19的传播。关键的想法是学习在多个国家的新日常案例的简短演变实例的字典模式,以便在词典模式中捕获其潜在相关结构。我们首先通过从整个时间序列学习Minibatch学习此类模式,然后通过在线NMF进一步使它们适应时间序列。随着我们逐步适应并改善学习的字典模式,我们还使用它们来通过部分拟合来进行一步预测。最后,通过递归应用一步预测,我们可以将我们的预测推断到不久的将来。由于其可解释性,我们的预测结果可以直接归因于学习的字典模式。
Predicting the spread and containment of COVID-19 is a challenge of utmost importance that the broader scientific community is currently facing. One of the main sources of difficulty is that a very limited amount of daily COVID-19 case data is available, and with few exceptions, the majority of countries are currently in the "exponential spread stage," and thus there is scarce information available which would enable one to predict the phase transition between spread and containment. In this paper, we propose a novel approach to predicting the spread of COVID-19 based on dictionary learning and online nonnegative matrix factorization (online NMF). The key idea is to learn dictionary patterns of short evolution instances of the new daily cases in multiple countries at the same time, so that their latent correlation structures are captured in the dictionary patterns. We first learn such patterns by minibatch learning from the entire time-series and then further adapt them to the time-series by online NMF. As we progressively adapt and improve the learned dictionary patterns to the more recent observations, we also use them to make one-step predictions by the partial fitting. Lastly, by recursively applying the one-step predictions, we can extrapolate our predictions into the near future. Our prediction results can be directly attributed to the learned dictionary patterns due to their interpretability.