English

Physics-Informed Uncertainty Enables Reliable AI-driven Design

Machine Learning 2026-01-27 v1 Computational Physics

Abstract

Inverse design is a central goal in much of science and engineering, including frequency-selective surfaces (FSS) that are critical to microelectronics for telecommunications and optical metamaterials. Traditional surrogate-assisted optimization methods using deep learning can accelerate the design process but do not usually incorporate uncertainty quantification, leading to poorer optimization performance due to erroneous predictions in data-sparse regions. Here, we introduce and validate a fundamentally different paradigm of Physics-Informed Uncertainty, where the degree to which a model's prediction violates fundamental physical laws serves as a computationally-cheap and effective proxy for predictive uncertainty. By integrating physics-informed uncertainty into a multi-fidelity uncertainty-aware optimization workflow to design complex frequency-selective surfaces within the 20 - 30 GHz range, we increase the success rate of finding performant solutions from less than 10% to over 50%, while simultaneously reducing computational cost by an order of magnitude compared to the sole use of a high-fidelity solver. These results highlight the necessity of incorporating uncertainty quantification in machine-learning-driven inverse design for high-dimensional problems, and establish physics-informed uncertainty as a viable alternative to quantifying uncertainty in surrogate models for physical systems, thereby setting the stage for autonomous scientific discovery systems that can efficiently and robustly explore and evaluate candidate designs.

Keywords

Cite

@article{arxiv.2601.18638,
  title  = {Physics-Informed Uncertainty Enables Reliable AI-driven Design},
  author = {Tingkai Xue and Chin Chun Ooi and Yang Jiang and Luu Trung Pham Duong and Pao-Hsiung Chiu and Weijiang Zhao and Nagarajan Raghavan and My Ha Dao},
  journal= {arXiv preprint arXiv:2601.18638},
  year   = {2026}
}
R2 v1 2026-07-01T09:20:40.949Z