Generative Pretraining for Black-Box Optimization
Satvik Mehul Mashkaria, Siddarth Krishnamoorthy, Aditya Grover
Abstract
Many problems in science and engineering involve optimizing an expensive black-box function over a high-dimensional space. In the offline black-box optimization (BBO) setting, we assume access to a fixed, offline dataset for pretraining and a small budget for online function evaluations. Prior approaches seek to utilize the offline data to approximate the function or its inverse but are not sufficiently accurate far from the data distribution. We propose BONET, a generative framework for pretraining a novel modelbased optimizer using offline datasets. In BONET, we train an autoregressive model on fixed-length trajectories derived from an offline dataset. We design a sampling strategy to synthesize trajectories from offline data using a simple heuristic of rolling out monotonic transitions from lowfidelity to high-fidelity samples. Empirically, we instantiate BONET using a causally masked Transformer (Radford et al., 2019) and evaluate it on Design-Bench (Trabucco et al., 2022) , where we rank the best on average, outperforming stateof-the-art baselines. Fortunately, for many such black-box optimization (BBO)
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext c400ddff-a912-4236-88dd-c7a864fe4306Cited by top-tier papers23
- Diffusion Models for Black-Box OptimizationSiddarth Krishnamoorthy, Satvik Mehul Mashkaria, Aditya GroverICML 2023 · 94 citations
- Masked Autoencoding for Scalable and Generalizable Decision MakingFangchen Liu, Hao Liu, Aditya Grover, Pieter AbbeelNeurIPS 2022 · 63 citations
- ExPT: Synthetic Pretraining for Few-Shot Experimental DesignTung Nguyen, Sudhanshu Agrawal, Aditya GroverNeurIPS 2023 · 27 citations
- Guided Trajectory Generation with Diffusion Models for Offline Model-based OptimizationTaeyoung Yun, Sujin Yun, Jaewoo Lee, Jinkyoo ParkNeurIPS 2024 · 23 citations
- Pretrained Optimization Model for Zero-Shot Black Box OptimizationXiaobin Li, Kai Wu, Yujian Betterest Li, Xiaoyu Zhang et al.NeurIPS 2024 · 23 citations
Builds on14
- Language Models are Few-Shot LearnersTom B. Brown, Benjamin Mann, Nick Ryder, Melanie Subbiah et al.NeurIPS 2020 · 64,255 citations
- Decision Transformer: Reinforcement Learning via Sequence ModelingLili Chen, Kevin Lu, Aravind Rajeswaran, Kimin Lee et al.NeurIPS 2021 · 2,557 citations
- Offline Reinforcement Learning as One Big Sequence Modeling ProblemMichael Janner, Qiyang Li, Sergey LevineNeurIPS 2021 · 950 citations
- Online Decision TransformerQinqing Zheng, Amy Zhang, Aditya GroverICML 2022 · 256 citations
- Convolutional Conditional Neural ProcessesJonathan Gordon, Wessel P. Bruinsma, Andrew Y. K. Foong, James Requeima et al.ICLR 2020 · 200 citations
Related papers
- Design-Bench: Benchmarks for Data-Driven Offline Model-Based OptimizationBrandon Trabucco, Xinyang Geng, Aviral Kumar, Sergey LevineICML 2022 · 126 citations
- Bootstrapped Transformer for Offline Reinforcement LearningKerong Wang, Hanye Zhao, Xufang Luo, Kan Ren et al.NeurIPS 2022 · 54 citations
- Training Diffusion Language Models for Black-Box OptimizationZipeng Sun, Can Chen, Ye Yuan, Haolun Wu et al.ICML 2026
- Offline Model-Based Optimization by Learning to RankRong-Xi Tan, Ke Xue, Shen-Huan Lyu, Haopu Shang et al.ICLR 2025
- Generative Adversarial Model-Based Optimization via Source Critic RegularizationMichael S. Yao, Yimeng Zeng, Hamsa Bastani, Jacob R. Gardner et al.NeurIPS 2024 · 14 citations
