We gratefully acknowledge support from
the Simons Foundation and member institutions.
Full-text links:

Download:

Current browse context:

cs.LG

Change to browse by:

References & Citations

DBLP - CS Bibliography

Bookmark

(what is this?)
CiteULike logo BibSonomy logo Mendeley logo del.icio.us logo Digg logo Reddit logo ScienceWISE logo

Computer Science > Machine Learning

Title: Influence Estimation and Maximization via Neural Mean-Field Dynamics

Abstract: We propose a novel learning framework using neural mean-field (NMF) dynamics for inference and estimation problems on heterogeneous diffusion networks. Our new framework leverages the Mori-Zwanzig formalism to obtain an exact evolution equation of the individual node infection probabilities, which renders a delay differential equation with memory integral approximated by learnable time convolution operators. Directly using information diffusion cascade data, our framework can simultaneously learn the structure of the diffusion network and the evolution of node infection probabilities. Connections between parameter learning and optimal control are also established, leading to a rigorous and implementable algorithm for training NMF. Moreover, we show that the projected gradient descent method can be employed to solve the challenging influence maximization problem, where the gradient is computed extremely fast by integrating NMF forward in time just once in each iteration. Extensive empirical studies show 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.
Comments: 26 pages, 6 figures. arXiv admin note: text overlap with arXiv:2006.09449
Subjects: Machine Learning (cs.LG); Computational Engineering, Finance, and Science (cs.CE); Social and Information Networks (cs.SI); Numerical Analysis (math.NA); Optimization and Control (math.OC)
MSC classes: 68U35, 65D99, 65Z05
Cite as: arXiv:2106.02608 [cs.LG]
  (or arXiv:2106.02608v1 [cs.LG] for this version)

Submission history

From: Xiaojing Ye [view email]
[v1] Thu, 3 Jun 2021 00:02:05 GMT (717kb,D)

Link back to: arXiv, form interface, contact.