BiX-NAS: Searching Efficient Bi-directional Architecture for Medical Image Segmentation
The recurrent mechanism has recently been introduced into U-Net in various medical image segmentation tasks. Existing studies have focused on promoting network recursion via reusing building blocks. Although network parameters could be greatly saved, computational costs still increase inevitably in accordance with the pre-set iteration time. In this work, we study a multi-scale upgrade of a bi-directional skip connected network and then automatically discover an efficient architecture by a novel two-phase Neural Architecture Search (NAS) algorithm, namely BiX-NAS. Our proposed method reduces the network computational cost by sifting out ineffective multi-scale features at different levels and iterations. We evaluate BiX-NAS on two segmentation tasks using three different medical image datasets, and the experimental results show that our BiX-NAS searched architecture achieves the state-of-the-art performance with significantly lower computational cost.
Code (1)
Tasks
Image SegmentationMedical Image SegmentationNeural Architecture SearchSegmentationSemantic SegmentationMethods 이 논문이 사용한 방법론
Similar Papers 제목 키워드 기반
UXNet: Searching Multi-level Feature Aggregation for 3D Medical Image Segmentation
Aggregating multi-level feature representation plays a critical role in achieving robust volumetric medical image segmentation, which is important for the auxiliary diagnosis and treatment. Unlike the recent neural archi…
DecoderImage SegmentationMedical Image SegmentationNeural Architecture Search+3Patch-MoE Mamba: A Patch-Ordered Mixture-of-Experts State Space Architecture for Medical Image Segmentation
CNN- and Transformer-based architectures have achieved strong performance in medical image segmentation, but CNNs are limited in modeling long-range dependencies, while Transformers often suffer from quadratic computatio…
Medical Image SegmentationSkin Lesion SegmentationPolyp SegmentationMixSearch: Searching for Domain Generalized Medical Image Segmentation Architectures
Considering the scarcity of medical data, most datasets in medical image analysis are an order of magnitude smaller than those of natural images. However, most Network Architecture Search (NAS) approaches in medical imag…
DecoderImage SegmentationMedical Image AnalysisMedical Image Segmentation+2DM-SegNet: Dual-Mamba Architecture for 3D Medical Image Segmentation with Global Context Modeling
Accurate 3D medical image segmentation demands architectures capable of reconciling global context modeling with spatial topology preservation. While State Space Models (SSMs) like Mamba show potential for sequence model…
AnatomyBrain Tumor SegmentationDecoderImage Segmentation+7SliceMamba with Neural Architecture Search for Medical Image Segmentation
Despite the progress made in Mamba-based medical image segmentation models, existing methods utilizing unidirectional or multi-directional feature scanning mechanisms struggle to effectively capture dependencies between …
Image SegmentationMambaMedical Image SegmentationNeural Architecture Search+4