Abstract:Deep neural networks have demonstrated impressive results in medical image analysis, but designing suitable architectures for each specific task is expertise dependent and time consuming. Neural architecture search (NAS) offers an effective means of discovering architectures. It has been highly successful in numerous applications, particularly in natural image classification. Yet, medical images possess unique characteristics, such as small regions and a wide variety of lesion sizes, that differentiate them from natural images. Furthermore, most current NAS methods struggle with high computational costs, especially when dealing with high-resolution image datasets. In this article, we present a novel evolutionary NAS method called multiscale training-free neural architecture search (MSTF-NAS) to address these challenges. Specifically, to accommodate the broad range of lesion region sizes in disease diagnosis, we develop a new reduction cell search space that enables the search algorithm to explicitly identify the optimal scale combination for multiscale feature extraction. To overcome the issue of high computational costs, we utilize training-free indicators as performance measures for candidate architectures, which allows us to search for the optimal architecture more efficiently. More specifically, by considering the capability and simplicity of various networks, we formulate a multiobjective optimization problem that involves two training-free indicators and model complexity for candidate architectures. Extensive experiments on a large medical image benchmark and a publicly available breast cancer detection dataset are conducted. The empirical results demonstrate that our MSTF-NAS outperforms both human-designed architectures and current state-of-the-art NAS algorithms on both datasets, indicating the effectiveness of our proposed method.

Tree-Based Codification in Neural Architecture Search for Medical Image Segmentation

MS-NAS: Multi-scale Neural Architecture Search for Medical Image Segmentation

NAS-Unet: Neural Architecture Search for Medical Image Segmentation

C2FNAS: Coarse-to-Fine Neural Architecture Search for 3D Medical Image Segmentation

NG-NAS: Node growth neural architecture search for 3D medical image segmentation

Scalable Neural Architecture Search for 3D Medical Image Segmentation

HyperSegNAS: Bridging One-Shot Neural Architecture Search with 3D Medical Image Segmentation Using HyperNet

GNAS-U2Net: A New Optic Cup and Optic Disc Segmentation Architecture with Genetic Neural Architecture Search

GP-CNAS: Convolutional Neural Network Architecture Search with Genetic Programming

Adaptive Lightweight Convolutional Neural Architecture Search for Segmentation Problem

Neural Architecture Search for Gliomas Segmentation on Multimodal Magnetic Resonance Imaging

MedNAS: Multi-Scale Training-Free Neural Architecture Search for Medical Image Analysis

Continuous Cartesian Genetic Programming based representation for Multi-Objective Neural Architecture Search

NexToU: Efficient Topology-Aware U-Net for Medical Image Segmentation

Neural Architecture Search based on Cartesian Genetic Programming Coding Method

A Gradient-Guided Evolutionary Neural Architecture Search

Evolving blocks by segmentation for neural architecture search

V-NAS: Neural Architecture Search for Volumetric Medical Image Segmentation

DiNTS: Differentiable Neural Network Topology Search for 3D Medical Image Segmentation

DCNAS: Densely Connected Neural Architecture Search for Semantic Image Segmentation