Inference-Time Steering Across Scaffolds and Epitopes for Sample-Efficient De Novo Nanobody Design
Mason Minot, Simon Friedensohn, Cédric R. Weber
Abstract
Generative molecular design typically relies on over-generation and post-hoc filtering. Inference-time steering methods like Feynman-Kac (FK) steering improve efficiency by biasing trajectories toward high-reward candidates, yet they optimize within a single fixed configuration. In de novo nanobody design, upfront commitment to conditioning choices like scaffold identity or target epitope creates a bottleneck. Since yield varies across these conditions, steering unpromising setups wastes compute. We recast generative condition selection as an inference-time resource-allocation problem. We propose two dynamic budget allocators: a joint Global Sequential Monte Carlo (SMC) formulation that steers and allocates compute across configurations concurrently, and a decoupled approach adapting Batched Successive Elimination (BaSE). Empirically, these dynamic allocators increase the yield of designable candidates per GPU-hour nearly 5× without sacrificing diversity across distinct therapeutic targets and generative configurations.
Create a lesson
Related papers
AI Models Can Provably Hide Arbitrary Capabilities
Andis Draguns, Sumeet Ramesh Motwani, Raymond Douglas et al.
KL Before and After Mixing: A Round-Complexity Separation in Parallel Sampling
Tianyi Wu
Designing Protein Structures from Natural Language
Joseph Lee, Yanjing Li, Bo Qiang et al.
Revisiting Tabular Deep Regression: Objectives and the Limits of Global Early Stopping
Danil Rudenko, Georgii Bulgakov, Nikolay Kartashev et al.
Prediction Is Not a Sufficient Memory State
Robert Jenkinson Alvarez
Partially Informed Stochastic Optimization via Zeroth-Order Residual Tracking
El Mahdi Chayti, Alireza Mirrokni, Martin Jaggi