Transformer-Based Models for Predicting Molecular Structures from Infrared Spectra Using Patch-Based Self-Attention

Infrared (IR) spectroscopy, a type of vibrational spectroscopy, provides extensive molecular structure details and is a highly effective technique for chemists to determine molecular structures. However, analyzing experimental spectra has always been challenging due to the specialized knowledge required and the variability of spectra under different experimental conditions. Here, we propose a transformer-based model with a patch-based self-attention spectrum embedding layer, designed to prevent the loss of spectral information while maintaining simplicity and effectiveness. To further enhance the model's understanding of IR spectra, we introduce a data augmentation approach, which selectively introduces vertical noise only at absorption peaks. Our approach not only achieves state-of-the-art performance on simulated data sets but also attains a top-1 accuracy of 55% on real experimental spectra, surpassing the previous state-of-the-art by approximately 10%. Additionally, our model demonstrates proficiency in analyzing intricate and variable fingerprint regions, effectively extracting critical structural information.