ViM-UNet: Vision Mamba for Biomedical Segmentation

arXiv:2404.07705v1 Announce Type: new
Abstract: CNNs, most notably the UNet, are the default architecture for biomedical segmentation. Transformer-based approaches, such as UNETR, have been proposed to replace them, benefiting from a global field of view, but suffering from larger runtimes and higher parameter counts. The recent Vision Mamba architecture offers a compelling alternative to transformers, also providing a global field of view, but at higher efficiency. Here, we introduce ViM-UNet, a novel segmentation architecture based on it and compare it …

arxiv biomedical cs.cv mamba segmentation type unet vim vision