English

Navigating heterogeneous protein landscapes through geometry-aware smoothing

Computational Engineering, Finance, and Science 2026-02-12 v1

Abstract

The evolutionary fitness landscape of biological molecules is extremely sparse and heterogeneous, with functional sequences forming isolated dense ``islands'' within a vast combinatorial space of largely non-functional variants. Protein sequences, in particular, exemplify this structure, yet most generative artificial intelligence models implicitly assume a homogeneous data distribution. We show that this assumption fundamentally breaks down in heterogeneous biological sequence spaces: fixed global noise levels impose a destructive trade-off, either oversmoothing dense functional clusters or fragmenting sparse regions and producing non-functional hallucinations. To address this limitation, we introduce \emph{Density-Dependent Smoothing} (DDS), a geometry-aware generative framework that adapts stochastic smoothing to the local density of the underlying sequence landscape. By inversely coupling diffusion noise to estimated sequence density, DDS enables gentle refinement in high-density functional regions while promoting controlled exploration across sparse regions. Implemented as a plug-in mechanism for discrete molecular sampling, DDS consistently outperforms state-of-the-art diffusion and autoregressive models across antibody repertoires, therapeutic antibody design, antimicrobial peptide generation and coronavirus antibody design. Together, these results show that fixed global smoothing assumptions fundamentally limit generative modeling in sparse biological sequence spaces, and that geometry-aware smoothing removes this constraint, enabling reliable exploration and design previously unattainable with fixed-noise generative models.

Keywords

Cite

@article{arxiv.2602.10422,
  title  = {Navigating heterogeneous protein landscapes through geometry-aware smoothing},
  author = {Srinivas Anumasa and Barath Chandran and Tingting Chen and Nuwaisir Mohammad Rahman and Yingtao Zhu and Rushi Shah and Hongyu He and Peisong Zhang and Yizhen Liao and Yiming Tang and Yong Shen and Tianfan Fu and Rui Qing and Xiao Li and Sebastian Maurer-Stroh and Xinyi Su and Zhizhuo Zhang and Dianbo Liu},
  journal= {arXiv preprint arXiv:2602.10422},
  year   = {2026}
}