STMHCpan, an accurate Star-Transformer-based extensible framework for predicting MHC I allele binding peptides

Peptide-major histocompatibility complex I (MHC I) binding affinity prediction is crucial for vaccine development, but existing methods face limitations such as small datasets, model overfitting due to excessive parameters and suboptimal performance. Here, we present STMHCPan (STAR-MHCPan), an open-source package based on the Star-Transformer model, for MHC I binding peptide prediction. Our approach introduces an attention mechanism to improve the deep learning network architecture and performance in antigen prediction. Compared with classical deep learning algorithms, STMHCPan exhibits improved performance with fewer parameters in receptor affinity training. Furthermore, STMHCPan outperforms existing ligand benchmark datasets identified by mass spectrometry. It can also handle peptides of arbitrary length and is highly scalable for predicting T-cell responses. Our software is freely available for use, training and extension through Github (https://github.com/Luckysoutheast/STMHCPan.git).