arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2602.15820cs.LG
收藏

通过D-最优统计稳定高维模拟代理的测试时适应

Stabilizing Test-Time Adaptation of High-Dimensional Simulation Surrogates via D-Optimal Statistics

ELLIS Unit, LIT AI Lab, Institute for Machine Learning, JKU Linz, Austria(ELLIS单位、LIT AI实验室、机器学习研究所、JKU林茨,奥地利);Emmi AI GmbH, Linz, Austria(Emmi AI公司、林茨,奥地利)

Anna Zimmel, Paul Setinek, Gianluca Galletti, Johannes Brandstetter, Werner Zellinger

AI总结:

本文提出基于D-最优统计的TTA框架,用于稳定高维模拟代理的测试时适应,实现分布外性能提升。

摘要:

机器学习代理越来越多地用于工程中以加速昂贵的模拟,但训练和部署之间的分布偏移往往会导致严重性能下降(例如,未见的几何形状或配置)。测试时适应(TTA)可以缓解此类偏移,但现有方法大多针对低维分类问题和结构化输出,以及视觉对齐的输入输出关系,使其在模拟中常见的高维、非结构化和回归问题中不稳定。我们通过提出基于存储最信息量(D-最优)统计的TTA框架来解决这一挑战,该框架联合实现了测试时的稳定适应和原理性的参数选择。当应用于预训练的模拟代理时,我们的方法在分布外情况下可获得高达7%的改进,计算成本 negligible。据我们所知,这是首次系统地展示了有效TTA在高维模拟回归和生成设计优化中的应用,已在SIMSHIFT和EngiBench基准上验证。

英文摘要:

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.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑