Hybrid Data Augmentation Combining Screening Based MCGAN and Manual Transformation for Few-Shot Tool Wear State Recognition

• Published in: IEEE sensors journal Vol. 24; no. 8; p. 1
• Main Authors: Quan, Yu, Liu, Changfu, Yuan, Zhuang, Yan, Boling
• Format: Journal Article
• Language: English
• Published: New York IEEE 15.04.2024; The Institute of Electrical and Electronics Engineers, Inc. (IEEE)
• Subjects: Artificial neural networks; Data augmentation; Data models; Datasets; Deep Learning; Fault diagnosis; Feature extraction; Few-Shot; Generative adversarial networks; Generators; Image enhancement; Machine learning; Performance degradation; Recognition; Sensors; Titanium alloys; Titanium base alloys; Tool Condition Monitoring; Tool wear; Training
• ISSN: 1530-437X; 1558-1748
• DOI: 10.1109/JSEN.2024.3372438

Deep learning has been widely applied in fault diagnosis and monitoring, but obtaining labeled data under abnormal conditions is challenging. This limitation makes the best current deep learning methods seem powerless as they require a large amount of labeled data for training. When the training dataset is small, overfitting is highly likely to occur, leading to performance degradation of deep neural networks. To address this issue, this paper proposes a hybrid data augmentation mechanism (HDAM) that utilizes a multi-category generative adversarial network (MCGAN) model and similarity-based selection criteria to generate high-quality data in few-shot scenarios. The selected samples are mixed with augmented samples that undergo image flipping, rotation, and noise addition to form the training set, enhancing the generalization capability of the recognition model after training. To validate the effectiveness of the proposed method, experiments are conducted on the PHM2010 dataset and the TC4 titanium alloy dataset. The experimental results demonstrate a significant improvement in the recognition accuracy of tool wear states in few-shot scenarios.