UNetMamba: An Efficient UNet-Like Mamba for Semantic Segmentation of High-Resolution Remote Sensing Images

2024-08-21 · Enze Zhu, Zhan Chen, Dingkai Wang, Hanru Shi, Xiaoxuan Liu, Lei Wang

Semantic segmentation of high-resolution remote sensing images is vital in downstream applications such as land-cover mapping, urban planning and disaster assessment.Existing Transformer-based methods suffer from the constraint between accuracy and efficiency, while the recently proposed Mamba is renowned for being efficient. Therefore, to overcome the dilemma, we propose UNetMamba, a UNet-like semantic segmentation model based on Mamba. It incorporates a mamba segmentation decoder (MSD) that can efficiently decode the complex information within high-resolution images, and a local supervision module (LSM), which is train-only but can significantly enhance the perception of local contents. Extensive experiments demonstrate that UNetMamba outperforms the state-of-the-art methods with mIoU increased by 0.87% on LoveDA and 0.39% on ISPRS Vaihingen, while achieving high efficiency through the lightweight design, less memory footprint and reduced computational cost. The source code is available at https://github.com/EnzeZhu2001/UNetMamba.

📄 PDF Abstract BibTeX arXiv:2408.11545

Code (1)

enzezhu2001/unetmamba 공식 구현 pytorch

Tasks

MambaSegmentation Semantic Segmentation The Semantic Segmentation Of Remote Sensing Imagery

Methods 이 논문이 사용한 방법론

Mamba Foundation models, now powering most of the exciting applications in deep learning, are almost universally based on the Transformer architecture and its core attention module.…

Mamba-UNet: UNet-Like Pure Visual Mamba for Medical Image Segmentation

2024-02-07 · Ziyang Wang, Jian-Qing Zheng, Yichi Zhang, Ge Cui 외

In recent advancements in medical image analysis, Convolutional Neural Networks (CNN) and Vision Transformers (ViT) have set significant benchmarks. While the former excels in capturing local features through its convolu…

Cardiac Segmentation Computational Efficiency Decoder Image Segmentation+5

Integrating Mamba Sequence Model and Hierarchical Upsampling Network for Accurate Semantic Segmentation of Multiple Sclerosis Legion

2024-03-26 · Kazi Shahriar Sanjid, Md. Tanzim Hossain, Md. Shakib Shahariar Junayed, Dr. Mohammad Monir Uddin

Integrating components from convolutional neural networks and state space models in medical image segmentation presents a compelling approach to enhance accuracy and efficiency. We introduce Mamba HUNet, a novel architec…

Decision Making Image Segmentation Lesion Segmentation Mamba+4

CM-UNet: Hybrid CNN-Mamba UNet for Remote Sensing Image Semantic Segmentation

2024-05-17 · Mushui Liu, Jun Dan, Ziqian Lu, Yunlong Yu 외

Due to the large-scale image size and object variations, current CNN-based and Transformer-based approaches for remote sensing image semantic segmentation are suboptimal for capturing the long-range dependency or limited…

Decoder Mamba Segmentation Semantic Segmentation

LightM-UNet: Mamba Assists in Lightweight UNet for Medical Image Segmentation

2024-03-08 · Weibin Liao, Yinghao Zhu, Xinyuan Wang, Chengwei Pan 외

UNet and its variants have been widely used in medical image segmentation. However, these models, especially those based on Transformer architectures, pose challenges due to their large number of parameters and computati…

Image Segmentation Mamba Medical Image Segmentation Semantic Segmentation+1

VM-UNET-V2 Rethinking Vision Mamba UNet for Medical Image Segmentation

2024-03-14 · Mingya Zhang, Yue Yu, Limei Gu, Tingsheng Lin 외

In the field of medical image segmentation, models based on both CNN and Transformer have been thoroughly investigated. However, CNNs have limited modeling capabilities for long-range dependencies, making it challenging …