Enhancing Protein Predictive Models via Proteins Data Augmentation: A Benchmark and New Directions

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Sun, Rui, Wu, Lirong, Lin, Haitao, Huang, Yufei, Li, Stan Z.
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866911787724570624
author Sun, Rui
Wu, Lirong
Lin, Haitao
Huang, Yufei
Li, Stan Z.
author_facet Sun, Rui
Wu, Lirong
Lin, Haitao
Huang, Yufei
Li, Stan Z.
contents Augmentation is an effective alternative to utilize the small amount of labeled protein data. However, most of the existing work focuses on design-ing new architectures or pre-training tasks, and relatively little work has studied data augmentation for proteins. This paper extends data augmentation techniques previously used for images and texts to proteins and then benchmarks these techniques on a variety of protein-related tasks, providing the first comprehensive evaluation of protein augmentation. Furthermore, we propose two novel semantic-level protein augmentation methods, namely Integrated Gradients Substitution and Back Translation Substitution, which enable protein semantic-aware augmentation through saliency detection and biological knowledge. Finally, we integrate extended and proposed augmentations into an augmentation pool and propose a simple but effective framework, namely Automated Protein Augmentation (APA), which can adaptively select the most suitable augmentation combinations for different tasks. Extensive experiments have shown that APA enhances the performance of five protein related tasks by an average of 10.55% across three architectures compared to vanilla implementations without augmentation, highlighting its potential to make a great impact on the field.
format Preprint
id arxiv_https___arxiv_org_abs_2403_00875
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Enhancing Protein Predictive Models via Proteins Data Augmentation: A Benchmark and New Directions
Sun, Rui
Wu, Lirong
Lin, Haitao
Huang, Yufei
Li, Stan Z.
Quantitative Methods
Artificial Intelligence
Machine Learning
Biomolecules
Augmentation is an effective alternative to utilize the small amount of labeled protein data. However, most of the existing work focuses on design-ing new architectures or pre-training tasks, and relatively little work has studied data augmentation for proteins. This paper extends data augmentation techniques previously used for images and texts to proteins and then benchmarks these techniques on a variety of protein-related tasks, providing the first comprehensive evaluation of protein augmentation. Furthermore, we propose two novel semantic-level protein augmentation methods, namely Integrated Gradients Substitution and Back Translation Substitution, which enable protein semantic-aware augmentation through saliency detection and biological knowledge. Finally, we integrate extended and proposed augmentations into an augmentation pool and propose a simple but effective framework, namely Automated Protein Augmentation (APA), which can adaptively select the most suitable augmentation combinations for different tasks. Extensive experiments have shown that APA enhances the performance of five protein related tasks by an average of 10.55% across three architectures compared to vanilla implementations without augmentation, highlighting its potential to make a great impact on the field.
title Enhancing Protein Predictive Models via Proteins Data Augmentation: A Benchmark and New Directions
topic Quantitative Methods
Artificial Intelligence
Machine Learning
Biomolecules
url https://arxiv.org/abs/2403.00875