Machine learning models with different cheminformatics data sets to forecast the power conversion efficiency of organic solar cells

Random Forest (RF) and Gradient Boosting Regression Trees (GBRT) regression models along with three cheminformatics data sets (RDkit, Mordred, Morgan) have been used to predict the power conversion efficiency (PCE) of organic solar cells (OSCs). The data consists of cheinformatics descriptors of the electron donor used in 433 OSCs for which the experimental PCE (target variable) is reported in the literature. The donor is either a polymer or a small organic molecule, and the acceptor the fullerene derivatives PCBM or PC71BM. Unlike previous methods, our ML approach considers the type of donor and the acceptor by adding four extra donor's features using the one-hot encoder tool. It is demonstrated that this additional information improves the prediction performance up to 34%. We have also exploited this feature to theoretically forecast the PCE of new OSCs by evaluating the ML model for a different acceptor. It is predicted that more than 50% of the OCSs obtained by exchanging the acceptor would have higher experimental PCE. The prediction accuracy of a given ML approach is analyzed for different PCE intervals. RF using RDkit descriptors resulted in the best ML approach with a Pearson's correlation coefficient for the training and testing sets equal to 0.96 and 0.62, respectively.
Comment: 12 pages, 4 Figures, 1 Table