Comparative study of various molecular feature representations for solvation free energy predictions of neutral species

Predicting molecular properties with the help of Neural Networks is a common way to substitute or enhance comprehensive quantum-chemical calculations. One of the problems facing researchers is the choice of vectorization approach to representing the solvent and the solute for the estimator model. In this work, 10 different approaches have been investigated for both organic solute and solvent including vectorizers that relied on macroscopic parameters, functional groups classification, molecular graphs, or atomic coordinates. A variation of the Bag of Bonds approach called JustBonds, trained on the MNSol database, showed the best overall performance resulting in RMSD <2 kcal/mol for the blind dataset that contains the solutes not presented in the training subset and <1 kcal/mol on records from Solv@TUM database, which is close to contemporary continuum models. We have also demonstrated that the most important bags usually contain heteroatom and play a key role in the solvation process. Furthermore, the small role of solvent vectorization was demonstrated and revealed that approaches based on functional groups or macroscopic solvent parameters are often enough to efficiently represent solvent media.