A Hybrid AI-Mathematical approach for epidemic threshold prediction in metapopulation networks: Integrating physics-guided neural networks with spectral graph theory

Etienne Kouokam

PLOS ONE, 2026, vol. 21, issue 6, 1-14

Abstract: Predicting the epidemic threshold τ in contact networks is a central challenge in computational epidemiology. Classical structural approaches based on spectral graph theory—most notably Quenched Mean-Field (QMF) and the recently proposed KSEL (K Spectral Energy of Laplacian) method—deliver fast but approximate predictions. We propose a hybrid AI-mathematical framework that integrates spectral graph features, epidemiological parameters, and epidemiologically motivated soft constraints derived from compartmental theory into a physics-guided neural network (PGNN). Rather than claiming state-of-the-art predictive performance, this work makes three complementary contributions: (i) a rigorous stochastic ground-truth estimation procedure with Monte Carlo uncertainty quantification (median στ*=0.0072, n = 775 networks); (ii) a systematic comparative evaluation of seven methods—including tree-based models (Random Forest, Gradient Boosting) trained on the same feature set—revealing the conditions under which deep learning surpasses and falls short of simpler baselines; and (iii) a full ablation study and SHAP interpretability analysis identifying the role of individual spectral features and physical constraints as structured regularisers. Evaluated on 775 synthetic networks spanning Erdős–Rényi, Barabási–Albert, Watts–Strogatz, and regular topologies, Gradient Boosting achieves the best predictive accuracy (R2 = 0.908, RMSE = 0.0731), while the PGNN (R2 = 0.093) offers complementary value through physical consistency and interpretability. These results are established on synthetic benchmarks; application to empirical contact networks (hospital, school, workplace settings) is a natural next step but requires dedicated validation beyond the scope of the present study. Ablation results show that the boundedness constraint is a beneficial regulariser while the stability constraint over-regularises in the low-data regime. Automatic gradient-based calibration of the KSEL coefficient yields topology-dependent optimal values (k*∈[0.803,1.458]), substantially departing from the universal constant k = 0.3 of prior work.

Date: 2026
References: Add references at CitEc
Citations:

Downloads: (external link)
https://journals.plos.org/plosone/article?id=10.1371/journal.pone.0344827 (text/html)
https://journals.plos.org/plosone/article/file?id= ... 44827&type=printable (application/pdf)

Related works:
This item may be available elsewhere in EconPapers: Search for items with the same title.

Export reference: BibTeX RIS (EndNote, ProCite, RefMan) HTML/Text

Persistent link: https://EconPapers.repec.org/RePEc:plo:pone00:0344827

DOI: 10.1371/journal.pone.0344827

Access Statistics for this article

More articles in PLOS ONE from Public Library of Science
Bibliographic data for series maintained by plosone ().