Multi-scale Sinusoidal Embeddings Enable Learning on High Resolution Mass Spectrometry Data

Gennady Voronov,Rose Lightheart,Joe Davison,Christoph A. Krettler,David Healey,Thomas Butler
DOI: https://doi.org/10.48550/arXiv.2207.02980
2023-05-05
Abstract:Small molecules in biological samples are studied to provide information about disease states, environmental toxins, natural product drug discovery, and many other applications. The primary window into the composition of small molecule mixtures is tandem mass spectrometry (MS2), which produces data that are of high sensitivity and part per million resolution. We adopt multi-scale sinusoidal embeddings of the mass data in MS2 designed to meet the challenge of learning from the full resolution of MS2 data. Using these embeddings, we provide a new state of the art model for spectral library search, the standard task for initial evaluation of MS2 data. We also introduce a new task, chemical property prediction from MS2 data, that has natural applications in high-throughput MS2 experiments and show that an average $R^2$ of 80\% for novel compounds can be achieved across 10 chemical properties prioritized by medicinal chemists. We use dimensionality reduction techniques and experiments with different floating point resolutions to show the essential role multi-scale sinusoidal embeddings play in learning from MS2 data.
Machine Learning,Quantitative Methods
What problem does this paper attempt to address?
The main problem that this paper attempts to solve is how to effectively extract and utilize information from high - resolution mass spectrometry data to improve the identification and prediction of small - molecule structures and properties. Specifically, the authors proposed a method of using multi - scale sinusoidal embeddings to process the m/z values in mass spectrometry data. This method can preserve the full - resolution information of the data, thereby achieving performance improvements on multiple tasks. These tasks include: 1. **Spectral Library Search**: This is a standard task in mass spectrometry data analysis and is used for the preliminary evaluation of MS2 data. By comparing the spectra of unknown compounds with those in the database, matches are found. The method proposed by the authors has reached a new technological level in this task. It can not only find completely correct compounds but also find structurally similar compounds, which is especially useful for compounds not included in the database. 2. **Chemical Property Prediction**: Predicting the chemical properties of compounds from MS2 data, which has a natural application in high - throughput mass spectrometry experiments. The authors showed that for 10 chemical properties prioritized by medicinal chemists, a prediction performance with an average \( R^2 \) of 80% can be achieved, which is of great significance for the preliminary screening of new compounds and the selection of candidate drug molecules. To verify the effectiveness of multi - scale sinusoidal embeddings, the authors also carried out the following experiments: - **The influence of different - precision inputs**: By converting the input m/z values into half - precision, single - precision, and double - precision floating - point numbers, the influence of input precision on the model performance was studied. The results showed that the performance of the sinusoidal embedding model was the best under double - precision input, while the performance under half - precision input decreased significantly, indicating that sinusoidal embeddings can learn more useful information from high - resolution data. - **Visual analysis of embeddings**: Using the UMAP dimensionality - reduction technique to visualize the embeddings generated by different methods, the results showed that high - precision sinusoidal embeddings can better capture the information of the decimal part of m/z while maintaining the m/z distance, which is difficult to achieve in other methods. In conclusion, this paper has solved the challenge of effective learning from high - resolution mass spectrometry data by introducing multi - scale sinusoidal embeddings, providing new technical support for multiple applications in the fields of metabolomics and drug discovery.