arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2006.09449cs.LGcs.SIstat.ML
收藏

基于神经平均场动力学的网络扩散

Network Diffusions via Neural Mean-Field Dynamics

Georgia State University(佐治亚州立大学)

Shushan He, Hongyuan Zha, Xiaojing Ye

摘要:

我们提出了一种新的学习框架,基于神经平均场动力学的推理和估计问题的扩散网络。我们的新框架来自Mori-Zwanzig形式主义,以获得节点感染概率的精确演化,这使得具有记忆积分的延迟微分方程由可学习的时间卷积算子近似,从而产生高度结构化和可解释的RNN。直接使用级联数据,我们的框架可以共同学习扩散网络的结构和感染概率的演变,这是重要的下游应用程序,如影响力最大化的基石。建立了参数学习与最优控制之间的联系。实证研究表明,我们的方法是通用的和强大的变化的基础扩散网络模型,并显着优于现有的方法在合成和真实世界的数据的准确性和效率。

英文摘要:

We propose a novel learning framework based on neural mean-field dynamics for inference and estimation problems of diffusion on networks. Our new framework is derived from the Mori-Zwanzig formalism to obtain an exact evolution of the node infection probabilities, which renders a delay differential equation with memory integral approximated by learnable time convolution operators, resulting in a highly structured and interpretable RNN. Directly using cascade data, our framework can jointly learn the structure of the diffusion network and the evolution of infection probabilities, which are cornerstone to important downstream applications such as influence maximization. Connections between parameter learning and optimal control are also established. Empirical study shows that our approach is versatile and robust to variations of the underlying diffusion network models, and significantly outperform existing approaches in accuracy and efficiency on both synthetic and real-world data.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑