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

被引:72
作者
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 条
[21]   Challenges and Recent Solutions for Image Segmentation in the Era of Deep Learning [J].
Goceri, Evgin .
2019 NINTH INTERNATIONAL CONFERENCE ON IMAGE PROCESSING THEORY, TOOLS AND APPLICATIONS (IPTA), 2019,
[22]  
Groh M., 2021, Evaluating deep neural networks trained on clinical images in dermatology with the fitzpatrick 17k dataset. Proceedings of the IEEE/CVF conference on computer vision and pattern recognition
[23]  
Guo YH, 2018, IEEE INT SYMP SIGNAL, P365, DOI 10.1109/ISSPIT.2018.8642669
[24]  
Gupta Suyog, 2020, ACCELERATOR AWARE NE
[25]  
Hameed N., 2018, INFORM MANAGEMENT AP
[26]   Multi-class multi-level classification algorithm for skin lesions classification using machine learning techniques [J].
Hameed, Nazia ;
Shabut, Antesar M. ;
Ghosh, Miltu K. ;
Hossain, M. A. .
EXPERT SYSTEMS WITH APPLICATIONS, 2020, 141
[27]  
Harangi B., 2018, 2018 40 ANN INT C IE
[28]  
Hegde PR, 2018, 2018 INTERNATIONAL CONFERENCE ON ADVANCES IN COMPUTING, COMMUNICATIONS AND INFORMATICS (ICACCI), P1825, DOI 10.1109/ICACCI.2018.8554512
[29]  
Hu J, 2018, PROC CVPR IEEE, P7132, DOI [10.1109/CVPR.2018.00745, 10.1109/TPAMI.2019.2913372]
[30]  
kaggle, ABOUT US