Semi-supervised image classification via attention mechanism and generative adversarial network

2020 ◽  
Author(s):  
Xuezhi Xiang ◽  
Zeting Yu ◽  
Ning Lv ◽  
Xiangdong Kong ◽  
Abdulmotaleb El Saddik
Keyword(s):  
Image Classification ◽  
Attention Mechanism ◽  
Generative Adversarial Network ◽  
Adversarial Network ◽  
Supervised Image Classification

scholarly journals SubSpace Capsule Network

2020 ◽  
Vol 34 (07) ◽  
pp. 10745-10753
Author(s):  
Marzieh Edraki ◽  
Nazanin Rahnavard ◽  
Mubarak Shah
Keyword(s):  
Image Classification ◽  
Lower Layer ◽  
Spatial Relation ◽  
Generative Models ◽  
Test Time ◽  
Generative Adversarial Network ◽  
Major Drawback ◽  
Computational Overhead ◽  
Adversarial Network ◽  
Supervised Image Classification

Convolutional neural networks (CNNs) have become a key asset to most of fields in AI. Despite their successful performance, CNNs suffer from a major drawback. They fail to capture the hierarchy of spatial relation among different parts of an entity. As a remedy to this problem, the idea of capsules was proposed by Hinton. In this paper, we propose the SubSpace Capsule Network (SCN) that exploits the idea of capsule networks to model possible variations in the appearance or implicitly-defined properties of an entity through a group of capsule subspaces instead of simply grouping neurons to create capsules. A capsule is created by projecting an input feature vector from a lower layer onto the capsule subspace using a learnable transformation. This transformation finds the degree of alignment of the input with the properties modeled by the capsule subspace.We show that SCN is a general capsule network that can successfully be applied to both discriminative and generative models without incurring computational overhead compared to CNN during test time. Effectiveness of SCN is evaluated through a comprehensive set of experiments on supervised image classification, semi-supervised image classification and high-resolution image generation tasks using the generative adversarial network (GAN) framework. SCN significantly improves the performance of the baseline models in all 3 tasks.

Attention-Based Generative Adversarial Network for Semi-supervised Image Classification

2019 ◽  
Vol 51 (2) ◽  
pp. 1527-1540 ◽  
Author(s):  
Xuezhi Xiang ◽  
Zeting Yu ◽  
Ning Lv ◽  
Xiangdong Kong ◽  
Abdulmotaleb El Saddik
Keyword(s):  
Image Classification ◽  
Generative Adversarial Network ◽  
Adversarial Network ◽  
Supervised Image Classification

scholarly journals An Imbalanced Image Classification Method for the Cell Cycle Phase

Information ◽  
2021 ◽  
Vol 12 (6) ◽  
pp. 249
Author(s):  
Xin Jin ◽  
Yuanwen Zou ◽  
Zhongbing Huang
Keyword(s):  
Cell Cycle ◽  
Deep Learning ◽  
Image Classification ◽  
Classification Accuracy ◽  
Data Augmentation ◽  
Cycle Phase ◽  
Generative Adversarial Network ◽  
Adversarial Network ◽  
Cellular Life

The cell cycle is an important process in cellular life. In recent years, some image processing methods have been developed to determine the cell cycle stages of individual cells. However, in most of these methods, cells have to be segmented, and their features need to be extracted. During feature extraction, some important information may be lost, resulting in lower classification accuracy. Thus, we used a deep learning method to retain all cell features. In order to solve the problems surrounding insufficient numbers of original images and the imbalanced distribution of original images, we used the Wasserstein generative adversarial network-gradient penalty (WGAN-GP) for data augmentation. At the same time, a residual network (ResNet) was used for image classification. ResNet is one of the most used deep learning classification networks. The classification accuracy of cell cycle images was achieved more effectively with our method, reaching 83.88%. Compared with an accuracy of 79.40% in previous experiments, our accuracy increased by 4.48%. Another dataset was used to verify the effect of our model and, compared with the accuracy from previous results, our accuracy increased by 12.52%. The results showed that our new cell cycle image classification system based on WGAN-GP and ResNet is useful for the classification of imbalanced images. Moreover, our method could potentially solve the low classification accuracy in biomedical images caused by insufficient numbers of original images and the imbalanced distribution of original images.

Sign in / Sign up

Export Citation Format

Share Document