Protein Graph Neural Networks for Heterogeneous Cryo-EM Reconstruction

2026-02-25

AI Summary

This work addresses the challenge of high-resolution three-dimensional reconstruction in cryo-electron microscopy (cryo-EM) caused by conformational heterogeneity. The authors propose a geometry-aware graph neural network (GNN) auto-decoder that models the protein backbone as a graph structure and jointly optimizes the mapping from latent variables to atomic-level conformations through a differentiable cryo-EM forward model augmented with geometric regularization. To handle unknown orientations, the method incorporates ellipsoidal support lifting (ESL). By integrating structural priors and graph-based representations into heterogeneous reconstruction for the first time, the approach introduces strong geometric inductive bias. Evaluated on synthetic data derived from molecular dynamics trajectories, it significantly outperforms comparably sized multilayer perceptrons, achieving higher accuracy in backbone conformation recovery.

Technology Category

• Machine Learning: Graph-based Machine Learning
• Reasoning under Uncertainty: Graphical Models
• Computer Vision: Other Foundations of Computer Vision

Application Category

• Graph Algorithms and Modeling for the Web: Graph neural networks and deep learning approaches for Web-related graphs
• Semantics and Knowledge: Scalable techniques for the creation, curation, publication, maintenance, and consumption of large, Web-based, structured, reusable, knowledge graphs and ontologies
• Search and Retrieval-Augmented AI: Web learning to rank, online learning, and counterfactual learning for ranking

Abstract

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.

Problem

• cryo-EM reconstruction
• heterogeneous reconstruction
• protein backbone conformation
• single-particle cryo-EM

Innovation

• Graph Neural Network
• Cryo-EM Reconstruction
• Geometric Inductive Bias
• Ellipsoidal Support Lifting
• Protein Backbone Modeling