Geometric Self-Supervised Pretraining on 3D Protein Structures using Subgraphs

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Chatzianastasis, Michail, Zhang, Yang, Dasoulas, George, Vazirgiannis, Michalis
Format: Preprint
Veröffentlicht: 2024
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866929550973206528
author Chatzianastasis, Michail
Zhang, Yang
Dasoulas, George
Vazirgiannis, Michalis
author_facet Chatzianastasis, Michail
Zhang, Yang
Dasoulas, George
Vazirgiannis, Michalis
contents Protein representation learning aims to learn informative protein embeddings capable of addressing crucial biological questions, such as protein function prediction. Although sequence-based transformer models have shown promising results by leveraging the vast amount of protein sequence data in a self-supervised way, there is still a gap in exploiting the available 3D protein structures. In this work, we propose a pre-training scheme going beyond trivial masking methods leveraging 3D and hierarchical structures of proteins. We propose a novel self-supervised method to pretrain 3D graph neural networks on 3D protein structures, by predicting the distances between local geometric centroids of protein subgraphs and the global geometric centroid of the protein. By considering subgraphs and their relationships to the global protein structure, our model can better learn the geometric properties of the protein structure. We experimentally show that our proposed pertaining strategy leads to significant improvements up to 6\%, in the performance of 3D GNNs in various protein classification tasks. Our work opens new possibilities in unsupervised learning for protein graph models while eliminating the need for multiple views, augmentations, or masking strategies which are currently used so far.
format Preprint
id arxiv_https___arxiv_org_abs_2406_14142
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Geometric Self-Supervised Pretraining on 3D Protein Structures using Subgraphs
Chatzianastasis, Michail
Zhang, Yang
Dasoulas, George
Vazirgiannis, Michalis
Quantitative Methods
Machine Learning
Biomolecules
Protein representation learning aims to learn informative protein embeddings capable of addressing crucial biological questions, such as protein function prediction. Although sequence-based transformer models have shown promising results by leveraging the vast amount of protein sequence data in a self-supervised way, there is still a gap in exploiting the available 3D protein structures. In this work, we propose a pre-training scheme going beyond trivial masking methods leveraging 3D and hierarchical structures of proteins. We propose a novel self-supervised method to pretrain 3D graph neural networks on 3D protein structures, by predicting the distances between local geometric centroids of protein subgraphs and the global geometric centroid of the protein. By considering subgraphs and their relationships to the global protein structure, our model can better learn the geometric properties of the protein structure. We experimentally show that our proposed pertaining strategy leads to significant improvements up to 6\%, in the performance of 3D GNNs in various protein classification tasks. Our work opens new possibilities in unsupervised learning for protein graph models while eliminating the need for multiple views, augmentations, or masking strategies which are currently used so far.
title Geometric Self-Supervised Pretraining on 3D Protein Structures using Subgraphs
topic Quantitative Methods
Machine Learning
Biomolecules
url https://arxiv.org/abs/2406.14142