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.

Neural Architecture Search for Adversarial Medical Image Segmentation.

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

Scalable Neural Architecture Search for 3D Medical Image Segmentation

Adversarial Consistency for Single Domain Generalization in Medical Image Segmentation

Adversarial Neural Networks in Medical Imaging Advancements and Challenges in Semantic Segmentation

A general approach to improve adversarial robustness of DNNs for medical image segmentation and detection

Semantic-Aware Generative Adversarial Nets for Unsupervised Domain Adaptation in Chest X-ray Segmentation

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

Resource Optimized Neural Architecture Search for 3D Medical Image Segmentation

SegAN: Adversarial Network with Multi-scale $L_1$ Loss for Medical Image Segmentation

AttENT: Domain-Adaptive Medical Image Segmentation Via Attention-Aware Translation and Adversarial Entropy Minimization

NAS-Unet: Neural Architecture Search for Medical Image Segmentation

Adversarial deep learning for improved abdominal organ segmentation in CT scans

Rethinking Adversarial Domain Adaptation: Orthogonal Decomposition for Unsupervised Domain Adaptation in Medical Image Segmentation

AM-AN: Adversarial Network Based on Attention Mechanism for Medical Image Segmentation

Neural Architecture Search for Gliomas Segmentation on Multimodal Magnetic Resonance Imaging

Deep Learning Architecture Design for Multi-Organ Segmentation

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

PnP-AdaNet: Plug-and-Play Adversarial Domain Adaptation Network with a Benchmark at Cross-modality Cardiac Segmentation

Auto-DeepLab: Hierarchical Neural Architecture Search for Semantic Image Segmentation