English

Generating Highly Designable Proteins with Geometric Algebra Flow Matching

Machine Learning 2024-11-11 v1 Machine Learning

Abstract

We introduce a generative model for protein backbone design utilizing geometric products and higher order message passing. In particular, we propose Clifford Frame Attention (CFA), an extension of the invariant point attention (IPA) architecture from AlphaFold2, in which the backbone residue frames and geometric features are represented in the projective geometric algebra. This enables to construct geometrically expressive messages between residues, including higher order terms, using the bilinear operations of the algebra. We evaluate our architecture by incorporating it into the framework of FrameFlow, a state-of-the-art flow matching model for protein backbone generation. The proposed model achieves high designability, diversity and novelty, while also sampling protein backbones that follow the statistical distribution of secondary structure elements found in naturally occurring proteins, a property so far only insufficiently achieved by many state-of-the-art generative models.

Keywords

Cite

@article{arxiv.2411.05238,
  title  = {Generating Highly Designable Proteins with Geometric Algebra Flow Matching},
  author = {Simon Wagner and Leif Seute and Vsevolod Viliuga and Nicolas Wolf and Frauke Gräter and Jan Stühmer},
  journal= {arXiv preprint arXiv:2411.05238},
  year   = {2024}
}

Comments

To be published in proceedings of NeurIPS 2024

R2 v1 2026-06-28T19:52:28.932Z