相同机能情况下，提早减轻 46%，参数减轻 25%。
目标检测系统的标杆 YOLO 系列，再次获得了重磅升级。
自今年 2 月 YOLOv9 发布之后， YOLO（You Only Look Once）系列的接力棒传到了清华大学研讨人员的手上。
上周末，YOLOv10 推出的消息引发了 AI 界的关注。它被认为是计算机视觉领域的突破性框架，以及时的端到端目标检测威力而闻名，通过提供结合效劳和准确性的强大解决方案，延续了 YOLO 系列的传统。
论文地址：https://arxiv.org/pdf/2405.14458
项目地址：https://github.com/THU-MIG/yolov10
新版本发布之后，很多人已经进行了部署测试，效果不错：
YOLO 因为机能强大、消耗算力较少，一直以来都是及时目标检测领域的主要范式。该框架被广泛用于各种实际应用，包括自动驾驶、监控和物流。其高效、准确的物体检测威力使其成为及时识别行人和车辆等任务的理想选择；在物流方面，它有助于库存管理和包裹跟踪，通过 AI 威力帮助人们在很多工作上提高了效劳。
几年来，研讨人员对 YOLO 的架构计划、优化目标、数据增强策略等进行了试探，取得了显著进展。然而，后处理对非极大值抑制（NMS）的依赖阻碍了 YOLO 的端到端部署，并对推理提早产生不利影响。此外，YOLO 中各个组件的计划缺乏全面彻底的检查，导致明显的计算冗余并限制了模型的威力。
YOLOv10 的突破就在于从后处理和模型架构方面进一步提升了 YOLO 的机能 – 效劳边界。
为此，研讨团队首次提出了 YOLO 无 NMS 训练的一致双重分派（consistent dual assignment），这使得 YOLO 在机能和推理提早方面有所改进。
研讨团队为 YOLO 提出了整体效劳 – 准确率启动的模型计划策略，从效劳和准确率两个角度全面优化 YOLO 的各个组件，大大降低了计算开销并增强了模型威力。
大量实验表明，YOLOv10 在各种模型规模上都兑现了 SOTA 机能和效劳。例如，YOLOv10-S 在 COCO 上的类似 AP 下比 RT-DETR-R18 快 1.8 倍，同时参数数量和 FLOP 大幅减轻。与 YOLOv9-C 相比，在机能相同的情况下，YOLOv10-B 的提早减轻了 46%，参数减轻了 25%。
方法介绍
为了兑现整体效劳 – 准确率启动的模型计划，研讨团队从效劳、准确率两方面分别提出改进方法。
为了提高效劳，该研讨提出了轻量级分类 head、空间通道（spatial-channel）解耦下采样和排序指导的块计划，以减轻明显的计算冗余并兑现更高效的架构。
为了提高准确率，研讨团队试探了大核卷积并提出了有效的部分自注意力（partial self-attention，PSA）模块来增强模型威力，在低成本下挖掘机能改进的潜力。基于这些方法，该团队成功兑现了一系列不同规模的及时端到端检测器，即 YOLOv10-N / S / M / B / L / X。
用于无 NMS 训练的一致双重分派
在训练期间，YOLO 通常利用 TAL 为每个实例分派多个正样本。一对多的分派方式产生了丰富的监督信号，促进了优化并使模型兑现了卓越的机能。
然而，这需要 YOLO 依赖于 NMS 后处理，这导致了部署时次优的推理效劳。虽然之前的研讨工作试探了一对一匹配来抑制冗余预测，但它们通常引入了额外的推理开销。
与一对多分派不同，一对一匹配对每个 ground truth 仅分派一个预测，避免 NMS 后处理。然而，这会导致弱监督，以至于准确率和收敛速度不理想。幸运的是，这种缺陷可以通过一对多分派来弥补。
该研讨提出的「双标签分派」结合了上述两种策略的优点。如下图所示，该研讨为 YOLO 引入了另一个一对一 head。它保留了与原始一对多分支相同的结构并采用相同的优化目标，但利用一对一匹配来获取标签分派。在训练过程中，两个 head 联合优化，以提供丰富的监督；在推理过程中，YOLOv10 会丢弃一对多 head 并利用一对一 head 做出预测。这使得 YOLO 能够进行端到端部署，而不会产生任何额外的推理成本。
整体效劳 – 准确率启动的模型计划
除了后处理之外，YOLO 的模型架构也对效劳 – 准确率权衡提出了巨大挑战。尽管之前的研讨工作试探了各种计划策略，但仍然缺乏对 YOLO 中各种组件的全面检查。因此，模型架构表现出不可忽视的计算冗余和威力受限。
YOLO 中的组件包括 stem、下采样层、带有基本构建块的阶段和 head。作者主要对以下三个部分执行效劳启动的模型计划。
轻量级分类 head
空间通道解耦下采样
排序指导的模块计划
为了兑现准确率启动的模型计划，研讨团队进一步试探了大核卷积和自注意力机制，旨在以最小的成本提升模型机能。
实验
如表 1 所示，清华团队所开发的的 YOLOv10 在各种模型规模上兑现了 SOTA 的机能和端到端提早。
该研讨还针对 YOLOv10-S 和 YOLOv10-M 进行了消融实验，实验结果如下表所示：
如下表所示，双标签分派兑现了最佳的 AP – 提早权衡，采用一致匹配度量可以达到最优机能。
如下表所示，每个计划组件，包括轻量级分类 head、空间通道解耦下采样和排序指导的模块计划，都有助于减轻参数数量、FLOPs 和提早。重要的是，这些改进是在保持卓越机能的同时所兑现的。
针对准确性启动的模型计划的分析。研讨人员展示了基于 YOLOv10-S/M 逐步集成准确性启动计划元素的结果。
如表 10 所示，采用大核卷积和 PSA 模块分别在提早最小增加 0.03ms 和 0.15ms 的情况下，使 YOLOv10-S 的机能有了 0.4% AP 和 1.4% AP 的显著提升。
参考内容：
https://visionplatform.ai/yolov10-object-detection/