GeoID-PINN: Identifiability-Aware Regional Epidemic Inference with Geographic Coupling
GeoID-PINN is a physics-informed neural network combining SIRD epidemic dynamics with spatial regularization to disentangle local transmission from external seeding and reporting effects in regional surveillance data A row-stochastic source-composition matrix models spatial dependence, regularized toward spatial priors based on distance, adjacency, commuting, or lead-lag information Simulation results show accurate trajectory fitting does not guarantee recovery of the true regional dependence st
Analysis
TL;DR
- GeoID-PINN is a physics-informed neural network combining SIRD epidemic dynamics with spatial regularization to disentangle local transmission from external seeding and reporting effects in regional surveillance data
- A row-stochastic source-composition matrix models spatial dependence, regularized toward spatial priors based on distance, adjacency, commuting, or lead-lag information
- Simulation results show accurate trajectory fitting does not guarantee recovery of the true regional dependence structure, with source-composition error rising from 0.099 (compatible prior) to 0.577 (misspecified prior)
- On 64 Louisiana COVID-19 counties, Forecast-Trained Geo-PINN reduced MSE by 65% (32,957 to 11,468) and MAE by 18% compared to an autoregressive negative-binomial baseline
- County adjacency regularization improved point accuracy but the baseline achieved better distributional fit (lower NLL), highlighting a trade-off between point prediction and probabilistic calibration
Why It Matters
This work addresses a fundamental identifiability challenge in epidemiological modeling: separating local transmission dynamics from external infection pressure and reporting artifacts using only regional surveillance data. For AI practitioners working in public health or spatial modeling, it demonstrates how physics-informed neural networks can encode domain knowledge (SIRD dynamics) while learning spatial coupling structures from data, offering a template for other spatiotemporal inference problems.
Technical Details
- Model architecture: GeoID-PINN integrates a SIRD (Susceptible-Infectious-Recovered-Deceased) compartmental model with a physics-informed neural network, using a row-stochastic source-composition matrix where each row assigns nonnegative weights summing to one across regions
- Spatial regularization: The source-composition matrix is regularized toward spatial priors constructed from distance, adjacency, commuting flows, or lead-lag correlations, enabling structured inference of regional dependence
- Simulation validation: In a four-region simulation with known ground truth, a compatible distance prior achieved source-composition error of 0.099, compared to 0.159 without regularization and 0.577 under strong prior misspecification, while trajectory fit remained similar across conditions
- Real-world evaluation: Applied retrospectively to COVID-19 data from 64 Louisiana counties; Forecast-Trained Geo-PINN reduced MSE from 32,957 to 11,468 and MAE from 70.60 to 57.73 versus an autoregressive negative-binomial baseline, though the baseline achieved lower NLL (5.158 vs 5.346)
- Prior sensitivity: A controlled 15-county comparison showed county adjacency reduced MSE by 6.85% and MAE by 3.1%; similar performance across plausible priors supports structured regularization but not unique edge recovery, necessitating prior-sensitivity and observation-model checks
Industry Insight
- The decoupling of trajectory accuracy from structural identifiability is a critical warning for epidemiological AI: models can produce excellent point forecasts while learning incorrect spatial dependence, meaning deployment in public health contexts requires rigorous prior-sensitivity analysis and uncertainty quantification beyond standard accuracy metrics
- The trade-off between point prediction accuracy (favored by Geo-PINN) and distributional fit (favored by the negative-binomial baseline) suggests hybrid approaches that combine physics-informed structure with probabilistic observation models could yield the best of both worlds for operational epidemic forecasting
- The framework's reliance on spatial priors (distance, adjacency, commuting) that must be specified a priori limits generalizability; future work should explore learned or adaptive priors that reduce human bias in constructing the spatial regularization structure
Disclaimer: The above content is generated by AI and is for reference only.