Eff2Net: An efficient channel attention-based convolutional neural network for skin disease classification

被引:70
作者
Karthik, R. [1 ]
Vaichole, Tejas Sunil [2 ]
Kulkarni, Sanika Kiran [2 ]
Yadav, Ojaswa [3 ]
Khan, Faiz [3 ]
机构
[1] Vellore Inst Technol, Ctr Cyber Phys Syst, Chennai, Tamil Nadu, India
[2] Vellore Inst Technol, Sch Comp Sci & Engn, Chennai, Tamil Nadu, India
[3] Vellore Inst Technol, Sch Elect Engn, Chennai, Tamil Nadu, India
关键词
EfficientNetV2; Efficient channel attention; Skin disease; Classification; CNN;
D O I
10.1016/j.bspc.2021.103406
中图分类号
R318 [生物医学工程];
学科分类号
0831 ;
摘要
The primary layer of protection for vital organs in the human body is the skin. It functions as a barrier to protect our internal organs from different sources. However, infections caused by fungus, viruses, or even dust can damage the skin. A tiny lesion on the skin can grow into something that can cause serious health problems. A good diagnosis can help the person suffering from a skin disease to recover quickly. This research aims to develop a system for detecting skin diseases using a Convolution Neural Network (CNN). The proposed model named Eff2Net is built on EfficientNetV2 in conjunction with the Efficient Channel Attention (ECA) block. This research attempts to replace the standard Squeeze and Excitation (SE) block in the EfficientNetV2 architecture with the ECA block. By doing so, it was observed that there was a significant drop in the total number of trainable pa-rameters. The proposed CNN learnt around 16 M parameters to classify the disease, which is comparatively less than the existing deep learning approaches reported in the literature. This skin disease classification was per -formed on four classes: acne, actinic keratosis (AK), melanoma, and psoriasis. The model achieved an overall testing accuracy of 84.70%.
引用
收藏
页数:11
相关论文
共 60 条
[1]   Discriminative Feature Learning for Skin Disease Classification Using Deep Convolutional Neural Network [J].
Ahmad, Belal ;
Usama, Mohd ;
Huang, Chuen-Min ;
Hwang, Kai ;
Hossain, M. Shamim ;
Muhammad, Ghulam .
IEEE ACCESS, 2020, 8 :39025-39033
[2]  
Al-Masni MA, 2019, 2019 IEEE EUR C BIOM
[3]   Multiple skin lesions diagnostics via integrated deep convolutional networks for segmentation and classification [J].
Al-masni, Mohammed A. ;
Kim, Dong-Hyun ;
Kim, Tae-Seong .
COMPUTER METHODS AND PROGRAMS IN BIOMEDICINE, 2020, 190 (190)
[4]  
ALEnezi N. S. A., 2019, Procedia Comput Sci, V163, P85, DOI [DOI 10.1016/J.PROCS.2019.12.090, 10.1016/j.procs.2019.12.090]
[5]  
[Anonymous], 2018, ADV INTELLIGENT SYST
[6]  
[Anonymous], IMAGE LIB
[7]  
[Anonymous], 2022, Home
[8]  
[Anonymous], Training imagenet in 3 hours for 25 minutes
[9]  
[Anonymous], 2016, SKIN CANC DETECTION
[10]  
[Anonymous], INTERACTIVE DERMATOL