Structure to Property: Chemical Element Embeddings and a Deep Learning Approach for Accurate Prediction of Chemical Properties
Abstract
We introduce the elEmBERT model for chemical classification tasks. It is based on deep learning techniques, such as a multilayer encoder architecture. We demonstrate the opportunities offered by our approach on sets of organic, inorganic and crystalline compounds. In particular, we developed and tested the model using the Matbench and Moleculenet benchmarks, which include crystal properties and drug design-related benchmarks. We also conduct an analysis of vector representations of chemical compounds, shedding light on the underlying patterns in structural data. Our model exhibits exceptional predictive capabilities and proves universally applicable to molecular and material datasets. For instance, on the Tox21 dataset, we achieved an average precision of 96%, surpassing the previously best result by 10%.
- Publication:
-
arXiv e-prints
- Pub Date:
- September 2023
- DOI:
- 10.48550/arXiv.2309.09355
- arXiv:
- arXiv:2309.09355
- Bibcode:
- 2023arXiv230909355S
- Keywords:
-
- Physics - Chemical Physics;
- Condensed Matter - Materials Science;
- Computer Science - Machine Learning;
- Physics - Atomic and Molecular Clusters;
- Quantitative Biology - Quantitative Methods
- E-Print:
- 26 pages, 16 figures, 6 tables, 50 references