Affordable Access

deepdyve-link
Publisher Website

Multiview Wasserstein generative adversarial network for imbalanced pearl classification

Authors
  • Gao, Shuang
  • Dai, Yun
  • Li, Yingjie
  • Liu, Kaixin
  • Chen, Kun
  • Liu, Yi
Type
Published Article
Journal
Measurement Science and Technology
Publisher
IOP Publishing
Publication Date
May 11, 2022
Volume
33
Issue
8
Identifiers
DOI: 10.1088/1361-6501/ac6224
Source
ioppublishing
Keywords
Disciplines
  • Imaging
License
Unknown

Abstract

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.

Report this publication

Statistics

Seen <100 times