MixSearch: Searching for Domain Generalized Medical Image Segmentation Architectures
arXiv:2102.13280
Abstract
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 images focused on specific datasets and did not take into account the generalization ability of the learned architectures on unseen datasets as well as different domains. In this paper, we address this point by proposing to search for generalizable U-shape architectures on a composited dataset that mixes medical images from multiple segmentation tasks and domains creatively, which is named MixSearch. Specifically, we propose a novel approach to mix multiple small-scale datasets from multiple domains and segmentation tasks to produce a large-scale dataset. Then, a novel weaved encoder-decoder structure is designed to search for a generalized segmentation network in both cell-level and network-level. The network produced by the proposed MixSearch framework achieves state-of-the-art results compared with advanced encoder-decoder networks across various datasets.
References in corpus (13)
- Very Deep Convolutional Networks for Large-Scale Image Recognition
- Faster R-CNN: Towards Real-Time Object Detection with Region Proposal Networks
- MobileNets: Efficient Convolutional Neural Networks for Mobile Vision Applications
- Rethinking Atrous Convolution for Semantic Image Segmentation
- mixup: Beyond Empirical Risk Minimization
- Attention U-Net: Learning Where to Look for the Pancreas
- Neural Architecture Search with Reinforcement Learning
- MultiResUNet : Rethinking the U-Net Architecture for Multimodal Biomedical Image Segmentation
- DARTS: Differentiable Architecture Search
- The Liver Tumor Segmentation Benchmark (LiTS)
- Skin Lesion Analysis Toward Melanoma Detection 2018: A Challenge Hosted by the International Skin Imaging Collaboration (ISIC)
- Recurrent Residual Convolutional Neural Network based on U-Net (R2U-Net) for Medical Image Segmentation
- Searching for Efficient Multi-Scale Architectures for Dense Image Prediction