Antibody sequence optimization with gradient-guided discrete walk-jump sampling

Published: 04 Mar 2024, Last Modified: 29 Apr 2024GEM PosterEveryoneRevisionsBibTeXCC BY 4.0
Track: Machine learning: computational method and/or computational results
Keywords: Machine learning, Antibody design, Biological sequence optimization
TL;DR: Combining denoising score model and discriminator model for discretized Langevin MCMC, our method generates high quality antibody sequences that optimize a given objective.
Abstract: For enhancing the rapid discovery and delivery of antibody-based drugs, antibody attribute optimization is essential for the real-world application of therapeutic antibody sequence design. Using a generative machine learning model for antibody design is a promising direction for such tasks. However, existing methods struggle in balancing error accumulation, scalability, and targeted attribute optimization. In this work, we propose gradient-guided discrete walk-jump sampling (gg-dWJS), a novel discrete sequence generation method for antibody attribute optimization. Leveraging gradient guidance in the noisy manifold, we sample from the smoothed data manifold by applying discretized Markov chain Monte Carlo (MCMC) using a denoising model with the gradient-guidance from a discriminative model. This is followed by jumping to the discrete data manifold using a conditional one-step denoising. Through evaluation on both discrete image and antibody sequence generation tasks, we show that our method generates high-quality samples that are well-optimized for specific tasks.
Submission Number: 67
Loading