IDEAS home Printed from https://ideas.repec.org/a/plo/pone00/0344827.html

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

Author

Listed:
  • Etienne Kouokam

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.

Suggested Citation

  • Etienne Kouokam, 2026. "A Hybrid AI-Mathematical approach for epidemic threshold prediction in metapopulation networks: Integrating physics-guided neural networks with spectral graph theory," PLOS ONE, Public Library of Science, vol. 21(6), pages 1-14, June.
  • Handle: RePEc:plo:pone00:0344827
    DOI: 10.1371/journal.pone.0344827
    as

    Download full text from publisher

    File URL: https://journals.plos.org/plosone/article?id=10.1371/journal.pone.0344827
    Download Restriction: no

    File URL: https://journals.plos.org/plosone/article/file?id=10.1371/journal.pone.0344827&type=printable
    Download Restriction: no

    File URL: https://libkey.io/10.1371/journal.pone.0344827?utm_source=ideas
    LibKey link: if access is restricted and if your library uses this service, LibKey will redirect you to where you can use your library subscription to access this item
    ---><---

    More about this item

    Statistics

    Access and download statistics

    Corrections

    All material on this site has been provided by the respective publishers and authors. You can help correct errors and omissions. When requesting a correction, please mention this item's handle: RePEc:plo:pone00:0344827. See general information about how to correct material in RePEc.

    If you have authored this item and are not yet registered with RePEc, we encourage you to do it here. This allows to link your profile to this item. It also allows you to accept potential citations to this item that we are uncertain about.

    We have no bibliographic references for this item. You can help adding them by using this form .

    If you know of missing items citing this one, you can help us creating those links by adding the relevant references in the same way as above, for each refering item. If you are a registered author of this item, you may also want to check the "citations" tab in your RePEc Author Service profile, as there may be some citations waiting for confirmation.

    For technical questions regarding this item, or to correct its authors, title, abstract, bibliographic or download information, contact: plosone (email available below). General contact details of provider: https://journals.plos.org/plosone/ .

    Please note that corrections may take a couple of weeks to filter through the various RePEc services.

    IDEAS is a RePEc service. RePEc uses bibliographic data supplied by the respective publishers.