Model Inversion Networks for Model-Based Optimization
Not on the map. Why?
Abstract (arXiv)
In this work, we aim to solve data-driven optimization problems, where the goal is to find an input that maximizes an unknown score function given access to a dataset of inputs with corresponding scores. When the inputs are high-dimensional and valid inputs constitute a small subset of this space (e.g., valid protein sequences or valid natural images), such model-based optimization problems become exceptionally difficult, since the optimizer must avoid out-of-distribution and invalid inputs. We propose to address such problem with model inversion networks (MINs), which learn an inverse mapping from scores to inputs. MINs can scale to high-dimensional input spaces and leverage offline logged data for both contextual and non-contextual optimization problems. MINs can also handle both purely offline data sources and active data collection. We evaluate MINs on tasks from the Bayesian optimization literature, high-dimensional model-based optimization problems over images and protein designs, and contextual bandit optimization from logged data.
Key facts. Cited by 83 papers in the map (data through 14 Oct 2026). First posted on arXiv 31 Dec 2019.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 7; 2022: 11; 2023: 14; 2024: 24; 2025: 14; 2026: 13.
83 papers.
Sort:
By area:
- 2022Biological Sequence Design with GFlowNetscited by 174
- 2022Design-Bench: Benchmarks for Data-Driven Offline Model-Based Optimizationcited by 97
- 2021Conservative Objective Models for Effective Offline Model-Based Optimizationcited by 88
- 2023Diffusion Models for Black-Box Optimizationcited by 76
- 2024An Overview of Diffusion Models: Applications, Guided Generation, Statistical Rates and Optimizationcited by 73
- 2021RoMA: Robust Model Adaptation for Offline Model-based Optimizationcited by 39
- 2022How to Robustify Black-Box ML Models? A Zeroth-Order Optimization Perspectivecited by 39
- 2021Offline Model-Based Optimization via Normalized Maximum Likelihood Estimationcited by 38
- 2022Generative Pretraining for Black-Box Optimizationcited by 36
- 2024Diffusion Models as Constrained Samplers for Optimization with Unknown Constraintscited by 35
- 2022Unsupervised Learning for Combinatorial Optimization with Principled Objective Relaxationcited by 32
- 2022Data-Driven Offline Decision-Making via Invariant Representation Learningcited by 31
38 more
- 2021Data-Driven Offline Optimization For Architecting Hardware Acceleratorscited by 29
- 2024Recent Advances on Machine Learning for Computational Fluid Dynamics: A Surveycited by 28
- 2023Denevil: Towards Deciphering and Navigating the Ethical Values of Large Language Models via Instruction Learningcited by 26
- 2024Diffusion Model for Data-Driven Black-Box Optimizationcited by 24
- 2023Bootstrapped Training of Score-Conditioned Generator for Offline Design of Biological Sequencescited by 23
- 2023ExPT: Synthetic Pretraining for Few-Shot Experimental Designcited by 22
- 2024GTA: Generative Trajectory Augmentation with Guidance for Offline Reinforcement Learningcited by 22
- 2024DiffuSolve: Diffusion-based Solver for Non-convex Trajectory Optimizationcited by 20
- 2024Offline Model-Based Optimization via Policy-Guided Gradient Searchcited by 19
- 2021Deep Extrapolation for Attribute-Enhanced Generationcited by 18
- 2022Plug & Play Directed Evolution of Proteins with Gradient-based Discrete MCMCcited by 17
- 2024Guided Trajectory Generation with Diffusion Models for Offline Model-based Optimizationcited by 15
- 2025Offline Model-Based Optimization: Comprehensive Reviewcited by 15
- 2024SurfPro: Functional Protein Design Based on Continuous Surfacecited by 14
- 2024Offline Multi-Objective Optimizationcited by 14
- 2021Adaptive machine learning for protein engineeringcited by 13
- 2024Improved Off-policy Reinforcement Learning in Biological Sequence Designcited by 13
- 2024Offline Model-Based Optimization by Learning to Rankcited by 13
- 2025Posterior Inference with Diffusion Models for High-dimensional Black-box Optimizationcited by 13
- 2023Importance Weighted Expectation-Maximization for Protein Sequence Designcited by 12
- 2025Preference-Guided Diffusion for Multi-Objective Offline Optimizationcited by 12
- 2024Functional Graphical Models: Structure Enables Offline Data-Driven Optimizationcited by 11
- 2024Design Editing for Offline Model-based Optimizationcited by 10
- 2023Design from Policies: Conservative Test-Time Adaptation for Offline Policy Optimizationcited by 8
- 2024"Noisier" Noise Contrastive Eestimation is (Almost) Maximum Likelihoodcited by 8
- 2024Network Inversion of Convolutional Neural Netscited by 8
- 2024Robust Guided Diffusion for Offline Black-Box Optimizationcited by 8
- 2025Simplifying Bayesian Optimization Via In-Context Direct Optimum Samplingcited by 7
- 2025Towards Universal Offline Black-Box Optimization via Learning Language Model Embeddingscited by 7
- 2022Reliable Conditioning of Behavioral Cloning for Offline Reinforcement Learningcited by 6
- 2025Boosting Offline Optimizers with Surrogate Sensitivitycited by 6
- 2024Generative Enzyme Design Guided by Functionally Important Sites and Small-Molecule Substratescited by 5
- 2025Incorporating Surrogate Gradient Norm to Improve Offline Optimization Techniquescited by 5
- 2025Posterior Inference in Latent Space for Scalable Constrained Black-box Optimizationcited by 5
- 2022Forecasting labels under distribution-shift for machine-guided sequence designcited by 4
- 2023Diffusing Gaussian Mixtures for Generating Categorical Datacited by 4
- 2023Latent Conservative Objective Models for Data-Driven Crystal Structure Predictioncited by 4
- 2025ROOT: Rethinking Offline Optimization as Distributional Translation via Probabilistic Bridgecited by 4
50 most cited of 83 All 83 (JSON)
- 2026ParetoTransport: Generative Optimization by Mass Transport Toward The Pareto Front7 Sep 2026
- 2026Rethinking Learnability in Offline Data-driven Optimization1 Sep 2026
- 2026Spec2Twin-Chain: Orchestrating Bi-Level Optimization with LLMs for Blockchain Digital Twin Construction30 Aug 2026
- 2026PTP: Previous-Token Prediction based LLM Inversion for Near-Exact Prompt Reconstruction31 Jul 2026
- 2026Generative Refinement for Low-Budget Black-Box Optimization1 Jul 2026
- 2026Conformal Candidate Certification for Offline Model-Based Optimization13 Jun 2026
- 2026Improving Bayesian Optimization via Training-Aware Conditional Diffusion Models7 Jun 2026
- 2026Support-Proximity Augmented Diffusion Estimation for Offline Black-Box Optimization11 May 2026
- 2026Black-Box Optimization From Small Offline Datasets via Meta Learning with Synthetic Tasks14 Apr 2026
- 2026Offline Materials Optimization with CliqueFlowmer6 Mar 2026
- 2026The Offline-Frontier Shift: Diagnosing Distributional Limits in Generative Multi-Objective Optimization11 Feb 2026
- 2026Pareto-Conditioned Diffusion Models for Offline Multi-Objective Optimization31 Jan 2026
38 more
- 2026Diffusion Large Language Models for Black-Box Optimization20 Jan 2026
- 2025ROOT: Rethinking Offline Optimization as Distributional Translation via Probabilistic Bridge19 Sep 2025
- 2025A Reward-Directed Diffusion Framework for Generative Design Optimization2 Aug 2025
- 2025Posterior Inference in Latent Space for Scalable Constrained Black-box Optimization1 Jul 2025
- 2025Conservative quantum offline model-based optimization24 Jun 2025
- 2025Towards Universal Offline Black-Box Optimization via Learning Language Model Embeddings8 Jun 2025
- 2025Learning Design-Score Manifold to Guide Diffusion Models for Offline Optimization6 Jun 2025
- 2025Simplifying Bayesian Optimization Via In-Context Direct Optimum Sampling29 May 2025
- 2025Network Inversion for Generating Confidently Classified Counterfeits26 Mar 2025
- 2025Preference-Guided Diffusion for Multi-Objective Offline Optimization21 Mar 2025
- 2025Offline Model-Based Optimization: Comprehensive Review21 Mar 2025
- 2025Incorporating Surrogate Gradient Norm to Improve Offline Optimization Techniques6 Mar 2025
- 2025Boosting Offline Optimizers with Surrogate Sensitivity6 Mar 2025
- 2025Overconfident Oracles: Limitations of In Silico Sequence Design Benchmarking24 Feb 2025
- 2025Posterior Inference with Diffusion Models for High-dimensional Black-box Optimization24 Feb 2025
- 2024Network Inversion and Its Applications26 Nov 2024
- 2024GROOT: Effective Design of Biological Sequences with Limited Experimental Data18 Nov 2024
- 2024Network Inversion for Training-Like Data Reconstruction22 Oct 2024
- 2024Cliqueformer: Model-Based Optimization with Structured Transformers17 Oct 2024
- 2024Sharpness-Aware Black-Box Optimization16 Oct 2024
- 2024Offline Model-Based Optimization by Learning to Rank15 Oct 2024
- 2024Improved Off-policy Reinforcement Learning in Biological Sequence Design6 Oct 2024
- 2024Robust Guided Diffusion for Offline Black-Box Optimization1 Oct 2024
- 2024Recent Advances on Machine Learning for Computational Fluid Dynamics: A Survey22 Aug 2024
- 2024Network Inversion of Convolutional Neural Nets25 Jul 2024
- 2024Guided Trajectory Generation with Diffusion Models for Offline Model-based Optimization29 Jun 2024
- 2024Offline Multi-Objective Optimization6 Jun 2024
- 2024GTA: Generative Trajectory Augmentation with Guidance for Offline Reinforcement Learning27 May 2024
- 2024"Noisier" Noise Contrastive Eestimation is (Almost) Maximum Likelihood27 May 2024
- 2024Design Editing for Offline Model-based Optimization22 May 2024
- 2024Generative Enzyme Design Guided by Functionally Important Sites and Small-Molecule Substrates13 May 2024
- 2024Offline Model-Based Optimization via Policy-Guided Gradient Search8 May 2024
- 2024SurfPro: Functional Protein Design Based on Continuous Surface7 May 2024
- 2024An Overview of Diffusion Models: Applications, Guided Generation, Statistical Rates and Optimization11 Apr 2024
- 2024Diffusion Model for Data-Driven Black-Box Optimization20 Mar 2024
- 2024Diffusion Models as Constrained Samplers for Optimization with Unknown Constraints28 Feb 2024
- 2024DiffuSolve: Diffusion-based Solver for Non-convex Trajectory Optimization22 Feb 2024
- 2024Functional Graphical Models: Structure Enables Offline Data-Driven Optimization8 Jan 2024
50 newest of 83 All 83 (JSON)
- 2023Diffusion Models for Black-Box Optimizationcited by 76
- 2024An Overview of Diffusion Models: Applications, Guided Generation, Statistical Rates and Optimizationcited by 73
- 2024Diffusion Models as Constrained Samplers for Optimization with Unknown Constraintscited by 35
- 2024Diffusion Model for Data-Driven Black-Box Optimizationcited by 24
- 2024DiffuSolve: Diffusion-based Solver for Non-convex Trajectory Optimizationcited by 20
- 2024Guided Trajectory Generation with Diffusion Models for Offline Model-based Optimizationcited by 15
- 2025Posterior Inference with Diffusion Models for High-dimensional Black-box Optimizationcited by 13
- 2025Preference-Guided Diffusion for Multi-Objective Offline Optimizationcited by 12
- 2024Design Editing for Offline Model-based Optimizationcited by 10
- 2024Robust Guided Diffusion for Offline Black-Box Optimizationcited by 8
- 2023Diffusing Gaussian Mixtures for Generating Categorical Datacited by 4
- 2025A Reward-Directed Diffusion Framework for Generative Design Optimizationcited by 2
12 most cited of 17 All 17 (JSON)
- 2026ParetoTransport: Generative Optimization by Mass Transport Toward The Pareto Front7 Sep 2026
- 2026Improving Bayesian Optimization via Training-Aware Conditional Diffusion Models7 Jun 2026
- 2026Pareto-Conditioned Diffusion Models for Offline Multi-Objective Optimization31 Jan 2026
- 2025A Reward-Directed Diffusion Framework for Generative Design Optimization2 Aug 2025
- 2025Learning Design-Score Manifold to Guide Diffusion Models for Offline Optimization6 Jun 2025
- 2025Preference-Guided Diffusion for Multi-Objective Offline Optimization21 Mar 2025
- 2025Posterior Inference with Diffusion Models for High-dimensional Black-box Optimization24 Feb 2025
- 2024Robust Guided Diffusion for Offline Black-Box Optimization1 Oct 2024
- 2024Guided Trajectory Generation with Diffusion Models for Offline Model-based Optimization29 Jun 2024
- 2024Design Editing for Offline Model-based Optimization22 May 2024
- 2024An Overview of Diffusion Models: Applications, Guided Generation, Statistical Rates and Optimization11 Apr 2024
- 2024Diffusion Model for Data-Driven Black-Box Optimization20 Mar 2024
12 newest of 17 All 17 (JSON)
- 2024Improved Off-policy Reinforcement Learning in Biological Sequence Designcited by 13
- 2024Sharpness-Aware Black-Box Optimizationcited by 2
- 2024Sharpness-Aware Black-Box Optimization16 Oct 2024
- 2024Improved Off-policy Reinforcement Learning in Biological Sequence Design6 Oct 2024
- 2022Biological Sequence Design with GFlowNetscited by 174
- 2022Design-Bench: Benchmarks for Data-Driven Offline Model-Based Optimizationcited by 97
- 2021Conservative Objective Models for Effective Offline Model-Based Optimizationcited by 88
- 2021RoMA: Robust Model Adaptation for Offline Model-based Optimizationcited by 39
- 2022How to Robustify Black-Box ML Models? A Zeroth-Order Optimization Perspectivecited by 39
- 2021Offline Model-Based Optimization via Normalized Maximum Likelihood Estimationcited by 38
- 2022Generative Pretraining for Black-Box Optimizationcited by 36
- 2022Unsupervised Learning for Combinatorial Optimization with Principled Objective Relaxationcited by 32
- 2022Data-Driven Offline Decision-Making via Invariant Representation Learningcited by 31
- 2021Data-Driven Offline Optimization For Architecting Hardware Acceleratorscited by 29
- 2024Recent Advances on Machine Learning for Computational Fluid Dynamics: A Surveycited by 28
- 2023Denevil: Towards Deciphering and Navigating the Ethical Values of Large Language Models via Instruction Learningcited by 26
12 most cited of 61 All 61 (JSON)
- 2026Rethinking Learnability in Offline Data-driven Optimization1 Sep 2026
- 2026Spec2Twin-Chain: Orchestrating Bi-Level Optimization with LLMs for Blockchain Digital Twin Construction30 Aug 2026
- 2026PTP: Previous-Token Prediction based LLM Inversion for Near-Exact Prompt Reconstruction31 Jul 2026
- 2026Generative Refinement for Low-Budget Black-Box Optimization1 Jul 2026
- 2026Conformal Candidate Certification for Offline Model-Based Optimization13 Jun 2026
- 2026Support-Proximity Augmented Diffusion Estimation for Offline Black-Box Optimization11 May 2026
- 2026Black-Box Optimization From Small Offline Datasets via Meta Learning with Synthetic Tasks14 Apr 2026
- 2026Offline Materials Optimization with CliqueFlowmer6 Mar 2026
- 2026The Offline-Frontier Shift: Diagnosing Distributional Limits in Generative Multi-Objective Optimization11 Feb 2026
- 2026Diffusion Large Language Models for Black-Box Optimization20 Jan 2026
- 2025ROOT: Rethinking Offline Optimization as Distributional Translation via Probabilistic Bridge19 Sep 2025
- 2025Posterior Inference in Latent Space for Scalable Constrained Black-box Optimization1 Jul 2025
12 newest of 61 All 61 (JSON)