用最小二乘重校准提升神经算子对随机障碍物的求解精度。
Fourier Neural Operators with Least-Squares Readout Refit for Learning Random Obstacle-to-Solution Maps

- 训练后对FNO的输出层做最小二乘重拟合,保持非线性特征不变。
- 在高振幅障碍物下,接触区域恢复和误差指标显著提升。
- 适合需要高精度求解且模型已部分收敛的随机偏微分方程场景。
我们研究了来自具有有限带宽自仿射随机障碍场的椭圆变分不等式的随机障碍物到解映射的算子学习问题。不引入显式的随机输入截断参数化,而是直接从固定网格上的障碍物样本中学习该映射。由于解不仅受障碍场影响,还受诱导接触集和自由边界几何结构制约,因此该问题极具挑战性。本文提出一种后训练最小二乘读出重校准方法(FNO-LS),在端到端训练完FNO后,冻结其非线性主干,通过求解所有训练样本与网格点上的线性最小二乘问题重新计算最终仿射读出层。该方法在保持学习特征不变的前提下,获得经验平方误差最优的读出权重。我们在两个不同振幅水平的障碍物集合上对比了原始DeepONet、POD-DeepONet、两阶段DeepONet基线、FNO及FNO-LS。数值结果表明,FNO-LS在所有测试模型中表现最佳,尤其在高振幅障碍物导致复杂接触几何时优势明显。该方法以极低额外成本显著提升了平均场精度、接触集恢复率和障碍物违反度指标,尤其适用于非线性主干已具备信息但尚未完全收敛的情况。这些结果表明,最小二乘读出重校准是一种简单而有效的后训练增强策略。
原文摘要 · Abstract (English)
We study operator learning for random obstacle-to-solution maps arising from elliptic variational inequalities with finite-band self-affine random obstacle fields. Instead of introducing an explicit truncated stochastic parametrization of the random input, we learn the map directly from sampled obstacle realizations on a fixed grid. This problem is challenging because the solution is governed not only by the obstacle field itself, but also by the induced contact set and free-boundary geometry. We introduce a post-training least-squares readout refit for the Fourier neural operator (FNO). After the FNO is trained end to end, its nonlinear backbone is frozen and the final affine readout is recomputed by solving the induced linear least-squares problem over all training samples and grid points. The refit yields the empirical squared-error optimal readout for the learned frozen features while leaving the nonlinear representation unchanged. We compare vanilla DeepONet, POD-DeepONet, a two-stage DeepONet baseline, FNO, and FNO with least-squares readout refit (FNO-LS) on two obstacle ensembles with different amplitude levels. Numerical results show that FNO-LS achieves the strongest overall performance among the tested models, particularly for higher-amplitude obstacles with more complex contact geometry. The method improves average field accuracy, contact-set recovery, and obstacle-violation metrics at low additional cost, especially when the FNO backbone is informative but not fully converged. These results suggest that least-squares readout refit is a simple and effective post-training enhancement for learning random obstacle-to-solution maps.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。