Ensemble learning-based strength prediction and model interpretability analysis of engineered cementitious composites
Yufei Wang , Junbo Sun , Xianda Liu , Yimeng Huang , Xiangyu Wang , Li Zuo , Dong Wang
AI in Civil Engineering ›› 2026, Vol. 5 ›› Issue (1) : 7
Accurate prediction of compressive strength is essential for improving the performance and durability of Engineered Cementitious Composites (ECC) in construction applications. Traditional methods often fall short in accounting for the complex interactions between material properties, such as fiber type, matrix composition, and curing conditions. To address this challenge, this study presents an advanced ensemble learning framework based on a dataset of 313 ECC samples characterized by 18 key features. The ensemble model integrates three base learners, namely Random Forest (RF), Gradient Boosting Decision Trees (GBDT), and Support Vector Regression (SVR), along with a meta-learner selected from ten candidate models. The proposed ensemble model demonstrates significantly higher prediction accuracy compared to conventional approaches. The results show that the ensemble model achieves a coefficient of determination (R2) of 0.896, a root mean square error (RMSE) of 5.734, and a mean absolute error (MAE) of 4.505, substantially outperforming individual models. Among the evaluated meta-learners, Lasso Regression was identified as the optimal choice. Its regularization capability effectively mitigated overfitting and enhanced generalization, leading to a notable improvement in the final predictive performance of the stacking framework. Furthermore, SHapley Additive exPlanations (SHAP) and Partial Dependence Plots (PDP) were employed for model interpretability and visualization. The analysis reveals that factors such as fiber elastic modulus, silica fume content, and fiber volume fraction significantly contribute to the enhancement of ECC compressive strength. This model provides practical insights for optimizing the design and application of ECC materials.
Engineered cementitious composites / Strength prediction / Ensemble learning / Model interpretability
| [1] |
|
| [2] |
|
| [3] |
|
| [4] |
|
| [5] |
|
| [6] |
|
| [7] |
|
| [8] |
Chen, T., & Guestrin, C. (2016). Xgboost: A scalable tree boosting system. Paper presented at the Proceedings of the 22nd acm sigkdd international conference on knowledge discovery and data mining |
| [9] |
|
| [10] |
|
| [11] |
|
| [12] |
|
| [13] |
|
| [14] |
|
| [15] |
|
| [16] |
|
| [17] |
|
| [18] |
|
| [19] |
|
| [20] |
|
| [21] |
Huang, D., Hu, D., He, J., & Xiong, Y. (2018). Structure damage detection based on ensemble learning. Paper presented at the 2018 9th International Conference on Mechanical and Aerospace Engineering (ICMAE) |
| [22] |
|
| [23] |
|
| [24] |
|
| [25] |
|
| [26] |
|
| [27] |
|
| [28] |
|
| [29] |
|
| [30] |
|
| [31] |
|
| [32] |
|
| [33] |
|
| [34] |
|
| [35] |
|
| [36] |
|
| [37] |
|
| [38] |
|
| [39] |
|
| [40] |
|
| [41] |
|
| [42] |
|
| [43] |
|
| [44] |
|
| [45] |
|
| [46] |
|
| [47] |
|
| [48] |
|
| [49] |
|
| [50] |
|
| [51] |
|
| [52] |
|
| [53] |
|
| [54] |
|
| [55] |
|
| [56] |
|
| [57] |
|
| [58] |
|
| [59] |
|
| [60] |
|
| [61] |
|
| [62] |
|
| [63] |
|
| [64] |
|
| [65] |
|
| [66] |
|
| [67] |
|
| [68] |
|
| [69] |
|
| [70] |
|
| [71] |
|
| [72] |
|
| [73] |
|
| [74] |
|
| [75] |
|
| [76] |
|
| [77] |
|
| [78] |
|
| [79] |
|
| [80] |
|
| [81] |
|
| [82] |
|
| [83] |
|
| [84] |
|
| [85] |
|
| [86] |
|
The Author(s)
/
| 〈 |
|
〉 |