Fourier Neural Operators with Least-Squares Readout Refit for Learning Random Obstacle-to-Solution Maps
For researchers in operator learning and computational mechanics, this work provides a simple post-training enhancement that improves accuracy for learning random obstacle-to-solution maps, though the improvement is incremental.
The paper tackles learning obstacle-to-solution maps for elliptic variational inequalities with random obstacles. The proposed FNO with least-squares readout refit (FNO-LS) achieves the strongest overall performance, improving average field accuracy, contact-set recovery, and obstacle-violation metrics, especially for higher-amplitude obstacles.
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.