Offline model-based optimization (MBO) aims to find designs that maximize a black-box function with only offline data. Under this setting, based on different practical application scenarios, many variant settings have been raised. For example, dimensional constrained MBO discusses situations where the achievable design space is constrained by environmental factors, thus requiring certain dimensions to maintain constant values during the optimization process. We find that it is the diverse contextual data situations that lead to different state-of-the-art methods being applicable to these settings respectively. Even within the optimization process of one setting, contextual data situations can also vary significantly, resulting in inconsistent performance across different datasets. To address this challenge, in this paper we define the problem of offline multi-context MBO, which takes different contextual data situations into consideration, and propose an innovative framework named Adaptive Retrieval-based gradient Planning for Optimization (ARPO), which leverages different retrieval strategies, including naive retrieval, gradient-direction retrieval and average-direction retrieval, to generate different gradient directions for various data contexts. ARPO is able to plan the gradients adaptively according to the evaluations of the retrieval model, since the model, as we will show, exhibits high predictive accuracy and implicitly introduces fine-grained conservatism against out-of-distribution designs. Experiments are conducted on the Hartmann test function and the Design-Bench benchmark, where ARPO outperforms the state-of-the-art methods in multi-context MBO tasks.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

Adaptive Retrieval-Based Gradient Planning for Offline Multi-context Model-Based Optimization

  • Haoran Zhao,
  • Mingcheng Chen,
  • Weinan Zhang,
  • Yong Yu

摘要

Offline model-based optimization (MBO) aims to find designs that maximize a black-box function with only offline data. Under this setting, based on different practical application scenarios, many variant settings have been raised. For example, dimensional constrained MBO discusses situations where the achievable design space is constrained by environmental factors, thus requiring certain dimensions to maintain constant values during the optimization process. We find that it is the diverse contextual data situations that lead to different state-of-the-art methods being applicable to these settings respectively. Even within the optimization process of one setting, contextual data situations can also vary significantly, resulting in inconsistent performance across different datasets. To address this challenge, in this paper we define the problem of offline multi-context MBO, which takes different contextual data situations into consideration, and propose an innovative framework named Adaptive Retrieval-based gradient Planning for Optimization (ARPO), which leverages different retrieval strategies, including naive retrieval, gradient-direction retrieval and average-direction retrieval, to generate different gradient directions for various data contexts. ARPO is able to plan the gradients adaptively according to the evaluations of the retrieval model, since the model, as we will show, exhibits high predictive accuracy and implicitly introduces fine-grained conservatism against out-of-distribution designs. Experiments are conducted on the Hartmann test function and the Design-Bench benchmark, where ARPO outperforms the state-of-the-art methods in multi-context MBO tasks.