ChemTorch: A Deep Learning Framework for Benchmarking and Developing Chemical Reaction Property Prediction Models

Read full text
Authors: Jasper De Landsheere, Anton Zamyatin, Johannes Karwounopoulos, Esther Heid
Cited by: 2
References: 53

Abstract

Modeling of chemical reactions is essential for understanding kinetic mechanisms and predicting possible outcomes of reacting systems. Quantum mechanical calculations are accurate but often prohibitively expensive. Deep learning has emerged as a faster alternative, but progress is slowed by a fragmented software ecosystem that hinders reuse, fair comparison, and reproducibility. We present ChemTorch, an open-source framework that streamlines model development, experimentation, hyperparameter tuning, and benchmarking through modular pipelines, standardized configuration, and built-in data splitters for in- and out-of-distribution evaluation. We envision ChemTorch as a foundation for community-driven method development and reproducible benchmarking in chemical reaction modeling. As a first step toward unified benchmarks, we compare four representative modalities for barrier-height prediction on the RDB7 data set, including fingerprint-, sequence-, graph-, and 3D-based approaches. Our results highlight clear advantages of structurally informed models and sharp performance drops under out-of-distribution conditions, highlighting the importance of rigorous benchmarking.

Fields of Study & Subjects

Machine Learning in Materials Science Computational Drug Discovery Methods Asymmetric Hydrogenation and Catalysis

Publication details

Journal / source
Journal of Chemical Information and Modeling
Publisher
American Chemical Society (ACS)
Volume
66
Issue
5
Pages
2434–2442
Publication date
24 February 2026
Language
EN