Search papers, labs, and topics across Lattice.
This paper addresses the challenge of distribution shifts in machine learning surrogates for engineering simulations by proposing a Test-Time Adaptation (TTA) framework that stores maximally informative (D-optimal) statistics. The method aims to stabilize adaptation and enable principled parameter selection for high-dimensional, unstructured regression problems common in simulation. Results demonstrate up to 7% out-of-distribution improvements on the SIMSHIFT and EngiBench benchmarks at negligible computational cost, showcasing the effectiveness of TTA for high-dimensional simulation regression and generative design optimization.
Achieve up to 7% performance gains in out-of-distribution simulation tasks by adapting high-dimensional surrogates at test time using D-optimal statistics.
Machine learning surrogates are increasingly used in engineering to accelerate costly simulations, yet distribution shifts between training and deployment often cause severe performance degradation (e.g., unseen geometries or configurations). Test-Time Adaptation (TTA) can mitigate such shifts, but existing methods are largely developed for lower-dimensional classification with structured outputs and visually aligned input-output relationships, making them unstable for the high-dimensional, unstructured and regression problems common in simulation. We address this challenge by proposing a TTA framework based on storing maximally informative (D-optimal) statistics, which jointly enables stable adaptation and principled parameter selection at test time. When applied to pretrained simulation surrogates, our method yields up to 7% out-of-distribution improvements at negligible computational cost. To the best of our knowledge, this is the first systematic demonstration of effective TTA for high-dimensional simulation regression and generative design optimization, validated on the SIMSHIFT and EngiBench benchmarks.
