A New Multitask Learning Method for Tool Wear Condition and Part Surface Quality Prediction

Abstract

Deep learning has been gradually used in the field of machining condition monitoring. However, at present only single-task prediction can be performed, which results in increased experimental costs, wasted datasets, and repetitive work. In this article, a new multitask learning method based on a deep belief network (DBN) is proposed, which can be used to predict the tool wear condition and part surface quality. The single-task data transmission of the last few hidden layers of the DBN network is improved to multitask parallel data transmission so that the improved DBN can realize multitask learning. The loss function of the multitask learning model is defined as the weighted sum of all single-task loss functions. According to the loss of different tasks in the iteration process, the weight of corresponding tasks can be adjusted automatically. Furthermore, the multitask deep learning method can realize information sharing, suppress overfitting, improve prediction accuracy, and require less computing time. Combined with the abovementioned improvements, a multitask model for tool wear and part surface quality was developed. Experimental verification was performed on a KVC850M three-axis vertical machining center. The results show that the accuracy of the proposed multitask prediction model is 99% for the tool wear prediction and 92.86% for part surface quality prediction.