Search papers, labs, and topics across Lattice.
The authors introduce TriProRep, a structure-aware pretraining method that jointly models amino-acid identity, backbone geometry, and local full-atom geometry using VQ-VAE tokenizers. TriProRep is pretrained to recover original tokens from generator-corrupted views, enabling the model to distinguish plausible but incorrect cross-view augmentations. They also introduce RepSP, a benchmark for evaluating protein representations in structure-predictive settings, and show that TriProRep improves performance on these tasks compared to sequence-only and prior structure-aware representation models.
Encoding proteins at the atom level unlocks better structure prediction, outperforming sequence-only and prior structure-aware methods on a new benchmark.
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.
