论文阅读笔记(五):YOLOV11-RGBT: TOWARDS A COMPREHENSIVE SINGLE-STAGE MULTISPECTRAL OBJECT DETECTION FRAMEWORK
现在可见光-红外(RGBT)多光谱检测有一堆融合方案,但大多是零散的、针对特定模型;而且默认RGB和红外一样重要,经常出现融合之后效果反而不如单模态。作者想做一套通用、单阶段、可灵活切换融合方式的RGBT检测框架,并且提出P3中层融合+MCF可控微调两个核心策略来解决模态权重失衡的痛点。
目录
论文阅读笔记(五):YOLOV11-RGBT: TOWARDS A COMPREHENSIVE SINGLE-STAGE MULTISPECTRAL OBJECT DETECTION FRAMEWORK
一、这篇论文解决了什么问题?
二、关于这个问题,其它学者提出了哪几类的解决方案,有何缺陷?
三、作者围绕该问题,是如何构建解决思路的?
四、从结果(含曲线)上看,作者是如何有力地证明他解决了问题?
五、这篇论文的缺陷在哪?
六、论文可以借鉴的地方?
文章信息
标题:YOLOV11-RGBT: TOWARDS A COMPREHENSIVE SINGLE-STAGE MULTISPECTRAL OBJECT DETECTION FRAMEWORK
来源:arXiv
年份:2025年
作者:Dahang Wan, Rongsheng Lu, Yang Fang, Xianli Lang, Shuangbao Shu, Jingjing Chen, Siyuan Shen, Ting Xu, Zecong Ye
一、这篇论文解决了什么问题?
在可见光-红外(RGBT)多光谱目标检测任务中,现有中层融合方案普遍存在模态权重不均衡的痛点:不同数据集的优势模态不一样(FLIR、LLVIP红外效果更好,VEDAI可见光更强),直接将RGB与红外特征融合时,弱势模态会引入噪声,经常出现融合之后精度反而不如单独使用优势单模态的情况;同时现有RGBT融合方案大多是定制化结构,只能适配某一个检测器,缺少一套通用、可对比多种融合范式的实验框架,很多研究直接在P3、P4、P5全部做融合,盲目增加融合节点带来特征冗余、计算开销上升。
本文目标:
- 搭建一套通用RGBT检测框架YOLOv11-RGBT,支持6种可插拔融合范式,能够迁移到 YOLOv3~v12、RT-DETR等多种单阶段检测器;
- 提出P3单节点中层融合方案,减少多位置融合带来的特征冗余;
- 提出MCF可控微调训练策略,解决模态不平衡、融合后性能退化的问题,保证融合结果保底不低于优势单模态,并且能够稳定涨点。
二、关于这个问题,其它学者提出了哪几类的解决方案,有何缺陷?
前人方案一共分为三大类,就是经典的早/中/晚融合路线,还有一些改进注意力融合模块:
1.早期融合(像素级融合)
- 做法:RGB图和红外图在输入阶段直接通道拼接,送入单个骨干网络提取特征。
- 缺陷:原始图像噪声大,两个模态像素分布差异巨大,容易互相干扰;对配准误差敏感,计算量大。
2.中层融合(特征级融合,当前主流,比如 CFT、ICAFusion 等)
- RGB、红外双分支各自提取特征,在 P3/P4/P5 多层特征图上做 Add/Concat 融合,再送入 Neck、Head 检测
- 缺陷:大多默认 RGB、红外两个模态同等重要,没有考虑模态不均衡;很多工作直接三层全部融合,信息冗余;训练时两个分支同时更新权重,优势模态学到的特征容易被弱势模态污染,容易出现融合掉点;模块往往是定制化的,通用性差,很难迁移到别的检测器
3.晚期融合 / 分数融合(决策级融合)
- 两个独立网络分别完成检测,最后只融合检测框 NMS 或者置信分数
- 缺陷:两个模态在特征层面完全没有信息交互,只在最后结果层面融合,模态之间互补信息利用不充分,涨点上限低。
4.权重共享方案:
- 套骨干权重同时处理 RGB、红外图像
- 缺陷:强迫网络去学习两种差异很大光谱的通用特征,表征能力有限,在复杂场景效果一般。
三、作者围绕该问题,是如何构建解决思路的?
整体思路分三步:搭建通用平台→优化融合位置→设计可控的训练范式
1.搭建YOLOv11-RGB 双分支通用框架
将以下6种融合方式做成可插拔模块,统一训练超参,作为公平对比的实验平台,支持YOLO系列、RT-DETR等多种检测器。
| 范式名称 | 核心流程 | 核心优点 | 主要局限 |
| 早期融合 | 输入层直接融合 | 简单直接,保留完整原始信息,信息损失极少 | 易受原始数据噪声干扰,模态间特征差异易引发冲突 |
| 中层融合 | 特征图层级融合 | 主流方案,多模态特征交互充分,融合效果较好 | 易引入冗余特征,融合策略不当可能导致性能下降 |
| 中后融合 | 临近检测头处融合 | 保留更多模态独有特征,兼顾交互与独立性 | 融合节点较晚,模态间信息互补的程度不够充分 |
| 晚期融合 | 结果层面决策融合 | 模态间互不干扰,鲁棒性强,系统运行稳定可靠 | 特征层面无交互,多模态信息利用率较低,性能上限有限 |
| 分数融合 | 置信度分数加权融合 | 工程实现简单,不改变检测框坐标,部署成本低 | 仅利用置信度分数,融合粒度较粗,难以挖掘深层特征 |
| 权重共享 | 骨干网络权重共享 | 模型参数量小,轻量化程度高,推理速度快 | 参数耦合度高,难以单独学习到各模态的特有特征 |
2.优化融合位置:P3单节点中层融合。
P3特征偏向细节、对小目标很关键;P4、P5 高层特征语义性强,跨模态交互收益低。只在 P3 位置做跨模态融合,放弃 P4、P5 的融合,减少冗余特征和噪声引入。
3.提出MCF可控微调策略(核心创新,解决模态失衡)
① 先单独训练优势模态(数据集上表现更好的那一路,比如FLIR数据集单独训练红外分支),训练完成后冻结这个优势分支全部权重;
② 次要模态单独走另一路分支,通过初始权重全0的ZeroConv把次要模态特征加到主分支上;
③ ZeroConv初始输出为0,训练初期完全依靠已经学好的优势模态保底,在训练过程中慢慢学习次要模态的补充信息,不会污染主分支学到的有效特征,从根源避免融合后掉点。
四、从结果(含曲线)上看,作者是如何有力地证明他解决了问题?
作者采用由浅入深的消融实验+多数据集验证+和SOTA对比+特征可视化来证明:
1.单模态基线对照
先在FLIR、LLVIP、M3FD、KAIST、VEDAI五个数据集上,统一超参训练所有YOLO版本的RGB、红外单模基线,确定每个数据集谁是优势模态,作为性能下限。
FLIR数据集仅使用可见光(RGB)的结果:
- FLIR数据集仅使用红外(IR)的结果:
这是一个重要基线,与表2对比发现,红外模型在FLIR上的性能远高于可见光模型(因为数据集多为夜间/恶劣天气)。
LLVIP数据集对比结果(不含预训练)
在行人检测专用数据集LLVIP上,对比各种模型在RGB、IR、RGB+IR、RGB+IR(P3)四种模式下的表现。同样发现纯IR表现极强,传统多光谱融合甚至不如纯IR。
M3FD数据集对比结果(不含预训练)
在白天/多场景数据集M3FD上,情况与FLIR(夜间偏多)不同,这里多光谱模型(RGB+IR)和P3融合模型普遍优于单模态模型,说明融合策略的效果高度依赖于数据集特性。
2.消融融合位置
对比传统三层 Midfusion (P3+P4+P5)与本文 P3 单点 Midfusion的 mAP 指标。实验证明,在 FLIR、LLVIP 这类红外占优数据集,仅 P3 融合优于三层融合,证明过多融合节点引入特征冗余。
FLIR数据集传统多光谱融合(Midfusion)结果
使用P3-P5多级融合策略的结果。部分模型优于纯RGB,但很多不如纯红外。
FLIR数据集P3中层融合(Midfusion-P3)结果
下表为作者提出的单节点(仅P3)融合结果。与表4对比,参数量大幅下降,且大部分模型的AP反而提升了(例如YOLOv11n从73.22%提升至74.41%),证明了“少即是多”。
3.消融训练策略(核心)
对比三组训练方案:①随机从头训练双分支;②COCO 预训练迁移微调;③P3 融合 + MCF 多光谱可控微调。实验发现直接 COCO 迁移联合训练经常打不过优势单模态;而 MCF 策略 mAP 稳定≥优势单模态,不会发生性能退化。
- 迁移学习结果
Fig.5证明了一个关键痛点——加载COCO预训练权重后,多光谱模型(斜线柱)的性能往往不如纯红外模型(因为初始化权重相似导致特征冗余)。同时证明了Midfusion-P3(深蓝柱)在多数情况下优于传统Midfusion。
MCF策略在不同超参数下的比较
Fig.6证明了MCF策略(实心蓝柱)全面超越了纯红外模型(白柱)和普通预训练模型(斜线柱)。说明通过冻结红外主分支、微调可见光分支,能够有效打破多光谱迁移学习的性能瓶颈。
FLIR MCF 超参消融
FLIR数据集综合对比(含预训练权重)
这是核心结果表,对比了最新的YOLOv11系列(纯RGB、纯IR、Midfusion、Midfusion-P3、MCF)以及近年顶会论文的方法(如YOLOv5-CFT、CSAA等)。证明了YOLOv11x-RGBT-MCF以47.61%的AP取得了最优成绩。
4.证明融合策略的差异
在M3FD数据集上对比发现,大多数情况下中期融合优于早期和晚期融合,这为提出P3融合提供了实证基础。
5.与现有 SOTA 方法横向对比,证明整体算法先进性
详见文中Tab.7(FLIR SOTA对比)和Tab.9(LLVIP SOTA对比),通过和 CFT、ICAFusion、Fusion‑Mamba 等近年RGBT检测算法对比 mAP,本文YOLOv11x‑RGBT‑MCF取得最优指标。
6.定性可视化(特征图 + 检测样例),直观解释融合为什么有效
通过以下热力图直观地证明了融合的有效性。例如在烟雾场景中,纯RGB特征丢失了行人,而融合模型(最右列)不仅捕捉到了行人(红框),还清晰显示了车辆,说明多光谱融合成功整合了不同波段的优势。
下图是YOLOv11-RGBT-MCF模型在M3FD和VEDAI数据集上的实际检测效果图,包含了复杂背景、低辨识度、不均匀光照、烟雾、雨天、夜间和低角度拍摄等场景,直观展示了模型在各种极端恶劣条件下的高检测精度和强鲁棒性。
五、这篇论文的缺陷在哪?
强依赖图像配准对齐:本文所有实验的RGB-红外图像都是已经精准配准好的成对图像;如果图像存在偏移、错位(未配准场景),融合效果会大幅下降,算法无法直接使用。
不是自适应方案:需要实验者提前分别跑单模态,人工判断哪个模态是优势模态,手动设置冻结主分支;网络不能自动判断当前场景哪一个模态更可靠,自动切换融合策略。
任务局限:只做目标检测,没有拓展到分割、关键点跟踪等其他多光谱视觉任务。
MCF 带来额外推理分支开销:虽然冻结了主分支,但是推理阶段仍然保留双分支,相比单模态模型,参数量、推理延迟更高,对嵌入式端部署不够友好;
部分数据集上收益有限:在 M3FD 这类两个模态本身都很强、互补性均衡的数据集上,MCF 的提升幅度比较小,提升主要集中在模态差异大的数据集(FLIR、LLVIP)。
六、论文可以借鉴的地方(创新点)?
1.提出 MCF 多模态可控微调策略(核心创新),解决模态不平衡痛点
借鉴点:直接加载可见光的COCO预训练权重到多光谱双分支网络,会导致两个分支提取的特征高度同质化(冗余),性能甚至不如单模态。作者借鉴了ControlNet的思想,提出了MCF策略。
启发:在做多模态融合时,不要理所当然地把各个模态视为同等重要,冻结表现更好的主模态(如红外)网络,利用初始权重为0的Zero Conv2d层,将另一模态(如可见光)的特征注入到冻结模型中。这种“主辅分支+零初始化注入”的微调方式,既保留了预训练模型的强大先验知识,又实现了跨模态信息的有效融合,非常值得在其他多模态(如文本+图像、点云+图像)任务中借鉴。
2.消融得出结论:浅层P3单点中层融合优于 P3+P4+P5 多层融合
借鉴点:传统多光谱融合喜欢在骨干网络的P3、P4、P5多个层级反复融合特征。这篇论文通过实验证明,过多的融合节点反而会引入特征冗余和干扰。
启发:作者提出仅在P3层进行一次融合(Midfusion-P3),在保证甚至提升mAP的同时,大幅降低了模型参数量和计算量。这提醒我们在做多模态/多特征融合时,“在正确的位置融合一次”往往比“到处融合”更有效。
3.高度统一的单阶段检测框架
借鉴点:很多多光谱论文只针对某一个特定模型(如YOLOv5)进行改进。本文构建了一个通用的框架,成功将六种融合模式(早期、中层、中后、晚期、分数、权重共享)适配到了YOLOv3到YOLOv12、RT-DETR、PP-YOLOE等多个主流单阶段模型上。
启发:在工程落地或写论文时,将改进方案模块化、通用化,不仅证明了方法的泛化能力,也大大提升了论文的工程价值和被引用的概率。