Orientation Estimation of Cryo-EM Images for Molecules with Arbitrary Symmetry by Semidefinite Programming
Authors: Yunpeng Shi, Amit Singer, Jinxin Wang, Ruiyi Yang
Organizations: Department of Mathematics, University of California, Davis · Department of Mathematics and Program in Applied and Computational Mathematics, Princeton University · Department of Statistics, University of Chicago · School of Mathematical Sciences, MOE-LSC, Institute of Natural Sciences, Shanghai Jiao Tong University
This paper extends the non-unique games framework introduced in Bandeira et al. (2020) to address symmetric structures. Precisely, we consider the problem
where G is a compact group, S⊂G is a finite subgroup of symmetry elements, and fij:G→R is a loss function. A notable example is common lines based orientation estimation of cryo-EM images for molecules with point group symmetry S, where G is the rotation group SO(3). With a semidefinite programming relaxation, we turn the problem into a convex program that is solved with the alternating direction method of multipliers. Unlike existing approaches which are tailored to specific symmetry groups, ours provides a unified computational framework for addressing all point group symmetries. We demonstrate the applicability of the proposed algorithm on the cryo-EM orientation estimation problem on both simulated and experimental datasets. In particular, our algorithm is effective in generating high-quality ab initio models from only a few class averages computed from experimental images.
Figures & tables
Figure 1: Simulated projection images of EMD-2660 at different SNR values.
SNR
Clean
1
1/2
1/4
1/8
C1 (without PR)
0.030 (18.10)
0.041 (30.53)
0.034 (33.50)
0.088 (44.10)
0.570 (62.08)
Synchronization
0.000 (7.58)
0.005 (23.25)
0.046 (33.27)
>1
>1
C2 (without PR)
0.032 (16.60)
0.026 (25.61)
0.047 (32.12)
0.195 (49.83)
>1
C2 (with PR)
0.017 (12.73)
0.017 (23.67)
0.027 (29.58)
0.043 (37.89)
>1
Synchronization
0.521 (8.29)
0.684 (40.00)
>1
>1
>1
C3 (without PR)
0.026 (14.53)
0.033 (24.38)
0.096 (38.16)
0.852 (60.91)
>1
Table 1: MSE ( 26 ) with N=40 images of size L=128 for different symmetry groups and SNRs. FSCs of the reconstructed volumes are reported in the parenthesis in angstrom. Here PR refers to the proximal refinement step as in Section 2.3 .
SNR
Clean
1
1/2
1/4
1/8
C1 (without PR)
0.022 (14.28)
0.028 (21.76)
0.039 (26.59)
0.076 (34.91)
0.398 (46.79)
Synchronization
0.000 (7.54)
0.003 (18.53)
0.021 (24.55)
0.150 (39.25)
>1
C2 (without PR)
0.019 (11.06)
0.021 (20.58)
0.043 (25.22)
0.102 (35.73)
>1
C2 (with PR)
0.013 (8.92)
0.017 (20.25)
0.027 (22.75)
0.035 (27.04)
>1
Synchronization
0.003 (7.54)
0.248 (24.35)
0.661 (42.46)
>1
>1
C3 (without PR)
0.036 (14.64)
0.032 (20.82)
0.138 (33.92)
0.568 (50.78)
>1
Table 2: MSE ( 26 ) with N=100 images of size L=128 for different symmetry groups and SNRs. FSCs of the reconstructed volumes are reported in the parenthesis in angstrom. Here PR refers to the proximal refinement step as in Section 2.3 .
SNR
Clean
1
1/2
1/4
1/8
D3 (without PR)
0.049 (35.52)
0.035 (52.32)
0.204 (61.37)
0.684 (64.93)
>1
D3 (with PR)
0.040 (17.86)
0.027 (51.54)
0.123 (60.68)
0.448 (60.07)
>1
D4 (without PR)
0.158 (60.60)
0.220 (61.69)
0.576 (43.23)
>1
>1
D4 (with PR)
0.082 (19.43)
0.097 (50.67)
0.511 (38.68)
>1
>1
D5 (without PR)
0.101 (12.52)
0.180 (61.69)
>1
>1
>1
D5 (with PR)
0.088 (12.15)
0.167 (49.68)
>1
>1
>1
Table 3: MSE ( 26 ) with N=40 images of size L=128 for different symmetry groups and SNRs. FSCs of the reconstructed volumes are reported in the parenthesis in angstrom. Here PR refers to the proximal refinement step as in Section 2.3 .
SNR
Clean
1
1/2
1/4
1/8
D3 (without PR)
0.085 (12.72)
0.064 (44.71)
0.118 (52.61)
0.903 (50.78)
>1
D3 (with PR)
0.082 (16.51)
0.072 (28.61)
0.096 (47.53)
0.646 (58.90)
>1
D4 (without PR)
0.029 (14.66)
0.160 (27.36)
0.455 (62.33)
>1
>1
D4 (with PR)
0.024 (16.71)
0.184 (26.25)
0.456 (62.08)
>1
>1
D5 (without PR)
0.110 (12.65)
0.185 (45.86)
0.808 (53.78)
>1
>1
D5 (with PR)
0.109 (16.95)
0.125 (44.83)
0.771 (42.61)
>1
>1
Table 4: MSE ( 26 ) with N=100 images of size L=128 for different symmetry groups and SNRs. FSCs of the reconstructed volumes are reported in the parenthesis in angstrom. Here PR refers to the proximal refinement step as in Section 2.3 .
N
20
50
100
200
300
Time (minutes)
2.0
6.8
21.3
78.7
176.2
Table 5: Computation time of solving the SDP ( 18 ) for different number of images with truncation level K=12 .
Figure 2: Raw images from EMPIAR 10081 (top) and class averages computed (bottom).
Figure 3: Visualizations of the ab initio model, the refined model, and the reference model.
Figure 4: Fourier shell correlations for the ab initio model and the refined model.
Figure 5: Raw images from EMPIAR 12036 (top) and class averages computed (bottom).
Figure 6: Visualizations of the ab initio model, the refined model, and the reference model.
Figure 7: Fourier shell correlations for the ab initio model and the refined model.
Cryo-electron microscopy reconstructs biological structures from images of frozen specimens, from cellular volumes to atomic-resolution molecular models.
Single-particle cryo-electron microscopy (cryo-EM) pose estimation is traditionally solved anew for each dataset, where iterative refinement is done from scratch while the estimator learns to store the molecule in its weights. In this work, we show that pose inference is a generalizable, specimen-agnostic operation when conditioned explicitly on a reference volume. We introduce ARCHER, an amortized contrastive classifier that models the pose posterior over a discrete rotation grid. Trained across a variety of protein structures, it operates zero-shot without retraining per structure. This transferability is grounded in Fourier-space information mechanics, where all specimen dependence is captured by the reference structure's power spectrum and spatial extent. ARCHER achieves a median angular error of 5.0° on 100 held-out test structures and 2.5° on experimental particles, matching dedicated estimators within 0.16 Å in 3D reconstruction. Crucially, downstream conformational signal is preserved. The leading conformational coordinate correlates at 0.97 with deposited benchmarks, faithfully reconstructing free-energy basins and mobile domains. These results overall demonstrate that cryo-EM pose estimation can be generalized across different structures.
Nhan D. Nguyen, Bao Pham
Pritzker School of Molecular Engineering, University of Chicago, Chicago, Illinois 60637, United States · Department of Computer Science, Rensselaer Polytechnic Institute, Troy, New York 12180, United States
Crystal-orientation maps are physical fields defined only up to crystal symmetry; electron backscatter diffraction (EBSD) resolves them experimentally, but acquisition-time constraints limit spatial resolution. Unlike conventional images, EBSD data lie on the quotient space SO(3)/G, where G is the crystal-symmetry group. Standard Euclidean interpolation can therefore mix symmetry-equivalent representations and blur grain boundaries. We introduce the Symmetry-Group-Aware Super-Resolution Attention Network (SG-SRAN), which incorporates crystal symmetry and boundary preservation by design. A frozen, locally isometric encoder maps equivalent orientations to a common latent representation in which Euclidean distance approximates misorientation. Super-resolution is performed in this space, with each high-resolution token restricted to a feature-consistent local support to prevent cross-boundary mixing. A dictionary-based decoder then recovers valid orientations. Across FCC and HCP benchmarks, SG-SRAN matches 15-16 million parameter backbones using only 27-49k trainable parameters, while achieving the lowest p68 errors, highest inverse-pole-figure fidelity, and zero-shot transfer to unseen alloys.
Umang Garg, Warren Zamudio, McLean P. Echlin +3
Electrical and Computer Engineering, University of California Santa Barbara, Santa Barbara, 93106, California, USA. · Mechanical Engineering, University of California Santa Barbara, Santa Barbara, 93106, California, USA. · Materials, University of California Santa Barbara, Santa Barbara, 93106, California, USA.
We present a geometry-aware method for heterogeneous single-particle cryogenic electron microscopy (cryo-EM) reconstruction that predicts atomic backbone conformations. To incorporate protein-structure priors, we represent the backbone as a graph and use a graph neural network (GNN) autodecoder that maps per-image latent variables to 3D displacements of a template conformation. The objective combines a data-discrepancy term based on a differentiable cryo-EM forward model with geometric regularization, and it supports unknown orientations via ellipsoidal support lifting (ESL) pose estimation. On synthetic datasets derived from molecular dynamics trajectories, the proposed GNN achieves higher accuracy compared to a multilayer perceptron (MLP) of comparable size, highlighting the benefits of a geometry-informed inductive bias.
Jonathan Krook, Axel Janson, Joakim Andén +2
KTH Royal Institute of Technology · Harvard University Department of Mathematics School of Engineering and Applied Sciences