Publication: BioNAS: neural architecture search for multiple data modalities in biomedical image classification
Program
KU-Authors
KU Authors
Co-Authors
Kuş Z., Kiraz B., Aydin M.
Publication Date
Language
Embargo Status
Journal Title
Journal ISSN
Volume Title
Alternative Title
Abstract
Neural Architecture Search (NAS) for biomedical image classification has the potential to design highly efficient and accurate networks automatically for tasks from different modalities. This paper presents BioNAS, a new NAS approach designed for multi-modal biomedical image classification. Unlike other methods, BioNAS dynamically adjusts the number of stacks, modules, and feature maps in the network to improve both performance and complexity. The proposed approach utilizes an opposition-based differential evolution optimization technique to identify the optimal network structure. We have compared our methods on two public multi-class classification datasets with different data modalities: DermaMNIST and OrganCMNIST. BioNAS outperforms hand-designed networks, automatic machine learning frameworks, and most NAS studies in terms of accuracy (ACC) and area under the curve (AUC) on the OrganCMNIST and DermaMNIST datasets. The proposed networks significantly outperform all other methods on the DermaMNIST dataset, achieving accuracy improvements of up to 4.4 points and AUC improvements of up to 2.6 points, and also surpass other studies by up to 5.4 points in accuracy and 0.6 points in AUC on OrganCMNIST. Moreover, the proposed networks have fewer parameters than hand-designed architectures like ResNet-18 and ResNet-50. The results indicate that BioNAS has the potential to be an effective alternative to hand-designed networks and automatic frameworks, offering a competitive solution in the classification of biomedical images. © The Author(s), under exclusive license to Springer Nature Switzerland AG 2024.
Source
Publisher
Springer Science and Business Media Deutschland GmbH
Subject
Physics
Citation
Has Part
Source
Lecture Notes in Networks and Systems
Book Series Title
Edition
DOI
10.1007/978-3-031-70924-1_41