Virtual data augmentation method for reaction prediction in small dataset scenario

09 February 2022, Version 1
This content is a preprint and has not undergone peer review at the time of posting.


To improve the performance of data-driven reaction prediction models, a new data augmentation method for augmenting data volumes is presented that aims to add fake data in training dataset. This method is only pay attention to small-scale reactions, and manually generates fake data which is chemical and credible molecules by replacing functional groups in reaction sites. And we call this method as virtual data augmentation. Additionally, the transformer model is introduced to explore the effectiveness of virtual data augmentation method in the task of reaction prediction based on small data sets. We apply our method to five classic coupling reactions, the results show that the overall performance of the transformer-baseline model and transformer-transfer learning model combined with virtual data augmentation method is obviously improved, compared to raw datasets. Especially for Suzuki reaction, combining transfer learning strategy and virtual data augmentation method, reaches top-1 accuracy of 97.8%. To sum up, virtual data augmentation can be used as a measure to face up the problem of insufficient data and significantly improves the performance of reaction prediction.


deep learning
reaction prediction
virtual data augmentation


Comments are not moderated before they are posted, but they can be removed by the site moderators if they are found to be in contravention of our Commenting Policy [opens in a new tab] - please read this policy before you post. Comments should be used for scholarly discussion of the content in question. You can find more information about how to use the commenting feature here [opens in a new tab] .
This site is protected by reCAPTCHA and the Google Privacy Policy [opens in a new tab] and Terms of Service [opens in a new tab] apply.