使用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中

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

相关推荐
Uopiasd1234oo5 小时前
上下文引导模块改进YOLOv26局部与全局特征融合能力双重提升
深度学习·yolo·机器学习
懷淰メ6 小时前
【AI加持】基于PyQt+YOLO+DeepSeek的钢材焊接缺陷检测系统(详细介绍)
yolo·目标检测·计算机视觉·pyqt·缺陷检测·deepseek·钢材缺陷
动物园猫6 小时前
工业织物缺陷目标检测数据集分享(适用于YOLO系列深度学习分类检测任务)
深度学习·yolo·目标检测
迪菲赫尔曼7 小时前
从 0 到 1 打造工业级推理控制台:UltraConsole(Ultralytics + FastAPI + React)开源啦!
前端·yolo·react.js·计算机视觉·开源·fastapi
极智视界10 小时前
分类数据集 - 遥感航空影像云量检测图像分类数据集下载
yolo·数据集·图像分类·算法训练·遥感航空影像云量检测
极智视界11 小时前
分类数据集 - 伪造人脸和真实人脸分类数据集下载
人工智能·yolo·数据集·图像分类·算法训练·人脸伪造检测
深度学习lover12 小时前
<数据集>yolo 常见对象检测<目标检测>
人工智能·python·yolo·目标检测·计算机视觉·常见对象检测
Westward-sun.12 小时前
YOLOv2算法全方位解析:从BatchNorm到聚类先验框的九大改进
算法·yolo·聚类
动物园猫14 小时前
工业粉尘检测数据集分享(适用于YOLO系列深度学习分类检测任务)
深度学习·yolo·分类
jay神1 天前
VisDrone2019-DET 无人机小目标检测数据集
人工智能·深度学习·yolo·目标检测·计算机视觉·毕业设计·无人机