融合流行病机制的连续传播图模型,提升传染病动态预测精度
Epidemiology-Aware Neural ODE with Continuous Disease Transmission Graph
- 将神经微分方程与流行病传播机制结合,建模连续区域传播模式
- 引入全局感染趋势引导局部传播,提升跨区域预测能力
- 适用于公共卫生决策与医疗资源调度,尤其适合突发疫情场景
有效的疫情预测对公共卫生策略和医疗资源高效配置至关重要,尤其在传染病快速传播背景下。现有深度学习方法常忽视疫情的动态特性,未能充分考虑疾病传播的具体机制。为此,本文提出一种端到端框架EARTH(Epidemiology-Aware Neural ODE with Continuous Disease Transmission Graph),通过EANO模块将神经微分方程与流行病学机制无缝结合,捕捉疫情演化中的复杂空间传播过程;同时引入GLTG建模全局感染趋势,并利用其信号动态引导局部传播。为兼顾全局趋势一致性与局部传播细微差异,设计交叉注意力机制融合关键信息。实验表明,EARTH在真实世界疫情预测任务中优于现有先进方法。代码将在https://github.com/Emory-Melody/EpiLearn发布。
原文摘要 · Abstract (English)
Effective epidemic forecasting is critical for public health strategies and efficient medical resource allocation, especially in the face of rapidly spreading infectious diseases. However, existing deep-learning methods often overlook the dynamic nature of epidemics and fail to account for the specific mechanisms of disease transmission. In response to these challenges, we introduce an innovative end-to-end framework called Epidemiology-Aware Neural ODE with Continuous Disease Transmission Graph (EARTH) in this paper. To learn continuous and regional disease transmission patterns, we first propose EANO, which seamlessly integrates the neural ODE approach with the epidemic mechanism, considering the complex spatial spread process during epidemic evolution. Additionally, we introduce GLTG to model global infection trends and leverage these signals to guide local transmission dynamically. To accommodate both the global coherence of epidemic trends and the local nuances of epidemic transmission patterns, we build a cross-attention approach to fuse the most meaningful information for forecasting. Through the smooth synergy of both components, EARTH offers a more robust and flexible approach to understanding and predicting the spread of infectious diseases. Extensive experiments show EARTH superior performance in forecasting real-world epidemics compared to state-of-the-art methods. The code will be available at https://github.com/Emory-Melody/EpiLearn.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。