Ensemble machine learning for predicting the homogenized elastic properties of unidirectional composites: A SHAP-based interpretability analysis

Wenzhao Wang,Yunmei Zhao,Yan Li
DOI: https://doi.org/10.1007/s10409-023-23301-x
IF: 3.5
2024-02-22
Acta Mechanica Sinica
Abstract:This study aims to develop an interpretable ensemble machine learning (EML) method for predicting the homogenized elastic properties of unidirectional fiber-reinforced composite. Three machine learning models—Random Forest (RF), eXtreme Gradient Boosting Machine (XGBoost), and Light Gradient Boosting Machine (LGBM) are selected for constructing the EML model. The ground-truth dataset is created via knowledge-based finite element (FE) simulations with representative volume element (RVE) models of composites considering randomly dispersed fibers. We thoroughly evaluate the EML model's performance using the metrics of accuracy, efficiency, interpretability, and generalization. The obtained results indicate that: (1) the EML model outperforms the base ML model with high precision ( R 2 = 0.962, MSE = 5.41); (2) the SHapley Additive explanations (SHAP) based on cooperative game theory is used to interpret the predictions, with the global interpretations identifying fiber volume content as the most influential variable on the composite and the local interpretations examining the key influencing mechanisms of each feature; (3) the EML model demonstrates good generalization ability on experimental data, and it can accurately forecast the homogenized properties of the composite with genuine fibers.
engineering, mechanical,mechanics
What problem does this paper attempt to address?