Excluding the Target Domain Improves Extrapolation: Deconfounded Hierarchical Physics Constraints
Not on the map. Why?
Abstract (arXiv)
Extrapolation to out-of-distribution conditions is a fundamental challenge for physics-constrained deep generative models. Existing methods apply physical constraints as a single static regularization term uniformly across the generation process, and address neither the hierarchical structure of physical laws and the confounding variable problem. We propose the Deconfounded Hierarchical Gate (DHG), which serves as a diagnostic and control mechanism: it identifies when and how strongly temperature confounding contaminates each constraint level, so that hierarchical gates reflect intrinsic physical inconsistency rather than spurious temperature effects. DHG combines counterfactual estimation via the do-operator with backdoor adjustment to remove confounding, then applies Coarse-to-Fine physical constraints progressively. We report a counter-intuitive finding in pretraining: excluding the target-domain data from pretraining outperforms including it by 39% in extrapolation performance (RMSE 0.224 vs. 0.324). This occurs because FNO learns domain-agnostic physical patterns that transfer more effectively when the target domain is withheld. On a lithium-ion battery temperature extrapolation benchmark (trained at 24 degrees Celsius, evaluated at 4.0--43.0 degrees Celsius), our method achieves RMSE = 0.215, a 46% improvement over the unconstrained baseline (Pure CFM: 0.397).
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 15 of its 34 references that are in the map. First posted on arXiv 8 May 2026.
Builds on
15 of 34 references.
References · 15
- 2020A Simple Framework for Contrastive Learning of Visual Representationscited by 11,979
- 2022Flow Matching for Generative Modelingcited by 5,426
- 2016Overcoming catastrophic forgetting in neural networkscited by 4,718
- 2020Fourier Neural Operator for Parametric Partial Differential Equationscited by 2,806
- 2019DeepONet: Learning nonlinear operators for identifying differential equations based on the universal approximation theorem of operatorscited by 1,965
- 2020Neural Operator: Graph Kernel Network for Partial Differential Equationscited by 722
- 2015Causal inference using invariant prediction: identification and confidence intervalscited by 533
- 2021Physics-Informed Neural Operator for Learning Partial Differential Equationscited by 516
- 2021U-FNO -- An enhanced Fourier neural operator-based deep-learning model for multiphase flowcited by 292
- 2024Physics-Informed Diffusion Modelscited by 88
- 2021Physics-Integrated Variational Autoencoders for Robust and Interpretable Generative Modelingcited by 59
- 2021Machine learning pipeline for battery state of health estimationcited by 28
Built on by
Not cited yet.