Classification of images using EfficientNet CNN model with convolutional block attention module (CBAM) and spatial group-wise enhance module (SGE)

Abstract

Classification of images is highly useful in medical, agriculture, industry, and other fields. Improving the accuracy of classification with a small quality of parameters is a challenging problem. This paper performs a study that relied on the use of EfficientNet and convolutional block attention module (CBAM). Especially, Spatial Group-wise Enhance (SGE) module is used to adjust the importance of sub features and suppress possible noise. Stochastic gradient 1. descent (SGD) is chosen as the optimizer. After experiments, the EfficientNet model with CBAM module and SGE module can achieve higher accuracy in image classification. This module has achieved high accuracy on Flowers (98.53%).

Topics & Concepts