Learning Latent Conformational Landscapes Encoded in Cryo-EM
1. The paper argues that single-particle cryo-EM already contains a continuous, probabilistic “conformational landscape” (not just a few discrete classes), where the latent-space density reflects state occupancy and can be interpreted as a relative free-energy surface via Boltzmann statistics.
2. Core method: CryoUNI, a universal cryo-EM particle-image encoder pretrained on CryoCRAB-Particle-22M (22M particles; 746 species; broad SCOP2 diversity and molecular-weight range). Pretraining is self-supervised and explicitly designed to separate structural signal from noise using odd/even half-image denoising supervision.
3. Downstream heterogeneity modeling: CryoUNI is adapted with a VAE-style framework (CryoDRGN-like) where a lightweight projector maps encoder features to latent variables, and an implicit neural volume decoder reconstructs densities (in the frequency/Hartley domain) conditioned on the latent code.
4. Landscape analysis: WAVE (Watershed Analysis of Variational Embeddings) converts particle embeddings into a density field (KDE), detects density peaks as conformational states, partitions basins via marker-controlled watershed, and traces transition pathways through high-density regions by solving an Eikonal/Fast-Marching problem on a cost field defined by negative log-density.
5. A key conceptual claim is made operational: relative energies between states are inferred directly from density ratios, ΔGr = −kBT ln(ρA/ρB). This yields energy profiles along WAVE-traced pathways and provides a quantitative way to discuss barriers and intermediates from cryo-EM data.
6. Physical grounding test (integrin αvβ8, EMPIAR-10345): independent 20 µs all-atom MD simulations define a thermodynamic landscape of leg motion in spherical coordinates (θ, ϕ). CryoUNI’s learned latent axes align strongly with these physical coordinates (PC1–θ r=0.982; PC2–ϕ r=0.963), and CryoUNI volumes sampled across the landscape match MD snapshots in leg orientation and structural details.
7. Benchmarking on heterogeneous mixtures: on CryoBench Ribosembly (16 discrete assembly states), WAVE on CryoUNI embeddings automatically recovers all 16 states with 99.4% accuracy, and pairwise distances between latent peaks correlate with true structural similarity (Mantel r = −0.887). On Tomotwin-100, CryoUNI yields near-perfect separation (~99.96–99.98% accuracy across clustering methods), highlighting representation quality as the main driver.
8. Continuous heterogeneity case: on IgG-1D (a conformational continuum), CryoUNI captures a continuous manifold and improves per-particle reconstruction fidelity (highest mean AUC-FSC with low variance), suggesting the approach is not limited to discrete classification problems.
9. Biological discovery example (LIS1-mediated dynein activation, EMPIAR-12715): a global landscape organizes particles into three major basins (open-bent, open-straight, motor-bound), while local/hierarchical landscapes reveal sub-states, including a low-population open-straight intermediate with distinct LIS1 stoichiometry (Straight/2x) that would be attenuated in consensus reconstructions; overall, 12 sub-states are resolved with reconstructions spanning ~2.75–6.44 Å.
10. Experimental “closing the loop” (KCTD5/CUL3NTD/Gβγ, EMPIAR-11734): latent-density-derived energy thresholds guide particle selection to improve reconstruction consistency/resolution; WAVE also extracts continuous transition trajectories connecting four known basins (A→D→C and A→B→C), providing experimentally derived pathways consistent with prior simulation/morphing interpretations.
💻Code:
github.com/Cellverse/cryouni
📜Paper:
biorxiv.org/content/10.64898…
#cryoEM #StructuralBiology #ComputationalBiology #MachineLearning #DeepLearning #ProteinDynamics #VariationalAutoencoder #MolecularDynamics #RepresentationLearning #SingleParticleAnalysis