Data Augmentation for Improved Melanoma Classification in Convolutional Neural Networks

被引:0
作者
Hermosilla, Pamela [1 ]
Soto, Ricardo [1 ]
Ponce, Jefte [1 ]
Suazo Jara, Cristian [1 ]
Crawford, Broderick [1 ]
Contreras, Sebastian [1 ]
机构
[1] Pontificia Univ Catolica Valparaiso, Valparaiso, Chile
来源
ADVANCED RESEARCH IN TECHNOLOGIES, INFORMATION, INNOVATION AND SUSTAINABILITY, ARTIIS 2024, PT II | 2025年 / 2346卷
关键词
Convolutional Neural Networks; Data augmentation; Skin cancer;
D O I
10.1007/978-3-031-83210-9_13
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Convolutional Neural Networks (CNNs) are highly effective in various computer vision tasks. However, their performance is heavily dependent on the quality and quantity of the training data. This study aims to optimise CNN performance using pattern augmentation techniques to address the problem of limited training data in medical image analysis. Using the DenseNet architecture and the HAM10000 dataset of melanoma and non-melanoma images, the researchers applied a range of augmentation methods, including random rotation, brightness adjustment, random zoom, horizontal flip, height and width shift, and random cropping. The methodology involved evaluating the impact of these augmentation techniques on CNN training. The results showed that random crop and horizontal flip significantly improved classification accuracy by 1-2% over 30 epochs. Extending training to 50 epochs further enhanced performance, though not all methods showed significant improvements; random crop and horizontal flip remained the most effective. These findings highlight the crucial role of selecting appropriate augmentation techniques and training epochs in optimizing CNN performance. The study underscores the importance of dataset diversity and augmentation strategies, particularly in medical image analysis where early and accurate diagnosis is critical. Future research should explore different network architectures and datasets to validate and generalize these results.
引用
收藏
页码:164 / 182
页数:19
相关论文
共 45 条
[1]  
American Academy of Dermatology, 2021, Skin cancer symptoms: What to look for
[2]  
American Cancer Society, 2020, What is cancer?
[3]  
Arslan E., 2023, J. Epigenetics, V10, P123
[4]  
Benbakreti S., 2023, J. Med. Imaging, V10, P245
[5]   LEARNING LONG-TERM DEPENDENCIES WITH GRADIENT DESCENT IS DIFFICULT [J].
BENGIO, Y ;
SIMARD, P ;
FRASCONI, P .
IEEE TRANSACTIONS ON NEURAL NETWORKS, 1994, 5 (02) :157-166
[6]   Multipath feature recalibration DenseNet for image classification [J].
Chen, Bolin ;
Zhao, Tiesong ;
Liu, Jiahui ;
Lin, Liqun .
INTERNATIONAL JOURNAL OF MACHINE LEARNING AND CYBERNETICS, 2021, 12 (03) :651-660
[7]   Randaugment: Practical automated data augmentation with a reduced search space [J].
Cubuk, Ekin D. ;
Zoph, Barret ;
Shlens, Jonathon ;
Le, Quoc, V .
2020 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION WORKSHOPS (CVPRW 2020), 2020, :3008-3017
[8]  
Di Biasi L., 2023, J. Med. Imaging, V15
[9]   Rethinking data augmentation for adversarial robustness [J].
Eghbal-zadeh, Hamid ;
Zellinger, Werner ;
Pintor, Maura ;
Grosse, Kathrin ;
Koutini, Khaled ;
Moser, Bernhard A. ;
Biggio, Battista ;
Widmer, Gerhard .
INFORMATION SCIENCES, 2024, 654
[10]   DenseNet-II: an improved deep convolutional neural network for melanoma cancer detection [J].
Girdhar, Nancy ;
Sinha, Aparna ;
Gupta, Shivang .
SOFT COMPUTING, 2023, 27 (18) :13285-13304