Rigidity-Aware Geometric Pretraining for
Protein Design and Conformational Ensembles
ICLR 2026

  • 1University of Illinois Urbana-Champaign
  • 2University of Washington
  • 3MPI for Intelligent Systems, Tübingen
  • 4National Research Council Canada
  • 5University of Ottawa
  • 6The Chinese University of Hong Kong

  • *Equal contribution

1. RigidSSL use inertial frames to establish a canonical reference system for protein structures.

2. RigidSSL learns protein dynamics by maximizing mutual information between two conformations $g_0$ and $g_1$.

3. RigidSSL variants improve protein structure designability, novelty, diversity, and secondary structure statistics.

4. RigidSSL generalizes to long protein chains of 700-800 residues.

Citation