arXiv:2412.02161cs.SIcs.DC2024-12

用联邦学习预测疫情传播,保护隐私还能提升预测精度。

Towards the efficacy of federated prediction for epidemics on networks

  • 基于联邦学习构建跨网络的疫情节点级预测框架。
  • STGAT模型在动态传播中捕捉时空依赖能力更强。
  • 适合关注隐私保护与公共卫生建模的研究者参考。

疫情预测对公共健康具有实际意义,可支持早期干预、资源调配与战略规划。然而,机构间健康数据共享常受隐私问题制约,限制了精准预测模型的发展。本文提出一种通用的隐私保护框架,用于基于联邦学习(FL)的网络节点级疫情预测。将多数据隔离子网络中的疫情时空传播建模为节点状态,代表社区内疫情严重程度的聚合值。设计纯时间序列的LSTM模型与时空图注意力网络(STGAT)以应对联邦疫情预测任务。在真实航空网络上对多种疫情过程进行大量实验,全面评估不同场景下联邦学习的效能。通过引入‘效能能量’指标衡量系统在不同客户端配置下的鲁棒性,系统探究影响性能的关键因素,包括客户端数量、聚合策略、图划分方式及感染报告缺失情况。数值结果表明:STGAT在动态过程中更优地捕捉时空依赖关系;而LSTM在简单模式下表现良好。研究强调客户端间特征一致性与数据量均匀性的平衡至关重要,并揭示信息丰富性与动态过程内在随机性之间的预测权衡。本研究为疫情管理中联邦学习的应用提供实践洞见,展现其在更广泛集体动态建模中的潜力。

原文摘要 · Abstract (English)

Epidemic prediction is of practical significance in public health, enabling early intervention, resource allocation, and strategic planning. However, privacy concerns often hinder the sharing of health data among institutions, limiting the development of accurate prediction models. In this paper, we develop a general privacy-preserving framework for node-level epidemic prediction on networks based on federated learning (FL). We frame the spatio-temporal spread of epidemics across multiple data-isolated subnetworks, where each node state represents the aggregate epidemic severity within a community. Then, both the pure temporal LSTM model and the spatio-temporal model i.e., Spatio-Temporal Graph Attention Network (STGAT) are proposed to address the federated epidemic prediction. Extensive experiments are conducted on various epidemic processes using a practical airline network, offering a comprehensive assessment of FL efficacy under diverse scenarios. By introducing the efficacy energy metric to measure system robustness under various client configurations, we systematically explore key factors influencing FL performance, including client numbers, aggregation strategies, graph partitioning, missing infectious reports. Numerical results manifest that STGAT excels in capturing spatio-temporal dependencies in dynamic processes whereas LSTM performs well in simpler pattern. Moreover, our findings highlight the importance of balancing feature consistency and volume uniformity among clients, as well as the prediction dilemma between information richness and intrinsic stochasticity of dynamic processes. This study offers practical insights into the efficacy of FL scenario in epidemic management, demonstrates the potential of FL to address broader collective dynamics.

联邦学习疫情预测图神经网络

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。