English

Conditionally Site-Independent Neural Evolution of Antibody Sequences

Machine Learning 2026-05-28 v4 Populations and Evolution

Abstract

Common deep learning approaches for antibody engineering focus on modeling the marginal distribution of sequences. By treating sequences as independent samples, however, these methods overlook affinity maturation as a rich and largely untapped source of information about the evolutionary process by which antibodies explore the underlying fitness landscape. In contrast, classical phylogenetic models explicitly represent evolutionary dynamics but lack the expressivity to capture complex epistatic interactions. We bridge this gap with CoSiNE, a continuous-time Markov chain parameterized by a deep neural network. Mathematically, we prove that CoSiNE provides a first-order approximation to the intractable sequential point mutation process, capturing epistatic effects with an error bound that is quadratic in branch length. Empirically, CoSiNE outperforms state-of-the-art language models in zero-shot variant effect prediction by explicitly disentangling selection from context-dependent somatic hypermutation. Finally, we introduce Guided Gillespie, a classifier-guided sampling scheme that steers CoSiNE at inference time, enabling efficient optimization of antibody binding affinity toward specific antigens.

Keywords

Cite

@article{arxiv.2602.18982,
  title  = {Conditionally Site-Independent Neural Evolution of Antibody Sequences},
  author = {Stephen Zhewen Lu and Aakarsh Vermani and Kohei Sanno and Jiarui Lu and Frederick A Matsen and Milind Jagota and Yun S. Song},
  journal= {arXiv preprint arXiv:2602.18982},
  year   = {2026}
}

Comments

28 pages, 15 figures. Accepted as a poster at ICML 2026