Using Active Learning to Develop Machine Learning Models for Reaction Yield Prediction

Johansson, Simon,1994Chalmers tekniska högskola,Chalmers University of Technology,AstraZeneca AB (author)

Computer aided synthesis planning, suggesting synthetic routes for molecules of interest, is a rapidly growing field. The machine learning methods used are often dependent on access to large datasets for training, but finite experimental budgets limit how much data can be obtained from experiments. This suggests the use of schemes for data collection such as active learning, which identifies the data points of highest impact for model accuracy, and which has been used in recent studies with success. However, little has been done to explore the robustness of the methods predicting reaction yield when used together with active learning to reduce the amount of experimental data needed for training. This study aims to investigate the influence of machine learning algorithms and the number of initial data points on reaction yield prediction for two public high-throughput experimentation datasets. Our results show that active learning based on output margin reached a pre-defined AUROC faster than random sampling on both datasets. Analysis of feature importance of the trained machine learning models suggests active learning had a larger influence on the model accuracy when only a few features were important for the model prediction.

Gummesson Svensson, Hampus,1996AstraZeneca AB,Chalmers tekniska högskola,Chalmers University of Technology(Swepub:cth)hamsven (author)
Bjerrum, E.AstraZeneca AB (author)
Schliep, Alexander,1967Gothenburg University,Göteborgs universitet,Institutionen för data- och informationsteknik (GU),Department of Computer Science and Engineering (GU),University of Gothenburg(Swepub:cth)schliep (author)
Haghir Chehreghani, Morteza,1982Chalmers tekniska högskola,Chalmers University of Technology(Swepub:cth)haghir (author)
Tyrchan, C.AstraZeneca AB (author)
Engkvist, Ola,1967Chalmers tekniska högskola,Chalmers University of Technology,AstraZeneca AB(Swepub:cth)olae (author)
Chalmers tekniska högskolaAstraZeneca AB (creator_code:org_t)

By the author/editor: Johansson, Simon ...; Gummesson Svenss ...; Bjerrum, E.; Schliep, Alexand ...; Haghir Chehregha ...; Tyrchan, C.; show more...; Engkvist, Ola, 1 ...; show less...

About the subject

NATURAL SCIENCES: NATURAL SCIENCES; and Computer and Inf ...; and Language Technol ...

NATURAL SCIENCES: NATURAL SCIENCES; and Computer and Inf ...; and Bioinformatics

NATURAL SCIENCES: NATURAL SCIENCES; and Biological Scien ...; and Bioinformatics a ...

By the university: University of Gothenburg; Chalmers University of Technology

Kungliga biblioteket hanterar dina personuppgifter i enlighet med EU:s dataskyddsförordning (2018), GDPR. Läs mer om hur det funkar här.
Så här hanterar KB dina uppgifter vid användning av denna tjänst.