CoSpred: Machine Learning Workflow to Predict Tandem Mass Spectrum in Proteomics

In mass spectrometry-based proteomics, the use of deep learning algorithms can help improve the identification rates of peptides and proteins through the generation of high-fidelity theoretical spectrum which can be used as the basis of a more complete spectral library than those presently available, especially for unobserved protein/genetic variants. Here we focus on providing an end-to-end user-friendly machine learning workflow, which we call Complete Spectrum Predictor (CoSpred). Using CoSpred users can create their own machine learning compatible training dataset and then train a machine learning model to predict both backbone and non-backbone ions. For the model a transformer encoder architecture is used to predict the complete MS/MS spectrum from a given peptide sequence. In addition to the transformer model provided in the package, the code is built modularly to allow for alternate ML models to be easily "plugged in," allowing for spectrum prediction optimization given different experimental conditions. The CoSpred workflow (preprocessing→training→inference) provides a path for state-of-art ML capabilities to be more accessible to proteomics scientists.