用高效算法优化高维超参数,提升求解微分方程与反问题的精度和稳定性。
Bilevel optimization for learning hyperparameters: Application to solving PDEs and inverse problems with Gaussian processes
- 通过高斯-牛顿线性化简化内层优化,实现闭式更新。
- 每次外层迭代仅需一次线性化PDE求解,计算成本大幅降低。
- 适用于高维超参数场景,尤其适合神经网络参数化核函数。
基于核方法与神经网络的偏微分方程(PDE)求解、反问题及监督学习任务,其性能高度依赖超参数选择。本文提出一种基于双层优化框架的高效超参数优化策略,通过高斯-牛顿线性化内层优化步骤,获得闭式更新公式,避免重复求解高成本的PDE。每次外层迭代仅需一次线性化PDE求解,随后进行显式梯度更新。在高斯过程模型中,针对非线性PDE与PDE反问题的实验表明,该方法显著提升精度与鲁棒性,优于传统随机初始化。添加核函数与神经网络参数化深度核的实验验证了方法在高维超参数优化中的可扩展性与有效性。
原文摘要 · Abstract (English)
Methods for solving scientific computing and inference problems, such as kernel- and neural network-based approaches for partial differential equations (PDEs), inverse problems, and supervised learning tasks, depend crucially on the choice of hyperparameters. Specifically, the efficacy of such methods, and in particular their accuracy, stability, and generalization properties, strongly depends on the choice of hyperparameters. While bilevel optimization offers a principled framework for hyperparameter tuning, its nested optimization structure can be computationally demanding, especially in PDE-constrained contexts. In this paper, we propose an efficient strategy for hyperparameter optimization within the bilevel framework by employing a Gauss-Newton linearization of the inner optimization step. Our approach provides closed-form updates, eliminating the need for repeated costly PDE solves. As a result, each iteration of the outer loop reduces to a single linearized PDE solve, followed by explicit gradient-based hyperparameter updates. We demonstrate the effectiveness of the proposed method through Gaussian process models applied to nonlinear PDEs and to PDE inverse problems. Extensive numerical experiments highlight substantial improvements in accuracy and robustness compared to conventional random hyperparameter initialization. In particular, experiments with additive kernels and neural network-parameterized deep kernels demonstrate the method's scalability and effectiveness for high-dimensional hyperparameter optimization.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。