Machine Learning Approach for Describing Water OH Stretch Vibrations

Abstract

A machine learning approach employing neural networks is developed to calculate the vibrational frequency shifts and transition dipole moments of the symmetric and antisymmetric OH stretch vibrations of a water molecule surrounded by water molecules. We employed the atom-centered symmetry functions (ACSFs), polynomial functions, and Gaussian-type orbital-based density vectors as descriptor functions and compared their performances in predicting vibrational frequency shifts using the trained neural networks. The ACSFs perform best in modeling the frequency shifts of the OH stretch vibration of water among the types of descriptor functions considered in this paper. However, the differences in performance among these three descriptors are not significant. We also tried a feature selection method called CUR matrix decomposition to assess the importance and leverage of the individual functions in the set of selected descriptor functions. We found that a significant number of those functions included in the set of descriptor functions give redundant information in describing the configuration of the water system. We here show that the predicted vibrational frequency shifts by trained neural networks successfully describe the solvent-solute interaction-induced fluctuations of OH stretch frequencies.