Journal
MEASUREMENT SCIENCE AND TECHNOLOGY
Volume 33, Issue 8, Pages -Publisher
IOP Publishing Ltd
DOI: 10.1088/1361-6501/ac6224
Keywords
imbalanced learning; pearl classification; generative adversarial network; deep learning; convolutional neural network
Funding
- National Natural Science Foundation of China [62022073, 61873241]
- Natural Science Foundation of Zhejiang Province [LQ22E090007]
Ask authors/readers for more resources
This work aims to enhance the automation level of industrial pearl classification through deep learning methods. To address the issue of imbalanced datasets, an enhanced generative adversarial network named MVWGAN is proposed to generate high-quality multiview images and balance the datasets. Experimental results demonstrate that the MVWGAN method effectively solves the imbalanced learning problem and improves the classification performance of pearl classification.
This work described in this paper aims to enhance the level of automation of industrial pearl classification through deep learning methods. To better extract the features of different classes and improve classification accuracy, balanced training datasets are usually needed for machine learning methods. However, the pearl datasets obtained in practice are often imbalanced; in particular, the acquisition cost of some classes is high. An enhanced generative adversarial network, named the multiview Wasserstein generative adversarial network (MVWGAN), is proposed for the imbalanced pearl classification problem. For the minority classes in the training datasets, the MVWGAN method can generate high-quality multiview images simultaneously to balance the original imbalanced datasets. The augmented balanced datasets are used to train a multistream convolution neural network (MS-CNN) for pearl classification. The experimental results show that MVWGAN can overcome the imbalanced learning problem and improve the classification performance of MS-CNN effectively. Moreover, feature visualization is implemented to intuitively explain the effectiveness of MVWGAN.
Authors
I am an author on this paper
Click your name to claim this paper and add it to your profile.
Reviews
Recommended
No Data Available