Title
Optimization of a Deep Learning Model for Skin Cancer Detection with Magnitude-Based Weight Pruning
Date Issued
01 January 2022
Access level
metadata only access
Resource Type
conference paper
Author(s)
Publisher(s)
Springer Science and Business Media Deutschland GmbH
Abstract
Deep Learning is a field of Artificial Intelligence that has recently become very important when building intelligent systems capable of classifying images automatically and with very high accuracy. The goal is always to build a neural network architecture with high accuracy which is especially important when used for medical applications, such as skin cancer detection. However, as new Deep Learning architectures are being created these new models become more complex, increasing their depth and size due to the number of hyper-parameters involved. This increase becomes a problem when trying to deploy them in hardware with limited capacity in terms of storage and memory, such as Edge devices. In this article we create a new convolutional neural network architecture that has an accuracy of 99% for skin cancer classification, and we then compress it using a technique called Magnitude-based weight pruning to reduce its size to 65% while retaining the same accuracy.
Start page
624
End page
629
Volume
468 LNNS
Language
English
OCDE Knowledge area
Informática y Ciencias de la Información
Ingeniería de sistemas y comunicaciones
Subjects
Scopus EID
2-s2.0-85130276409
ISBN
9783031048258
Source
Lecture Notes in Networks and Systems
ISSN of the container
23673370
Conference
10th World Conference on Information Systems and Technologies, WorldCIST 2022 Budva 12 April 2022 through 14 April 2022
Sources of information:
Directorio de Producción Científica
Scopus