Biology

Atom-level Protein Representation Learning Improves Protein Structure Prediction

AI Insight

This study introduces TriProRep, a protein representation learning method that jointly models three complementary views of protein structure at the residue level: amino-acid identity, backbone geometry, and local full-atom geometry. Each view is encoded using VQ-VAE tokenizers, and the model is pretrained to reconstruct original tokens from corrupted inputs, enabling it to distinguish plausible but incorrect structural configurations from authentic ones. The authors also introduce RepSP, a benchmark designed to evaluate protein representations specifically in structure-predictive tasks, including homodimer co-folding, residue-level interaction property prediction, and representation-aligned monomer structure prediction, across which TriProRep consistently outperforms sequence-only and prior structure-aware baselines.


Improved protein structure prediction methods have direct implications for drug discovery, protein engineering, and understanding disease-related molecular mechanisms, as more accurate structural representations can enhance the reliability of computational tools used in these fields.


arXiv:2605.22133v1 Announce Type: new
Abstract: Recent advances in generative modeling show that pretrained representations can improve generation as conditioning features or alignment targets. Motivated by this, we study protein representations for predicting structures beyond conventional function annotation. We propose TriProRep, a structure-aware pretraining method that jointly models three aligned residue-level views: amino-acid identity, backbone geometry, and local full-atom geometry, discretely encoded via VQ-VAE tokenizers. By pretraining to recover original tokens from generator-corrupted views, TriProRep learns to distinguish plausible but incorrect cross-view augmentations from the original protein. We further introduce RepSP, a benchmark for evaluating protein representations in structure-predictive settings. RepSP tests three uses of representations: homodimer co-folding from apo-chain representations, residue-level prediction of homodimer-derived interaction properties, and representation-aligned monomer structure prediction. Across these tasks, TriProRep improves over sequence-only and prior structure-aware representation models, while maintaining competitive performance on conventional benchmarks.

Source: Atom-level Protein Representation Learning Improves Protein Structure Prediction