Surrogate models for diffusion on graphs via sparse polynomials
Giuseppe Alessio D'Inverno, Kylian Ajavon, Simone Brugiapaglia
TL;DR
This work introduces sparse polynomial surrogate models for parametric diffusion on graphs with community structure and proves holomorphic regularity of the parametric diffusion map, enabling recovery guarantees for least squares and compressed sensing. By establishing that the parameter-to-solution map is holomorphic, the authors derive quasi-optimal best $s$-term convergence rates and provide probabilistic guarantees for both LS and CS recoveries, with explicit bounds that account for dimensionality and polylog factors. The methodology is validated on synthetic stochastic block model graphs and real networks (Twitter and Facebook), demonstrating robustness across dimension, basis size, and graph scale. The results offer a scalable framework for uncertainty quantification and fast surrogate modeling of diffusion processes on graphs, with potential extensions to time-dependent diffusion and connections to graph neural network architectures.
Abstract
Diffusion kernels over graphs have been widely utilized as effective tools in various applications due to their ability to accurately model the flow of information through nodes and edges. However, there is a notable gap in the literature regarding the development of surrogate models for diffusion processes on graphs. In this work, we fill this gap by proposing sparse polynomial-based surrogate models for parametric diffusion equations on graphs with community structure. In tandem, we provide convergence guarantees for both least squares and compressed sensing-based approximations by showing the holomorphic regularity of parametric solutions to these diffusion equations. Our theoretical findings are accompanied by a series of numerical experiments conducted on both synthetic and real-world graphs that demonstrate the applicability of our methodology.
