CryoNet.Refine: A One-step Diffusion Model for Rapid Refinement of Structural Models with Cryo-EM Density Map Restraints
Authors: Fuyao Huang, Xiaozhu Yu, Kui Xu, Qiangfeng Cliff Zhang
Organizations: State Key Laboratory of Membrane Biology–Membrane Structure and Artificial Intelligence Biology Branch, Hangzhou, China · State Key Laboratory of Membrane Biology, Beijing Tsinghua Institute for Frontier Interdisciplinary Innovation, Beijing, China · State Key Laboratory of Membrane Biology, Beijing Frontier Research Center of Biological Structures, Tsinghua-Peking Joint Center for Life Sciences, School of Life Sciences, Tsinghua University, Beijing, China
High-resolution structure determination by cryo-electron microscopy (cryo-EM) requires the accurate fitting of an atomic model into an experimental density map. Traditional refinement pipelines such as Phenix.real_space_refine and Rosetta are computationally expensive, demand extensive manual tuning, and present a significant bottleneck for researchers. We present CryoNet.Refine, an end-to-end deep learning framework that automates and accelerates molecular structure refinement. Our approach utilizes a one-step diffusion model that integrates a density-aware loss function with robust stereochemical restraints, enabling rapid optimization of a structure against experimental data. CryoNet.Refine provides a unified and versatile solution capable of refining protein complexes as well as DNA/RNA-protein complexes. In benchmarks against Phenix.real_space_refine, CryoNet.Refine consistently achieves substantial improvements in both model-map correlation and overall geometric quality metrics. By offering a scalable, automated, and powerful alternative, CryoNet.Refine aims to serve as an essential tool for next-generation cryo-EM structure refinement. Web server: https://cryonet.ai/refine; Source code: https://github.com/kuixu/cryonet.refine.
Figures & tables
Figure 1: (a) Workflow of CryoNet.Refine . Atomic models are colored by model–map correlation coefficients (CC), with blue indicating high CC and red indicating low CC. (b) The input atomic model, refined atomic model within cryo-EM density map.
Figure 2: (a) Overview of the CryoNet.Refine framework. The CryoNet.Refine Network consists of five modules: Atom encoder, Sequence embedder, Pairformer module, Diffusion module, and Density generator. The input atomic model is first processed by the encoders, and the resulting features are fed into a one-step diffusion module to generate an initial refined atomic model. Subsequently, the Density generator creates a synthetic density map, which is used to compute density loss against the input density map and geometry loss based on geometry restraints. These losses are then backpropagated to optimize the diffusion module, while the atomic model is further refined through multiple recycle steps until convergence. (b) Model-map correlation coefficient trajectory over 234 recycling CryoNet.Refine iterations on the structure of the human concentrative nucleoside transporter CNT3( ( Bank, ) ). The input density map is EMD-0775, the input atomic model is predicted by AlphaFold3 .
Category
Metrics
AlphaFold3
Phenix.real_space_refine
CryoNet.Refine
Model-map Correlation Coefficients
CCmask↑
0.38
0.54
0.59
CCbox↑
0.41
0.53
0.57
CCmc↑
0.40
0.55
0.60
CCsc↑
0.39
0.55
0.58
CCpeaks↑
0.27
0.40
0.45
CCvolume↑
0.42
0.55
0.60
Table 1: Performance on benchmark of 110 protein complexes(↑: higher is better, ↓: lower is better).
Figure 3: Model–map correlation coefficients on protein complex benchmark.
Figure 4: Model geometric metrics. (Color gradient: blue for better, red for worse.)
Figure 5: The input atomic models from AlphaFold3 , the refined atomic model from Phenix.real_space_refine and CryoNet.Refine on the Medicago truncatula HISN5 protein (PDB- 7oj5 ; EMD-22692) and the human concentrative nucleoside transporter CNT3 (PDB- 6ksw , EMD-0775) complex. Inserts in the right panel show that the main-chains and side-chains generated from CryoNet.Refine model align well with the density map.
Category
Metrics
AlphaFold3
Phenix.real_space_refine
CryoNet.Refine
Model–map Correlation Coefficients
CCmask↑
0.40
0.57
0.65
CCbox↑
0.49
0.61
0.67
CCmc↑
0.45
0.62
0.61
CCsc↑
0.42
0.58
0.67
CCpeaks↑
0.35
0.51
0.60
CCvolume↑
0.48
0.61
0.69
Table 2: Performance on DNA/RNA–protein complexes.
Figure 6: Model–map correlation coefficients on DNA/RNA-protein complex.
Figure 7: The input atomic models from AlphaFold3 , the refined atomic models from Phenix.real_space_refine and CryoNet.Refine are shown on the Cas12i(E894A)–crRNA–dsDNA (PDB- 6w5c , EMD-21541) and the active-state Dot1 bound to the H4K16ac nucleosome (PDB- 7k6q , EMD-22692) complex. Inserts in the right panel show that the main-chains and side-chains generated by CryoNet.Refine align well with the density map.
Figure 8: Runtime comparison.
Appendix figures & tables10 assets
Supplementary material from the paper’s appendix.
Appendix
Figure 9: Dataset statistics of density map resolution, number of chains, and number of residues.
Figure 10: Data Patterns and Visualizations of the Density Generator Output
Loss Functions
Weight
Value
Density loss ( Lden )
γden
20.0
Ramachandran loss ( Lrama )
γrama
500
Rotation loss ( Lrot )
γrot
500
Angle loss ( Langle )
γangle
2
C β loss ( LCβ )
γCβ
50
Violation loss ( Lviol )
γviol
1000
Appendix
Table 3: Loss function weights
Figure 11: Visualization of Ramachandran distributions
Figure 12: Model-map Correlation Coefficients (CC) analysis of protein complexes across different recycling numbers (We provide a trajectory video over the entire 234 refinement recyclings in supplementary material).
Figure 13: Model-map Correlation Coefficients (CC) analysis of DNA/RNA–protein complex across different diffusion sampling steps.
Category
Metrics
CryoNet.Refine (One-step)
Classical Diffusion (200 steps)
Model-map Correlation Coefficients
CCmask↑
0.65
0.30
CCbox↑
0.58
0.35
CCmc↑
0.66
0.31
CCsc↑
0.63
0.31
CCpeaks↑
0.47
0.24
CCvolume↑
0.65
0.34
Appendix
Table 4: Quantitative comparison between CryoNet.Refine (One-step) and Classical Diffusion (200 steps) across 27 cases. The multi-step approach fails to preserve structural biology constraints when applied to refinement tasks.
Category
Metrics
γden=0
γrama=0
γrot=0
CryoNet.Refine
Model-map Correlation Coefficients
CCmask↑
0.41
0.65
0.64
0.65
CCbox↑
0.42
0.58
0.57
0.58
CCmc↑
0.42
0.66
0.65
0.66
CCsc↑
0.41
0.64
0.64
0.63
CCpeaks↑
0.24
0.48
0.47
0.47
CCvolume↑
0.43
0.65
0.64
0.65
Appendix
Table 5: Ablation study: how loss functions influence model-map correlation coefficients and model geometric metrics
Category
Metrics
AlphaFold3
Phenix. real_space_refine
Numerical Optimization
CryoNet. Refine
Model-map Correlation Coefficients
CCmask↑
0.44
0.61
0.46
0.65
CCbox↑
0.44
0.55
0.47
0.58
CCmc↑
0.45
0.62
0.46
0.66
CCsc↑
0.44
0.61
0.45
0.64
CCpeaks↑
0.31
0.44
0.32
0.47
CCvolume↑
0.47
0.61
0.48
0.65
Appendix
Table 6: Comparison between CryoNet.Refine and Direct Numerical Optimization (SGD) on a subset of 27 protein complexes. The numerical baseline optimizes geometry well but fails to fit the density map.
Component
Time (ms)
Percentage (%)
Data Loading & Preprocessing
150
∼ 6%
Model Forward Pass (One-step)
200
∼ 7%
Loss Calculation (Density + Geometry)
940
∼ 35%
Backpropagation & Optimization
1420
∼ 52%
Total per Recycle
2710
100%
Appendix
Table 7: Runtime profile of a single refinement recycle (averaged over total recycle).
Cryo-electron microscopy reconstructs biological structures from images of frozen specimens, from cellular volumes to atomic-resolution molecular models.
Protein automodeling from cryo-EM density maps faces unique challenges in enforcing physicochemical validity and managing conformational heterogeneity. Current solvers are often limited to static predictions or require computationally intensive heuristic searches. We present CryoACE, an end-to-end framework that reconstructs precise atomic graphs for both homogeneous and heterogeneous structures. Our method features two key innovations: an atom-centric reconstruction paradigm, where density features are sampled directly at atomic coordinates and iteratively recycled to refine structures, replacing expensive voxel convolutions for efficient multimodal fusion; and a training-free guidance mechanism that leverages predicted local resolution priors to resolve dynamic ambiguity. Validated on a newly constructed high-quality dataset, CryoACE significantly outperforms existing baselines on static benchmarks and, for the first time, unveils atomic-level dynamic conformations on complex real-world datasets like EMPIAR-10345 without relying on pre-built static structures.
Minzhang Li, Mingrui Li, Weichen Qin +5
School of Information Science and Technology, ShanghaiTech University, Shanghai, China · Cellverse Co, Ltd., Shanghai, China.
We introduce PhenixCraft, a fully automated pipeline for building atomic models from cryo-EM density maps. By integrating AlphaFold predictions, we enhance the map-segmentation step in Phenix during model building, addressing challenges posed by noise and artifacts that traditionally hinder this step. Our results demonstrate PhenixCraft's superior performance in TM-scores and sequence accuracy, significantly improving upon the limitations and inefficiencies of traditional model building using Phenix.
Chenwei Zhang
Department of Computer Science The University of British Columbia Vancouver, Canada
Single-particle cryo-electron microscopy (cryo-EM) has become a widely adopted technique for biomolecular structure determination. The conventional cryo-EM computational pipeline first combines many particle images to reconstruct an electrostatic potential (ESP) map and then fits an atomic model to the recovered map. Density reconstruction has high sample complexity, requiring large numbers of particle images and making structure determination high-cost and low-throughput, particularly for heterogeneous samples. Downstream atomic model building, in turn, becomes increasingly difficult as the resolution of the reconstructed map deteriorates. Protein structure prediction models provide strong sequence-derived priors on atomic structure, and experiment-guided approaches can use these priors to recover structures consistent with experimental measurements. Yet, in cryo-EM, such priors are typically integrated only after density reconstruction during atomic model fitting. We introduce Fold'EM, an inference-time framework that combines priors from protein generative models directly with cryo-EM particle images to determine atomic models from a small number of single particle images, bypassing both intermediate density reconstruction and downstream model building against the reconstructed map. Across synthetic and experimental cryo-EM datasets, Fold'EM recovers accurate atomic structures both with known particle orientations and in an ab-initio setting where orientations are inferred jointly with structure. In heterogeneous datasets, Fold'EM further resolves distinct conformational states from mixed particle populations without separately reconstructing a density map and building an atomic model for each state. We believe these results open new avenues for structure determination in the low-sample regime and for characterizing low-population conformational states directly from cryo-EM particles.
Advaith Maddipatla, Märt-Erik Mäeots, Marco Pegoraro +5
Institute of Science and Technology Austria · University of Zurich · Frankfurt Institute for Advanced Studies +2