中文

RoMA:面向离线基于模型优化的鲁棒模型自适应

机器学习 2021-10-28 v1

摘要

我们考虑在给定输入-输出查询的静态数据集下,搜索使黑盒目标函数最大化的输入的问题。解决该问题的一个流行方法是维护一个代理模型,例如深度神经网络(DNN),来近似真实目标函数。此处的主要挑战是如何避免在搜索过程中对抗性优化的输入,即DNN高度高估真实目标函数的那些输入。为处理该问题,我们提出了一个新框架,称为鲁棒模型自适应(RoMA),基于在DNN上对已优化输入进行梯度优化。具体而言,它包含两步:(a) 鲁棒训练代理模型的预训练策略,以及 (b) 将代理模型自适应以对特定候选解集合具有鲁棒估计的新颖过程。在高层上,我们的方案利用局部平滑先验来克服DNN的脆弱性。在各种任务下的实验显示了RoMA相较于先前方法的有效性,取得了最先进(state-of-the-art)的结果,例如,RoMA在6项任务中的4项上优于所有方法,并在其余任务上获得亚军结果。

关键词

引用

@article{arxiv.2110.14188,
  title  = {RoMA: Robust Model Adaptation for Offline Model-based Optimization},
  author = {Sihyun Yu and Sungsoo Ahn and Le Song and Jinwoo Shin},
  journal= {arXiv preprint arXiv:2110.14188},
  year   = {2021}
}

备注

NeurIPS 2021