arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2406.04532cs.CV

MambaDepth:增强自监督细结构单目深度估计的长程依赖

MambaDepth: Enhancing Long-range Dependency for Self-Supervised Fine-Structured Monocular Depth Estimation

Ionuţ Grigore, Călin-Adrian Popa

AI总结:

本文提出MambaDepth,将Mamba状态空间模型与U-Net编码器-解码器框架结合,通过跳跃连接和新型Mamba块集成技术增强长程依赖建模,在KITTI等数据集上实现自监督单目深度估计的最先进性能。

摘要:

在自监督深度估计领域,卷积神经网络(CNNs)和Transformer长期以来占据主导地位。然而,由于局部关注或计算需求,这两种架构都难以高效处理长程依赖。为克服这一局限,我们提出MambaDepth,一个专为自监督深度估计设计的通用网络。受Mamba架构优势的启发——该架构以擅长处理长序列并通过状态空间模型(SSM)高效捕获全局上下文而闻名——我们引入MambaDepth。这一创新架构将U-Net在自监督深度估计中的有效性与Mamba的先进能力相结合。MambaDepth围绕纯Mamba编码器-解码器框架构建,引入跳跃连接以在网络各层级保持空间信息。该配置促进了广泛的特征学习过程,能够捕获深度图中的细粒度细节和更广泛上下文。此外,我们在Mamba块内开发了一种新颖的集成技术,以促进编码器和解码器组件之间不间断的连接和信息流动,从而提高深度精度。在已建立的KITTI数据集上的全面测试表明,MambaDepth在自监督深度估计任务上优于领先的基于CNN和Transformer的模型,实现了最先进的性能。此外,MambaDepth在Make3D和Cityscapes等其他数据集上证明了其优越的泛化能力。MambaDepth的性能预示着自监督深度估计中高效长程依赖建模的新时代。

英文摘要:

In the field of self-supervised depth estimation, Convolutional Neural Networks (CNNs) and Transformers have traditionally been dominant. However, both architectures struggle with efficiently handling long-range dependencies due to their local focus or computational demands. To overcome this limitation, we present MambaDepth, a versatile network tailored for self-supervised depth estimation. Drawing inspiration from the strengths of the Mamba architecture, renowned for its adept handling of lengthy sequences and its ability to capture global context efficiently through a State Space Model (SSM), we introduce MambaDepth. This innovative architecture combines the U-Net's effectiveness in self-supervised depth estimation with the advanced capabilities of Mamba. MambaDepth is structured around a purely Mamba-based encoder-decoder framework, incorporating skip connections to maintain spatial information at various levels of the network. This configuration promotes an extensive feature learning process, enabling the capture of fine details and broader contexts within depth maps. Furthermore, we have developed a novel integration technique within the Mamba blocks to facilitate uninterrupted connectivity and information flow between the encoder and decoder components, thereby improving depth accuracy. Comprehensive testing across the established KITTI dataset demonstrates MambaDepth's superiority over leading CNN and Transformer-based models in self-supervised depth estimation task, allowing it to achieve state-of-the-art performance. Moreover, MambaDepth proves its superior generalization capacities on other datasets such as Make3D and Cityscapes. MambaDepth's performance heralds a new era in effective long-range dependency modeling for self-supervised depth estimation.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑