Exploring accuracy and uncertainty quantification in physics-informed neural networks for inferring microbial community dynamics

2025/09/02 by Pedro Fontanarrosa, C. Barnes, Fontanarrosa, Pedro +1
Biochemistry, Genetics and Molecular Biology · Computer Science · Physics and Astronomy · #FOS: Biological sciences #Gaussian Processes and Bayesian Inference #Gene Regulatory Network Analysis #Model Reduction and Neural Networks #Quantitative Methods (q-bio.QM)

paper · pdf · doi:10.48550/arxiv.2509.02282

openalex publication_date 2025/09/02 · openalex created_date 2025/10/10 · openalex updated_date 2026/07/28

Abstract

Physics-Informed Neural Networks (PINNs) have become a popular way to infer interpretable interaction parameters from noisy microbial time series, but practitioners face many tunable design choices (loss weights, regularisers, scaling, training schedules) with little guidance, and uncertainty is rarely quantified. We present a two-part study using a DeepXDE PINN for a six-species generalised Lotka-Volterra (gLV) model with a known step input and fixed, species-specific step amplitudes, under synthetic noise. Part A: Accuracy ablation. Starting from a broad-init baseline, we systematically vary constrained initialisation, parameter scaling, L2 regularisation with loss-weighting, split training with auxiliary observations, function constraints, adaptive collocation, hyperparameter tuning, and optimisers (Adam,L-BFGS). The largest gains come from simple parameter scaling. Hyperparameter tuning and adaptive collocation achieve competitive pairwise interaction errors. Part B: Uncertainty quantification. We benchmark deep ensembles and Monte Carlo (MC) dropout and report multi-replicate fits with different initial conditions. An N=10 deep ensemble yields the best single-trajectory growth-rate accuracy, while MC-dropout provides predictive bands. Overall, scaling plus either tuning or ensembles delivers robust interaction inference; our loss-weight sweep and additional regularisation settings offer practical defaults. We release code, metrics, and figures to serve as baselines for accuracy and uncertainty quantification in microbiome PINNs.

Citations

• Uncertainty quantification for noisy inputs-outputs in physics-informed neural networks and neural operators
• Systems Biology: Identifiability analysis and parameter identification via systems-biology informed neural networks
• Universal Differential Equations for Scientific Machine Learning
• DeepXDE: A Deep Learning Library for Solving Differential Equations
• Can You Trust Your Model's Uncertainty? Evaluating Predictive Uncertainty Under Dataset Shift
• Physics-informed neural networks: A deep learning framework for solving forward and inverse problems involving nonlinear partial differential equations
• Neural Ordinary Differential Equations
• Sparsity information and regularization in the horseshoe and other shrinkage priors
• On Calibration of Modern Neural Networks
• Simple and Scalable Predictive Uncertainty Estimation using Deep Ensembles
• Data-driven discovery of partial differential equations
• Dropout as a Bayesian Approximation: Representing Model Uncertainty in Deep Learning
• The horseshoe estimator for sparse signals

Related

• Physics-constrained neural ordinary differential equation models to discover and predict microbial community dynamics
• Fast fitting of neural ordinary differential equations by Bayesian neural gradient matching to infer ecological interactions from time series data
• Reconstructing large networks with time-varying interactions
• Predicting Microbial Interactions Using Graph Neural Networks
• Fast fitting of neural ordinary differential equations by Bayesian neural gradient matching to infer ecological interactions from time‐series data