G. Lambard, E. Gracheva
National Institute for Materials Science,
Japan
Keywords: SMILES-X, autonomous molecular compounds, characterization, small datasets with descriptors
Summary:
Background: There is more and more evidence that machine learning can be successfully applied in materials science and related fields. However, datasets in these fields are often quite small (from tens to several thousands of samples). It makes the most advanced machine learning techniques remain neglected, as they are considered to be applicable to big data only. Moreover, materials informatics methods often rely on human-engineered descriptors, that should be carefully chosen, or even created, to fit the physicochemical property that one intends to predict. Methods: We propose a new method that tackles both the issue of small datasets and the difficulty of task-specific descriptors development. The SMILES-X is an autonomous pipeline for molecular compounds characterisation based on a {Embed-Encode-Attend-Predict} neural architecture with a data-specific Bayesian hyper-parameters optimisation. The only input to the architecture — the SMILES strings — are decanonicalised in order to efficiently augment the data. One of the key features of the architecture is the attention mechanism, which enables the interpretation of output predictions without extra computational cost. Results: The SMILES-X achieves state-of-the-art results in the inference of aqueous solubility (RMSEtest ≃ 0.57 ± 0.07 mols/L), hydration free energy (RMSEtest ≃ 0.81 ± 0.22 kcal/mol, which is ∼ 24.5% better than molecular dynamics simulations), and octanol/water distribution coefficient (RMSEtest ≃ 0.59 ± 0.02 for LogD at pH 7.4) of molecular compounds. Conclusions: The SMILES-X is intended to become an important asset in the toolkit of materials scientists and chemists. The source code for the SMILES-X is available at github.com/GLambard/SMILES-X.