Search papers, labs, and topics across Lattice.
This paper benchmarks five message-passing neural network (MPNN) architectures—SchNet, FieldSchNet, SO3Net, PaiNN, and MACE—for predicting infrared spectra of small organic molecules. The study evaluates the models' computational efficiency, accuracy in predicting energies, forces, and dipole moments, and robustness in capturing harmonic frequencies and infrared spectra. Results show that while all models perform well on training data, SO3Net, PaiNN, and MACE demonstrate superior generalization to unseen systems, with MACE achieving the highest spectral accuracy and transferability.
Equivariant message-passing neural networks like MACE can predict infrared spectra with higher accuracy and transferability than invariant models, offering a path to more reliable and generalizable atomistic simulations.
Machine learning has transformed the field of atomistic simulations by enabling the development of interatomic potentials that are computationally efficient and highly accurate. These advances have opened the door to modeling molecular vibrations and predicting infrared spectra with near ab-initio accuracy at a fraction of the computational cost. Among these approaches, message-passing neural networks (MPNNs) have emerged as a particularly powerful class of models for representing complex atomic interactions. In this study, we benchmark five MPNN architectures, SchNet, FieldSchNet, SO3Net, PaiNN, and MACE, for predicting infrared spectra of small organic molecules. SchNet and FieldSchNet are invariant models, while SO3Net, PaiNN, and MACE are equivariant, explicitly accounting for rotational symmetries in molecular representations. We evaluate their performance in terms of computational efficiency, accuracy, and robustness. All models accurately predict properties, such as energies, forces, and dipole moments, required for infrared spectra calculations. They also capture harmonic frequencies and infrared spectra derived from molecular dynamics with high fidelity for molecules in the training set. However, SchNet and FieldSchNet show limited transferability to unseen systems, while SO3Net, PaiNN, and MACE generalize more effectively. In terms of computational efficiency, SchNet is the most efficient and FieldSchNet enables field-dependent response modeling but with higher cost. PaiNN achieves the best balance between accuracy and efficiency, MACE provides the highest spectral accuracy and transferability, and SO3Net performs between PaiNN and MACE.