EAGP: an efficient generative augmentation framework for phage protein classification under severe class imbalance

The accurate classification of phage proteins is critical for advancing bacteriophage research. Despite the proliferation of machine learning approaches in this domain, the persistent issue of data imbalance continues to hinder performance, particularly for rare protein sequences. Previous attempts to address this by re-weighting minority classes have faced limitations due to insufficient feature extraction capabilities.

In this paper, we introduce EAGP, a novel approach that integrates a generative model-functionally equivalent to a WGAN yet tailored for one-dimensional data-with the Evolutionary Scale Modeling (ESM) protein large language model for robust feature extraction. EAGP exhibits exceptional performance in binary classification and protein function annotation tasks. Crucially, our method not only improves overall classification efficacy but also significantly alleviates the performance degradation typically observed in minority classes.

The data and code underlying this article are available in GitHub at https://github.com/Innerly/EAGP and have been archived on Zenodo at https://doi.org/10.5281/zenodo.19928069.