Predicting the Properties of Metamaterials Consisting of Curved-Wall Triangles Using Ensemble Neural Networks with Interpretability

Shaowei Zhu,Mingdu Wen,Zhuo Lv,Liming Chen,Tao Liu,Xianbo Hou
DOI: https://doi.org/10.1016/j.engappai.2024.109408
2024-01-01
Abstract:Machine learning has emerged as a promising tool for predicting the properties of metamaterials, owing to its substantially faster prediction speed compared to conventional methods. However, the lack of interpretability of black-box machine learning models has impeded their adoption in this field. In this work, we develop an ensemble of neural networks (NNE) for predicting the mechanical properties of metamaterials consisting of curved-wall triangles. Our method achieves high prediction accuracy for the effective Young's modulus and Poisson's ratio for most samples. The average relative errors of the NNE are less than 5% for predicting the dimensionless Young's modulus and less than 10% for predicting Poisson's ratios. Additionally, our method is orders of magnitude faster than finite element simulations. More importantly, by explaining the predictions of NNE using SHapley Additive exPlanations (SHAP), we are able to quantify the relationships between geometric features and metamaterial properties. This reveals insights into parameter couplings and nonlinear effects. New physical insights can thus be obtained and new design rules discovered through this machine learning interpretability approach. Overall, our interpretable machine learning model can accelerate the design and discovery of metamaterials while providing physical insights into the inner workings of the models.
What problem does this paper attempt to address?