Explainable Data-Driven Ensemble Learning Models for the Mechanical Properties Prediction of Concrete Confined by Aramid Fiber-Reinforced Polymer Wraps Using Generative Adversarial Networks

Abstract

The current study offers a data-driven methodology to predict the ultimate strain and compressive strength of concrete reinforced by aramid FRP wraps. An experimental database was collected from the literature, on which seven different machine learning (ML) models were trained. The diameter and length of the cylindrical specimens, the compressive strength of unconfined concrete, the thickness, elasticity modulus and ultimate tensile strength of the FRP wrap were used as the input features of the machine learning models, to predict the ultimate strength and strain of the specimens. The experimental dataset was further enhanced with synthetic data using the tabular generative adversarial network (TGAN) approach. The machine learning models’ performances were compared to the predictions of the existing strain capacity and compressive strength prediction equations for aramid FRP-confined concrete. The accuracy of the predictive models was measured using state-of-the-art statistical metrics such as the coefficient of determination, mean absolute error and root mean squared error. On average, the machine learning models were found to perform better than the available equations in the literature. In particular, the extra trees regressor, XGBoost and K-nearest neighbors algorithms performed significantly better than the remaining algorithms, with R2 scores greater than 0.98. Furthermore, the SHapley Additive exPlanations (SHAP) method and individual conditional expectation (ICE) plots were used to visualize the effects of various input parameters on the predicted ultimate strain and strength values. The unconfined compressive strength of concrete and the ultimate tensile strength of the FRP wrap were found to have the greatest impact on the machine learning model outputs.