使用Mambaout替换YOLObackbone 整合全局信息,提升遮挡目标检测中定位能力,以及小目标、多尺度

近年来,Transformer 架构虽在各类任务中成为主流,但注意力机制的二次复杂度对长序列处理构成挑战。为此,类似 RNN 的模型如 Mamba 被引入,其核心是状态空间模型(SSM),旨在以线性复杂度处理长序列。然而,将 Mamba 应用于视觉任务时,其性能常不及卷积和基于注意力的模型。研究发现,Mamba 更适合兼具长序列和自回归特性的任务,而多数视觉任务(如图像分类)并不满足这些特性。基于此,研究者提出 MambaOut,通过移除 Mamba 块中的 SSM,仅保留门控 CNN 结构,验证 SSM 在视觉任务中的必要性。

1.Mambaout原理

MambaOut 的核心原理是基于对 Mamba 适用场景的分析:Mamba 的 SSM 机制具有 RNN 特性,适合处理长序列且需因果 token 混合(自回归)的任务,而视觉理解任务(如分类)无需因果限制,且短序列场景下 SSM 优势不明显。MambaOut 通过堆叠门控 CNN 块构建模型,门控 CNN 块与 Mamba 块的区别在于不含 SSM,其 token 混合依赖深度卷积,结合 MLP 和门控机制,在保证计算效率的同时,避免了 SSM 在非长序列视觉任务中的冗余性。实验表明,该结构在 ImageNet 分类中超越含 SSM 的视觉 Mamba 模型,验证了 SSM 的非必要。

MambaOut 采用类似 ResNet 的分层架构,包含四个阶段,每个阶段堆叠门控 CNN 块。门控 CNN 块的具体结构为:输入经归一化后,通过线性层分为门控信号(g)、输入信号(i)和卷积信号(c);c 经深度卷积(如 7×7 kernel)处理后,与 i 拼接并与 g 的激活值相乘,再通过线性层输出,最终与残差连接相加。模型配置根据尺寸(如 Femto、Tiny、Small、Base)调整通道数、块数量等参数,例如 Small 版本包含(3, 4, 27, 3)个块,通道数为(96, 192, 384, 576)。

2.Mambaout习作思路​

​​MambaOut 采用 7×7 深度 wise 卷积作为 Token Mixer,通过局部感受野聚合信息,相较于 Mamba 的 SSM 模块更专注于空间局部模式提取,在遥感缺陷检测中能精准捕获裂缝、破损等小目标缺陷的边缘与结构特征,其类似 ResNet 的四阶段分层架构可从不同尺度解析缺陷形态,避免长序列依赖引入的无关上下文干扰,结合门控机制与深度卷积的高效特征交互,在抵御云层、地物纹理等背景干扰的同时,充分利用 GPU 并行计算能力,实现高分辨率遥感图像中多尺度缺陷的精准定位与快速检测。

3. YOLO与Mambaout 的结合

MambaOut 替换 YOLO 主干时,通过门控 CNN 强化局部特征提取,7×7 深度卷积扩大感受野,显著提升小目标检测能力。其架构移除 SSM 避免因果约束,能更好整合全局信息,提升遮挡目标检测中定位能力。

4. Mambaout代码部分

使用Mambaout替换YOLO backbone 整合全局信息,提升遮挡目标检测中定位能力,以及小目标、多尺度_哔哩哔哩_bilibili

代码获取: https://github.com/tgf123/YOLOv8_improve

5. Mambaout引入到YOLOv12中

将百度网盘的压缩包下载后解压,用编辑器打开运行即可

6. Mambaout引入到YOLOv11中

将百度网盘的压缩包下载后解压,用编辑器打开运行即可

相关推荐
YOLO数据集集合3 小时前
小麦穗检测数据集 | 小麦穗检测 植物表型 密集目标检测 农业AI9182期
人工智能·yolo·目标检测·机器学习·目标跟踪·小麦穗
FL162386312921 小时前
城市道路叶子叶堆检测数据集VOC+YOLO格式351张2类别
人工智能·yolo·机器学习
AI日报派送佬1 天前
Ultralytics YOLO 模型训练技巧与最佳实践
人工智能·python·深度学习·yolo·机器学习·计算机视觉
FL16238631292 天前
外来入侵植物检测数据集VOC+YOLO格式1111张5类别
人工智能·yolo·机器学习
菜鸟变凤凰20232 天前
基于深度学习 YOLOv8 的各种识别系统(Vue3 + Element Plus + Django 全栈项目)
深度学习·yolo·django
冷雨夜中漫步2 天前
YOLO快速入门——(2)YOLO V26安装与快速入门
图像处理·算法·yolo·计算机视觉
YOLO数据集集合2 天前
遥感滑坡图像识别数据集 | 滑坡识别 遥感影像 语义分割 灾害监测 无人机航拍 目标检测 深度学习数据集 计算机视觉9172期
人工智能·yolo·目标检测·计算机视觉·无人机·滑坡
YOLO数据集集合2 天前
空中鸟类检测数据集 | 单目标 鸟类检测 生态监测 鸟击防范 目标检测9174期
安全·yolo·目标检测·飞鸟数据集·飞鸟
YOLO数据集集合2 天前
学生课堂行为智能检测系统 | 课堂行为检测 YOLOv13 DeepSeek 智慧教育 行为识别 9178期
yolo·目标检测·计算机视觉·课堂·学生课堂行为
YOLO数据集集合3 天前
Deepseek融合yolo的行人车辆多目标检测系统 |行人检测 车辆检测 多目标检测 YOLO DeepSeek9165期
人工智能·yolo·目标检测·计算机视觉·车辆检测·行人检测