Integrating Wide and Deep Neural Networks with Squeeze-and-Excitation Blocks for Multi-Target Property Prediction in Additively Manufactured Fiber Reinforced Composites
Behzad Parvaresh, Rahmat K. Adesunkanmi, Adel Alaeddini
TL;DR
This work tackles the challenge of predicting multiple mechanical and manufacturing properties for CFRC-AM under a large, mixed-parameter design space. It introduces a data-efficient SE-WDNN that fuses a wide and deep architecture with squeeze-and-excitation blocks and uses LHS-guided experimentation to build an eight-target predictive model, achieving a best overall MAPE of $MAPE = 12.33\%$. SHAP analysis reveals that reinforcement strategy (e.g., number of reinforced layers and fiber type) is the dominant factor shaping mechanical performance, guiding design decisions towards fiber-driven optimization. The combination of LHS sampling, SE-WDNN modeling, and interpretability enables efficient, interpretable multi-target predictions that can steer parameter selection and design exploration in CFRC-AM, while highlighting areas for future validation and integration with optimization workflows.
Abstract
Continuous fiber-reinforced composite manufactured by additive manufacturing (CFRC-AM) offers opportunities for printing lightweight materials with high specific strength. However, their performance is sensitive to the interaction of process and material parameters, making exhaustive experimental testing impractical. In this study, we introduce a data-efficient, multi-input, multi-target learning approach that integrates Latin Hypercube Sampling (LHS)-guided experimentation with a squeeze-and-excitation wide and deep neural network (SE-WDNN) to jointly predict multiple mechanical and manufacturing properties of CFRC-AMs based on different manufacturing parameters. We printed and tested 155 specimens selected from a design space of 4,320 combinations using a Markforged Mark Two 3D printer. The processed data formed the input-output set for our proposed model. We compared the results with those from commonly used machine learning models, including feedforward neural networks, Kolmogorov-Arnold networks, XGBoost, CatBoost, and random forests. Our model achieved the lowest overall test error (MAPE = 12.33%) and showed statistically significant improvements over the baseline wide and deep neural network for several target variables (paired t-tests, p <= 0.05). SHapley Additive exPlanations (SHAP) analysis revealed that reinforcement strategy was the major influence on mechanical performance. Overall, this study demonstrates that the integration of LHS and SE-WDNN enables interpretable and sample-efficient multi-target predictions, guiding parameter selection in CFRC-AM with a balance between mechanical behavior and manufacturing metrics.
