Authors
Tianyu Wu, Lin Zhu
Published in
Bioinformatics (Oxford, England). Volume 42. Issue Supplement_2. Aug 01, 2026.
Abstract
Generative models for protein backbone design have to simultaneously ensure geometric validity, sampling efficiency, and scalability to long protein chains. However, most existing approaches rely on iterative refinement, quadratic attention mechanisms, or post-hoc geometry correction, leading to a persistent trade-off between computational efficiency and structural fidelity.
We present Physics-Informed Mamba (PI-Mamba), a generative model that enforces exact local covalent geometry by construction while enabling linear-time inference. PI-Mamba integrates a differentiable constraint-enforcement operator into a flow-matching framework and couples it with a Mamba-based state-space architecture. To improve optimization stability and backbone realism, we introduce a spectral initialization derived from the Rouse polymer model and an auxiliary cis-proline awareness head. Across benchmark tasks, PI-Mamba demonstrates the advantage in scalable, physically valid backbone generation: on a single A5000 GPU (24GB), it generates backbones beyond 2000 residues, producing 2000-residue samples in 9.49 s with only 0.91 GB peak VRAM, while preserving exact local geometry with 0.0% local geometry violations and maintaining strong designability on short-chain benchmarks (mean scTM = 0.910 at L = 100).
Code and distilled data are available at https://github.com/forxhunter/PI-mamba.
PMID:
42635247
Bibliographic data and abstract were imported from PubMed on 24 Aug 2026.
Read full publication at:
Please sign in
to see all details.
Advertisement
Stats
- Recommendations n/a n/a positive of 0 vote(s)
- Views 5
- Comments 0