LGB-Stack: Stacked Generalization with LightGBM for Highly Accurate Predictions of Polymer Bandgap

Abstract

Recently, the Ramprasad group reported a quantitative structure–property relationship (QSPR) model for predicting the E gap values of 4209 polymers, which yielded a test set R 2 score of 0.90 and a test set root-mean-square error (RMSE) score of 0.44 at a train/test split ratio of 80/20. In this paper, we present a new QSPR model named LGB-Stack , which performs a two-level stacked generalization using the light gradient boosting machine. At level 1, multiple weak models are trained, and at level 2, they are combined into a strong final model. Four molecular fingerprints were generated from the simplified molecular input line entry system notations of the polymers. They were trimmed using recursive feature elimination and used as the initial input features for training the weak models. The output predictions of the weak models were used as the new input features for training the final model, which completes the LGB-Stack model training process. Our results show that the best test set R 2 and the RMSE scores of LGB-Stack at the train/test split ratio of 80/20 were 0.92 and 0.41, respectively. The accuracy scores further improved to 0.94 and 0.34, respectively, when the train/test split ratio of 95/5 was used.

Artificial Intelligence for Conjugated Polymers

“…Several examples of these algorithms are shown in Figure 3. (1) Regression algorithms, such as linear regression and support vector regression, are adept at predicting continuous property values such as the band gaps 30 and power conversion efficiency (PCE) of conjugated polymers. The flexibility and interpretability of these algorithms contribute to their widespread adoption.…”

Data-Driven Method for Vacuum Prediction in the Underwater Pump of a Cutter Suction Dredger

“…Then, all these base learners output the original data, and the outputs of these models are stacked in columns to form (m, p)-dimensional new data, with m representing the number of samples and p representing the number of base learners. Finally, the new sample data are given to the second layer model for fitting [23][24][25]. Here is an example of the training process for stacked models.…”

Calibration of CAMS PM2.5 data over Hungary: a machine learning approach

“…LightGBM algorithms help advance cutting-edge technology by boosting our understanding of complicated data patterns and decision-making processes across numerous industries [41][42][43][44]. It was created by merging two unique data sampling and classification algorithms (EFB (Exclusive Feature Bundling) and GOSS (Gradientbased One Side Sampling)) [39].…”