Equivariant neural networks improve vibrational spectrum prediction accuracy

Response-state Learning for Transferable Vibrational Spectroscopic Characterization with Electron Prior

Machine Learning

Summary

Predicting how molecules vibrate and interact with light is hard when small changes in their environment affect key features. The authors created a special kind of neural network called an Equivariant Neural Kalman Network that learns from molecules’ shapes and electronic information to predict vibrational spectra more accurately and reliably. Their method works well on simple molecules and more complex ones like drugs and biomolecules, showing it can handle a wide variety of chemical systems. This approach also helps diagnose when the predictions might be uncertain and calibrates them to stay consistent with known physics.

What this means in practice

  • For computational chemists: Generate more accurate vibrational spectra predictions for molecules across diverse chemical spaces including drug-like and biomolecular systems.
  • For materials scientists: Assess vibrational features of functional molecular materials with improved reliability and physics-informed calibration.

Authors

Zetong Li, Zhuosong Xie, Hengyu Fan, Jiaao Yu, Qiyao Hua, Zheng Lu, Liming Xu, Juanni Wu, Honglin Li

Abstract

Vibrational spectral prediction can become inaccurate when localized stereoelectronic environments perturb intermediate response states and high-risk response units dominate characteristic spectral fingerprints, making prediction across external chemical space difficult. SO(3) Equivariant Neural Kalman Networks (SENK) form a response-state cascade that combines an equivariant transformer backbone for Hessian, dipole-derivative and polarizability-derivative learning, an Equivariant Neural Kalman bridge for state-dependent refinement and reliability sensing, and an NBO-informed electronic-prior pathway coupling consistency regularization with bounded, branch-specific guided spectral calibration. SENK outperforms DetaNet on QM9S and QMe14S while preserving full-spectrum IR and Raman fidelity from small molecules to drug-like systems. SENK remains stable and selectively improves spectrally sensitive features in biomolecular systems with complex stereoelectronic effects. It therefore integrates tensor prediction, reliability diagnosis and physics-informed calibration, supporting transferable vibrational spectroscopy from molecular systems to functional molecular materials.