期刊
ISA TRANSACTIONS
卷 111, 期 -, 页码 337-349出版社
ELSEVIER SCIENCE INC
DOI: 10.1016/j.isatra.2020.11.005
关键词
Deep learning; Small dataset; Data augmentation; Regularization method; Batch Normalization
资金
- National Key Research and Development Program of China [2019YFF0302204]
- National Natural Science Foundation of China [51875436, 91960106, U1933101, 61633001, 51421004]
- China Postdoctoral Science Foundation [2020T130509, 2018M631145]
- Shaanxi Natural Science Foundation, China [2019JM041]
A supervised data augmentation method (SDA) based on 1-D mechanical vibration signal characteristics is proposed, which utilizes Cut-Flip and Mix-Normal to provide more effective training samples and significantly improve classification accuracy.
Data-driven intelligent diagnosis model plays a key role in the monitoring and maintenance of mechanical equipment. However, due to practical limitations, the fault data is difficult to obtain, which makes model training unsatisfactory and results in poor testing performance. Based on the characteristics of 1-D mechanical vibration signal, this paper proposes Supervised Data Augmentation (SDA) as a regularization method to provide more effective training samples, which includes Cut-Flip and Mix-Normal. Cut-Flip is used directly on the raw sample without parameter selection. Mix-Normal mixes the data and labels of a random sample with a random normal sample at a certain ratio. The proposed SDA is verified on two bearing datasets with some popular intelligent diagnosis networks. Besides, we also design a Batch Normalization CNN (BNCNN) to learn the small dataset. Results show that SDA can significantly improve the classification accuracy of BNCNN by 10%-30% under 1-8 samples of each class. The proposed method also shows a competitive performance with existing advanced methods. Finally, we further discuss each data augmentation method through a series of ablation experiments and summarize the advantages and disadvantages of the proposed SDA. (c) 2020 ISA. Published by Elsevier Ltd. All rights reserved.
作者
我是这篇论文的作者
点击您的名字以认领此论文并将其添加到您的个人资料中。
推荐
暂无数据