Neural Architecture Search for Skin Lesion Classification

Arkadiusz Kwasigroch; Michał Grochowski; Agnieszka Mikołajczyk-Bareła

doi:10.1109/access.2020.2964424

Neural Architecture Search for Skin Lesion Classification

Abstract

Deep neural networks have achieved great success in many domains. However, successful deployment of such systems is determined by proper manual selection of the neural architecture. This is a tedious and time-consuming process that requires expert knowledge. Different tasks need very different architectures to obtain satisfactory results. The group of methods called the neural architecture search (NAS) helps to find effective architecture in an automated manner. In this paper, we present the use of an architecture search framework to solve the medical task of malignant melanoma detection. Unlike many other methods tested on benchmark datasets, we tested it on practical problem, which differs greatly in terms of difficulty in distinguishing between classes, resolution of images, data balance within the classes, and the number of data available. In order to find a suitable network structure, the hill-climbing search strategy was employed along with network morphism operations to explore the search space. The network morphism operations allow for incremental increases in the network size with the use of the previously trained network. This kind of knowledge reusing allows significantly reducing the computational cost. The proposed approach produces structures that achieve similar results to those provided by manually designed structures, at the same time making use of almost 20 times fewer parameters. What is more, the search process lasts on average only 18h on single GPU.

Citations

5 1

CrossRef
0

Web of Science
5 2

Scopus

Authors (3)

Cite as

Full text

download paper

downloaded 128 times

Publication version: Accepted or Published Version
License: open in new tab

Keywords

DEEP LEARNING , CONVOLUTIONAL NEURAL NETWORK , NEURAL ARCHITECTURE SEARCH , NETWORK MORPHISM , MALIGNANT MELANOMA

Details

Category:: Articles
Type:: artykuły w czasopismach
Published in:: IEEE Access no. 8, pages 9061 - 9071,
ISSN: 2169-3536
Language:: English
Publication year:: 2020
Bibliographic description:: Kwasigroch A., Grochowski M., Mikołajczyk A.: Neural Architecture Search for Skin Lesion Classification// IEEE Access -Vol. 8, (2020), s.9061-9071
DOI:: Digital Object Identifier (open in new tab) 10.1109/access.2020.2964424
Verified by:: Gdańsk University of Technology