A Causally Constrained Framework Coupling Causal Discovery and SEIR Mechanisms for Interpretable Epidemic Modeling
Infectious disease transmission is a complex dynamic process governed by intrinsic causal mechanisms rather than simple statistical correlations. Although deep learning paradigms have demonstrated powerful nonlinear representation capabilities, their “black-box” and purely data-driven nature often lead to a severe lack of causal consistency and logical transparency. To bridge this gap, this paper proposes CCSANet (Causally Constrained SEIR-Aware Network), an interpretable forecasting framework that seamlessly embeds epidemiological priors directly into the neural architecture. The model integrates SEIR dynamics into a temporal causal discovery framework, utilizing a mechanism-aware prior loss to guide a CausalFormer in learning a global temporal causal graph from multi-source heterogeneous data. This ensures that the identified relationships strictly adhere to the fundamental evolutionary logic of contagion. Subsequently, the extracted causal subgraphs are encoded as structural priors within a Causal-SCI-Block via a specialized masking mechanism, effectively forcing information to propagate exclusively along epidemiologically legitimate pathways. To ensure deep alignment between neural representations and physical reality, a causal strength alignment loss is introduced to synchronize the network’s attention weights with actual transmission intensities. Experimental evaluations on real-world multi-city datasets demonstrate that this integrated approach significantly outperforms baselines such as LSTM, Informer, and its predecessor, ESASNet. Under a 7-day sliding window configuration, the model maintains a Coefficient of Determination R2 stably above 0.97, achieving an accuracy improvement of 5.5% to 6.2% and an 8% to 10% reduction in SMAPE, thereby demonstrating that coupling causal discovery with SEIR constraints substantially enhances both predictive precision and physical interpretability.
Authors
- Zhixiong Fang (ORCID: https://orcid.org/0000-0002-9463-6207)
- Yijiang Zhao (ORCID: https://orcid.org/0000-0002-8356-5474)
- Rui Zhu
- Yizhi Liu
Institutions
- Hunan University of Science and Technology (CN)
- Hunan University (CN)
- Changsha University (CN)
Publication Details
- Journal
- Mathematics
- Published
- 2026-05-21
- DOI
- https://doi.org/10.3390/math14101776
- Primary Topic
- Machine Learning in Healthcare
- Type
- article
- Field-Weighted Citation Impact
- 0.00
Funders
- Natural Science Foundation of Hainan Province