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

• Published in: IEEE Transactions on Evolutionary Computation Vol. 28; no. 3; pp. 668 - 681
• Main Authors: Wang, Yan, Zhen, Liangli, Zhang, Jianwei, Li, Miqing, Zhang, Lei, Wang, Zizhou, Feng, Yangqin, Xue, Yu, Wang, Xiao, Chen, Zheng, Luo, Tao, Goh, Rick Siow Mong, Liu, Yong
• Format: Journal Article
• Language: English
• Published: New York IEEE 01.06.2024; Institute of Electrical and Electronics Engineers (IEEE); The Institute of Electrical and Electronics Engineers, Inc. (IEEE)
• Subjects: 60 APPLIED LIFE SCIENCES; Artificial neural networks; Computing costs; Datasets; Empirical analysis; Evolutionary optimization; Feature extraction; Image analysis; Image classification; Image resolution; Indicators; Lesions; Medical diagnostic imaging; medical image classification; Medical imaging; Multiple objective analysis; Neural architecture search; neural architecture search (NAS); Neural networks; Optimization; Search algorithms; training-free architecture search
• ISSN: 1089-778X; 1941-0026; 1941-0026
• DOI: 10.1109/TEVC.2024.3352641

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.