Deformable Mamba for Wide Field of View Segmentation

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Hu, Jie, Zheng, Junwei, Wei, Jiale, Zhang, Jiaming, Stiefelhagen, Rainer
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866912269039828992
author Hu, Jie
Zheng, Junwei
Wei, Jiale
Zhang, Jiaming
Stiefelhagen, Rainer
author_facet Hu, Jie
Zheng, Junwei
Wei, Jiale
Zhang, Jiaming
Stiefelhagen, Rainer
contents Recent advancements in the Mamba architecture, with its linear computational complexity, being a promising alternative to transformer architectures suffering from quadratic complexity. While existing works primarily focus on adapting Mamba as vision encoders, the critical role of task-specific Mamba decoders remains under-explored, particularly for distortion-prone dense prediction tasks. This paper addresses two interconnected challenges: (1) The design of a Mamba-based decoder that seamlessly adapts to various architectures (e.g., CNN-, Transformer-, and Mamba-based backbones), and (2) The performance degradation in decoders lacking distortion-aware capability when processing wide-FoV images (e.g., 180° fisheye and 360° panoramic settings). We propose the Deformable Mamba Decoder, an efficient distortion-aware decoder that integrates Mamba's computational efficiency with adaptive distortion awareness. Comprehensive experiments on five wide-FoV segmentation benchmarks validate its effectiveness. Notably, our decoder achieves a +2.5% performance improvement on the 360° Stanford2D3D segmentation benchmark while reducing 72% parameters and 97% FLOPs, as compared to the widely-used decoder heads.
format Preprint
id arxiv_https___arxiv_org_abs_2411_16481
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Deformable Mamba for Wide Field of View Segmentation
Hu, Jie
Zheng, Junwei
Wei, Jiale
Zhang, Jiaming
Stiefelhagen, Rainer
Computer Vision and Pattern Recognition
Recent advancements in the Mamba architecture, with its linear computational complexity, being a promising alternative to transformer architectures suffering from quadratic complexity. While existing works primarily focus on adapting Mamba as vision encoders, the critical role of task-specific Mamba decoders remains under-explored, particularly for distortion-prone dense prediction tasks. This paper addresses two interconnected challenges: (1) The design of a Mamba-based decoder that seamlessly adapts to various architectures (e.g., CNN-, Transformer-, and Mamba-based backbones), and (2) The performance degradation in decoders lacking distortion-aware capability when processing wide-FoV images (e.g., 180° fisheye and 360° panoramic settings). We propose the Deformable Mamba Decoder, an efficient distortion-aware decoder that integrates Mamba's computational efficiency with adaptive distortion awareness. Comprehensive experiments on five wide-FoV segmentation benchmarks validate its effectiveness. Notably, our decoder achieves a +2.5% performance improvement on the 360° Stanford2D3D segmentation benchmark while reducing 72% parameters and 97% FLOPs, as compared to the widely-used decoder heads.
title Deformable Mamba for Wide Field of View Segmentation
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2411.16481