Autofocused oracles for model-based design
Clara Fannjiang, Jennifer Listgarten
Abstract
Data-driven design is making headway into a number of application areas, including protein, small-molecule, and materials engineering. The design goal is to construct an object with desired properties, such as a protein that binds to a therapeutic target, or a superconducting material with a higher critical temperature than previously observed. To that end, costly experimental measurements are being replaced with calls to high-capacity regression models trained on labeled data, which can be leveraged in an in silico search for design candidates. However, the design goal necessitates moving into regions of the design space beyond where such models were trained. Therefore, one can ask: should the regression model be altered as the design algorithm explores the design space, in the absence of new data? Herein, we answer this question in the affirmative. In particular, we (i) formalize the data-driven design problem as a non-zero-sum game, (ii) develop a principled strategy for retraining the regression model as the design algorithm proceedswhat we refer to as autofocusing, and (iii) demonstrate the promise of autofocusing empirically. 1 For many applications in protein, molecule, and material design, even if one performs iterative rounds of data acquisition, at some point, the acquisition phase concludes due to finite resources. 34th Conference on Neural Information Processing Systems (NeurIPS 2020),
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.
Cited by top-tier papers37
- Design-Bench: Benchmarks for Data-Driven Offline Model-Based OptimizationBrandon Trabucco, Xinyang Geng, Aviral Kumar, Sergey LevineICML 2022 · 126 citations
- Conservative Objective Models for Effective Offline Model-Based OptimizationBrandon Trabucco, Aviral Kumar, Xinyang Geng, Sergey LevineICML 2021 · 119 citations
- Diffusion Models for Black-Box OptimizationSiddarth Krishnamoorthy, Satvik Mehul Mashkaria, Aditya GroverICML 2023 · 94 citations
- Offline Model-Based Optimization via Normalized Maximum Likelihood EstimationJustin Fu, Sergey LevineICLR 2021 · 59 citations
- Bidirectional Learning for Offline Infinite-width Model-based OptimizationCan Chen, Yingxue Zhang, Jie Fu, Xue (Steve) Liu et al.NeurIPS 2022 · 56 citations
Builds on4
- Performative PredictionJuan C. Perdomo, Tijana Zrnic, Celestine Mendler-Dünner, Moritz HardtICML 2020 · 422 citations
- Deep Imitative Models for Flexible Inference, Planning, and ControlNicholas Rhinehart, Rowan McAllister, Sergey LevineICLR 2020 · 159 citations
- Model-based reinforcement learning for biological sequence designChristof Angermüller, David Dohan, David Belanger, Ramya Deshpande et al.ICLR 2020 · 159 citations
- Model Inversion Networks for Model-Based OptimizationAviral Kumar, Sergey LevineNeurIPS 2020 · 129 citations
Related papers
- Implicitly Guided Design with PropEn: Match your Data to Follow the GradientNatasa Tagasovska, Vladimir Gligorijevic, Kyunghyun Cho, Andreas LoukasNeurIPS 2024 · 10 citations
- Improving Molecular Design by Stochastic Iterative Target AugmentationKevin Yang, Wengong Jin, Kyle Swanson, Regina Barzilay et al.ICML 2020 · 31 citations
- Generative property enhancer: implicit guided generation through conditional density estimationPedro O. Pinheiro, Pan Kessel, Aya Abdelsalam Ismail, Sai Pooja Mahajan et al.NeurIPS 2025
- Reliable Algorithm Selection for Machine Learning-Guided DesignClara Fannjiang, Ji Won ParkICML 2025
- Bridging Model-Based Optimization and Generative Modeling via Conservative Fine-Tuning of Diffusion ModelsMasatoshi Uehara, Yulai Zhao, Ehsan Hajiramezanali, Gabriele Scalia et al.NeurIPS 2024 · 31 citations
