Open Access
Chemical Science, volume 12, issue 31, pages 10622-10633
ChemPix: automated recognition of hand-drawn hydrocarbon structures using deep learning
Hayley Weir
1, 2
,
Keiran C Thompson
1, 2
,
Amelia Woodward
1
,
Benjamin Choi
3
,
Augustin Braun
1
,
Todd J. Martinez
1, 2
Publication type: Journal Article
Publication date: 2021-07-03
Journal:
Chemical Science
scimago Q1
wos Q1
SJR: 2.333
CiteScore: 14.4
Impact factor: 7.6
ISSN: 20416520, 20416539
PubMed ID:
34447555
General Chemistry
Abstract
Inputting molecules into chemistry software, such as quantum chemistry packages, currently requires domain expertise, expensive software and/or cumbersome procedures. Leveraging recent breakthroughs in machine learning, we develop ChemPix: an offline, hand-drawn hydrocarbon structure recognition tool designed to remove these barriers. A neural image captioning approach consisting of a convolutional neural network (CNN) encoder and a long short-term memory (LSTM) decoder learned a mapping from photographs of hand-drawn hydrocarbon structures to machine-readable SMILES representations. We generated a large auxiliary training dataset, based on RDKit molecular images, by combining image augmentation, image degradation and background addition. Additionally, a small dataset of ∼600 hand-drawn hydrocarbon chemical structures was crowd-sourced using a phone web application. These datasets were used to train the image-to-SMILES neural network with the goal of maximizing the hand-drawn hydrocarbon recognition accuracy. By forming a committee of the trained neural networks where each network casts one vote for the predicted molecule, we achieved a nearly 10 percentage point improvement of the molecule recognition accuracy and were able to assign a confidence value for the prediction based on the number of agreeing votes. The ensemble model achieved an accuracy of 76% on hand-drawn hydrocarbons, increasing to 86% if the top 3 predictions were considered.
Nothing found, try to update filter.
Are you a researcher?
Create a profile to get free access to personal recommendations for colleagues and new articles.
Profiles