An adversarial bidirectional serial-parallel LSTM-based QTD framework for product quality prediction

摘要

In order to capture temporal interactions among processes in manufacturing and assembly processes, an end-to-end unified product quality prediction framework called QTD is proposed in this paper. It consists of three modules: quality embedding model pool, temporal-interactive model, and decoding model. Besides, to handle the information transfer and integration problems in the time direction of parallel processes, a novel bidirectional serial-parallel LSTM (Bi-SP-LSTM) is devised as an instantiated model of temporal-interactive model. Bi-SP-LSTM is an extension of bidirectional long short-term memory. Moreover, an unsupervised task and a loss function named adversarial focal loss have been designed to give the framework the ability to assess heteroscedastic uncertainty in classification task due to intrinsic uncertainty in data. Furthermore, experiments are devised based on a subset of a public dataset from Kaggle competition to demonstrate the validity of the proposed framework. Compared with other latest methods, the proposed framework is verified to be more accurate and robust. Taking Matthews correlation coefficient as an example, the adversarial Bi-SP-LSTM-based QTD framework is superior to the best existing methods with 95% confidence interval in most cases, and its mean MCC is 4.88% higher than the best existing method. The results suggest that the proposed framework has a broad application prospect for quality prediction in manufacturing and assembly processes.