Deep Extrapolation for Attribute-Enhanced Generation
Alvin Chan, Ali Madani, Ben Krause, Nikhil Naik
Abstract
Attribute extrapolation in sample generation is challenging for deep neural networks operating beyond the training distribution. We formulate a new task for extrapolation in sequence generation, focusing on natural language and proteins, and propose GENhance, a generative framework that enhances attributes through a learned latent space. Trained on movie reviews and a computed protein stability dataset, GENhance can generate strongly-positive text reviews and highly stable protein sequences without being exposed to similar data during training. We release our benchmark tasks and models to contribute to the study of generative modeling extrapolation and data-driven design in biology and chemistry: https://github.com/salesforce/genhance .
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 9fccd78f-c47a-46e5-a8cd-332ed8755185Cited by top-tier papers11
- Parallel-mentoring for Offline Model-based OptimizationCan Chen, Christopher Beckham, Zixuan Liu, Xue (Steve) Liu et al.NeurIPS 2023 · 36 citations
- Bootstrapped Training of Score-Conditioned Generator for Offline Design of Biological SequencesMinsu Kim, Federico Berto, Sungsoo Ahn, Jinkyoo ParkNeurIPS 2023 · 30 citations
- Bidirectional Learning for Offline Model-based Biological Sequence DesignCan Chen, Yingxue Zhang, Xue Liu, Mark CoatesICML 2023 · 30 citations
- Extrapolative Controlled Sequence Generation via Iterative RefinementVishakh Padmakumar, Richard Yuanzhe Pang, He He, Ankur P. ParikhICML 2023 · 13 citations
- Contrastive losses as generalized models of global epistasisDavid H. Brookes, Jakub Otwinowski, Sam SinaiNeurIPS 2024 · 8 citations
Builds on6
- Plug and Play Language Models: A Simple Approach to Controlled Text GenerationSumanth Dathathri, Andrea Madotto, Janice Lan, Jane Hung et al.ICLR 2020 · 1,166 citations
- Delving into Deep Imbalanced RegressionYuzhe Yang, Kaiwen Zha, Ying-Cong Chen, Hao Wang et al.ICML 2021 · 385 citations
- How Neural Networks Extrapolate: From Feedforward to Graph Neural NetworksKeyulu Xu, Mozhi Zhang, Jingling Li, Simon Shaolei Du et al.ICLR 2021 · 364 citations
- Model Inversion Networks for Model-Based OptimizationAviral Kumar, Sergey LevineNeurIPS 2020 · 129 citations
- Autofocused oracles for model-based designClara Fannjiang, Jennifer ListgartenNeurIPS 2020 · 90 citations
Related papers
- Learning Extrapolative Sequence Transformations from Markov ChainsSophia Hager, Aleem Khan, Andrew Wang, Nicholas AndrewsICML 2025
- Generating Novel, Designable, and Diverse Protein Structures by Equivariantly Diffusing Oriented Residue CloudsYeqing Lin, Mohammed AlQuraishiICML 2023 · 105 citations
- Probabilistic Transformer: Modelling Ambiguities and Distributions for RNA Folding and Molecule DesignJörg K. H. Franke, Frederic Runge, Frank HutterNeurIPS 2022 · 19 citations
- Dynamics-Informed Protein Design with Structure ConditioningUrszula Julia Komorowska, Simon V. Mathis, Kieran Didi, Francisco Vargas et al.ICLR 2024 · 7 citations
- GenomeQA: Benchmarking General Large Language Models for Genome Sequence UnderstandingWeicai Long, Yusen Hou, Junning Feng, Houcheng Su et al.ACL 2026
