A two-layer stacking regression model for predicting yarn quality of small samples

Abstract

In addressing the prevalent challenges associated with the limited generalization ability and weak adaptability to small samples observed in neural network models in contemporary yarn quality prediction research, this study proposes a two-layer stacking regression model. The initial layer of the stacking model integrates three base models: random forest, gradient boosting decision tree, and adaptive boosting. The secondary layer employs the linear regression algorithm. The efficacy of the stacking model was juxtaposed against the performance assessment of a multilayer perceptron neural network model, in addition to evaluations of the single random forest, gradient boosting decision tree, and adaptive boosting models, respectively. The generalization ability of the models was assessed through experiments that randomly partitioned the sample dataset based on 10 distinct random seeds. The adaptability to small samples of the models was evaluated through experiments spanning nine varying training set sample sizes. The experiment results underscore that the stacking model outperforms in aspects of prediction accuracy, generalization ability, and adaptability to small samples.