| Title |
Performance Improvement of Transfer Learning-based Motor Fault Diagnosis Using Mix-up Data Considering Data Imbalance |
| Authors |
최의진(Eui-Jin Choi) ; 한지훈(Ji-Hoon Han) ; 홍선기(Sun-Ki Hong) |
| DOI |
https://doi.org/10.5370/KIEE.2023.72.2.255 |
| Keywords |
Deep learning; Mix-up; Motor fault diagnosis; Data augmentation; Data imbalance |
| Abstract |
Most of the data obtained for motor fault diagnosis is the data of normal motors. Therefore, the model is trained using normal data collected in the field and fault data collected in experiment. In addtion, transfer learning is used for fast learning convergence. Fault diagnosis performance may be reduced by the number of unbalanced data in each state during transfer learning.
In this study, a transfer learning method using mix-up data is proposed to solve this data imbalance. A mix-up is a technique of adding insufficient data to the data with clear decision boundaries, and can be used to balance a large number of normal state data with a small number of fault-state data. The proposed method is compared with GAN, another technique of data augmentation. As a result of the experiment, it was confirmed that the proposed mix-up technique is a method to improve the performance of fault diagnosis. |