LLM2D
基于流行病学的连续疾病传播图神经常微分方程
Epidemiology-Aware Neural ODE with Continuous Disease Transmission Graph
作者: Guancheng Wan, Zewen Liu, Max S. Y. Lau, B. Aditya Prakash, Wei Jin
发布日期: 10/2/2024
arXiv ID: oai:arXiv.org:2410.00049v1

摘要

有效的流行病预测对于公共卫生策略和医疗资源的有效分配至关重要,尤其是在快速传播的传染病面前。然而,现有的深度学习方法往往忽略了流行病的动态特性,并且没有考虑到疾病传播的具体机制。为了应对这些挑战,本文提出了一种名为“具有连续疾病传播图的流行病感知神经常微分方程 (EARTH)” 的端到端创新框架。为了学习连续和区域性的疾病传播模式,我们首先提出了 EANO,它将神经常微分方程方法与流行病机制无缝整合,考虑了流行病演变过程中的复杂空间传播过程。此外,我们引入了 GLTG 来模拟全球感染趋势,并利用这些信号来动态指导局部传播。为了同时适应流行病趋势的全球一致性和流行病传播模式的局部细微差别,我们构建了一种交叉注意机制,以融合最具意义的信息进行预测。通过这两个组件的平滑协同,EARTH 为理解和预测传染病的传播提供了更加稳健和灵活的方法。大量实验表明,与最先进的方法相比,EARTH 在预测现实世界中的流行病方面表现出优异的性能。代码将发布在 https://github.com/Emory-Melody/EpiLearn。