Token导航 LogoToken导航

10万参数模型在COCO击败亿参数检测器

更新时间 2026-10-19来源 科技行者正文 8096字阅读约 26分钟

你可能没意识到这件事有多反常。

过去十年,计算机视觉领域有一条几乎被当成铁律的经验:想让模型更聪明,就得让它更大。从ResNet的两千五百万参数,到ConvNeXt的近亿参数,再到DINOv3的七十亿参数,整个行业都在往一个方向卷——参数越多,见过的图片越多,效果就越好。Meta的DINOv2模型为了学会稳定识别物体,硬生生喂了它1.42亿张图片。

现在有一篇论文说:我们用不到10万参数,在COCO目标检测榜单上跑赢了参数量是我们一千多倍的RF-DETR-2XL。不是接近,是超过——65.3%的mAP50-95,对方是60.1%。

这不是在讲一个"更省资源的小模型"的老套故事。这背后藏着一个很扎心的问题:如果我们一直以来解决视觉问题的方式本身就有点绕远路呢?

图片对机器来说,到底是什么

先说说问题出在哪儿。

当一张照片被输入到今天绝大多数的AI视觉模型里,它首先会变成一堆数字。每个像素点记录一个亮度或颜色值,就是这么简单粗暴。一张30×50像素的小图,在8位RGB编码下,理论上能有256的4500次方种排列组合——这是一个天文数字,比宇宙里的原子数量还夸张。

这就意味着,模型要"看懂"一张图,其实是在做两件事:第一,从这堆几乎是随机排列的数字里,重新猜出这张图原本的结构(哪里是边缘、哪里是同一个物体、哪块和哪块是相连的);第二,才是去完成真正的任务,比如判断这是不是一只猫。

这就好比你去一个完全陌生的城市找一家餐厅,但导航软件给你的不是地图,而是一份写着每栋楼层高、墙体材质、窗户数量的原始建材清单。你得先花大把时间从这堆数据里"猜"出这座城市的街道结构,搞清楚哪条是主干道、哪个转角能拐弯,然后才能开始真正找餐厅这件事。如果一开始给你的就是一张画好的地图,你能省下的时间和脑力是完全不同量级的。

**这篇论文的核心主张,就是给模型一张"地图",而不是一堆原始建材清单。**

这个"地图"就叫TAPe。

> TAPe(Theory of Active Perception,主动感知理论):一种表示图像的新方式,图像不再被拆解成一个个孤立的像素点,而是被描述成一组"T-bit"——每个T-bit对应图像里一批相互关联、承载信息量更大的视觉元素,天生就带着元素之间的相似和关联关系。

这个设计思路的关键在哪?一个像素只知道自己是什么颜色,不知道自己和别的像素有什么关系。而一个T-bit从诞生的那一刻起,就已经记录了"我和周围哪些元素是相关的"。这就相当于给每块建材都提前标注好了"这块砖属于哪面墙,这面墙属于哪个房间",模型不用再自己去猜这些关系,直接拿来用。

论文里给出了一个很有说服力的对照实验。用完全一样的3层CNN架构,大约51.6万参数,在Imagenette数据集(ImageNet的10分类子集)上训练,唯一的区别是输入数据:一组用TAPe表示,另一组用原始像素。结果TAPe那组跑出92%的验证准确率,像素那组只有47%——差了将近一倍。

架构一样,数据量一样,训练方式一样,唯一变量就是"喂给模型的是什么样的输入"。这个实验干净到近乎极端,它想说明的就是一件事:视觉模型的瓶颈,很多时候根本不在网络设计上,而在你给它看的东西本身。

拆掉庞大的检测器,换成一群"专才"协作

光有好的输入表示还不够,TAPe+ML这套系统在架构设计上也做了一次挺大的手术。

早期版本的TAPe+ML,检测部分是一个整体式的大模型——一个模型吃进TAPe表示,自己端到端把检测任务全做完,本质上还是传统"一个大脑管所有事"的思路。到了v3版本,团队彻底换了个做法:把检测任务拆成几个简单的小任务,分别交给几个专门的小模型去做,再用一个"协调员"把它们的输出组织起来。

具体拆成了三个角色。

第一个叫背景模型,专门负责把图里"没意思"的区域(背景、纹理均匀的部分)先过滤掉,只留下可能藏着物体的区域。它是一层一层筛的:第一层能在保留99%真实物体的前提下,排除掉68%的背景;第二层继续筛,又能排除掉剩下背景里的58%。两层筛完,背景只剩17%,而物体还是保留了98%。

第二个叫指针模型,负责在剩下的区域里精确定位物体在哪儿。因为已经放弃了transformer架构里那种"每个位置都要和所有其他位置计算关联"的全局自注意力机制,指针模型用的是更轻量的局部关联方式。

第三个叫原型聚类模型,这是整个系统里最反常识的部分:它不训练一个传统意义上的分类器。传统做法是用梯度下降去调一个线性分类层的参数,TAPe+ML v3用的是k-means聚类,把同一类物体的TAPe元素聚成簇,簇的中心就是这个类别的"代表形象"。新物体来了,不是去问一个训练好的分类器"你觉得这是什么",而是直接去找离哪个簇中心最近,就归为哪一类。

> k-means:一种无监督聚类算法,通过反复计算数据点到簇中心的距离,把相近的数据点自动分成若干组,每组用一个中心点代表。

这三个小模型之上,还有一个协调员统筹全局。它决定先调哪个模型、后调哪个模型,根据数据集的复杂程度自动增减子模型的数量——如果某个损失函数收敛得不好,协调员就往那个环节多加几个子模型。

这个设计思路让我想起一个场景:传统的大型检测器,像是雇了一个什么都懂一点的全能选手去处理所有环节;而TAPe+ML v3更像是组建了一个小分队,背景筛查交给专门盯梁上尘土的人,定位交给专门找目标的人,分类交给专门认脸的人。如果硬要一个全能选手一肩挑,他确实什么都能做,但每样都做得不够精,而且体型必须练得极其庞大才扛得住这么多职责——这正是那些动辄上亿参数的检测器为什么这么"重"的原因。分工之后,每个专才只需要处理自己那一亩三分地,个体可以做得又小又准。

整套系统的总参数量控制在10万以内。作为对比,RF-DETR 2XL有1.27亿参数,是它的一千多倍。

放弃梯度下降,分类准确率反而涨了

这里有个细节值得单独说一说,因为它有点反直觉。

现在几乎所有的深度学习模型,最后那层分类器都是用梯度下降训练出来的,这意味着你得费心去调学习率——学习率设太大,模型学不稳;设太小,训练慢得要命还容易卡在局部最优解里出不来。这种对学习率的依赖,本身就是一种不稳定因素。

> 学习率:梯度下降算法里控制"每一步调整幅度"的参数,直接影响模型训练是收敛还是震荡发散。

TAPe+ML v3的做法是,分类这一步完全不用梯度下降,换成前面说的k-means聚类,配合坐标下降来完成模型校准。这意味着分类这个环节里,不再需要纠结学习率怎么调、调度策略怎么设计——因为这个环节根本没有梯度下降这一步。

论文里给出的消融实验数据是:把依赖学习率的分类头换成这套无梯度方案后,在相同条件下,准确率提升了大约3个百分点。

这就像是,你原本要过一条河,得靠一根竹竿不断试探水深、调整步伐、稍有不慎就会失足打滑,现在换成直接走一座已经建好的、结构固定的桥。桥不需要你临场调整步幅,走多快走多稳都不影响你能不能过河。少了一个变量,系统整体反而更稳,效果还更好。

视频场景检测:10秒钟处理一小时视频

前面说的都是静态图片,论文里还有一个挺有意思的横向对比实验,是在视频场景分割任务上做的。

任务很直接:把电影《星际穿越》的一个片段,按场景切分开来。团队用同一套聚类算法(HDBSCAN),分别搭配13种以上的机器学习模型(包括ConvNeXt、SwinT、ViT、DINOv2等)和6种经典方法(HOG、Sobel、Canny边缘检测、光流、直方图等),再加上TAPe,在完全相同的硬件条件下(一台Intel Xeon E5-2697 v2 CPU,256GB内存,没有用GPU)跑了一遍。

> HDBSCAN:一种基于密度的聚类算法,能自动发现数据里"密集聚集"的区域并分组,不需要提前指定要分成几类。

结果差距大到有点离谱。TAPe建立索引只花了10到11秒,聚类再花1秒左右,整个索引文件不到1MB。而DINOv2 ViT-B/14,用16核CPU跑,建索引花了5220秒(差不多87分钟),索引文件将近10MB。经典方法里的光流更夸张,建索引花了2959秒,聚类阶段又花了8739秒,文件大小飙到3.2GB。

这个差距不是几倍,是几十倍到上千倍的量级。而且效果上,TAPe切出来的场景划分是最稳定的,连《星际穿越》里那些空荡荡的太空场景、光线昏暗几乎看不清物体的画面,TAPe都能准确划分,而DINOv2这类依赖识别"有没有明显物体"的模型,在这种空镜头上反而表现变差。

这就好比找一本书里所有讲到"爱情"这个主题的章节。一种做法是把全书从头到尾逐字读一遍,记住每个字每句话,再回头去分析哪些段落算一类;另一种做法是提前知道每个章节在讲什么核心关系和情节走向,直接按关系归类。前一种方法更"通用",但代价是要处理的信息量巨大;后一种方法轻快得多,前提是你得先有一份靠谱的"关系摘要"。TAPe选的是后一种路子,而它的索引之所以能小到1MB,是因为它从根上就没有把整张画面的像素细节都记下来,只记了真正有辨别力的结构关系。

COCO和LVIS上的硬指标

说了这么多设计思路,回到最实打实的benchmark数据上。

在COCO目标检测任务上,TAPe+ML v3的mAP50是84.7%,mAP50-95是65.3%,参数量小于10万。

> mAP50 / mAP50-95:目标检测领域衡量准确率的核心指标。mAP50是指预测框和真实框的重叠度(IoU)超过50%就算检测成功时的平均准确率;mAP50-95则更严格,要在0.5到0.95之间取10个不同的重叠度门槛分别计算再平均,对定位精度的要求更高。

对比RF-DETR-2XL,mAP50-95是60.1%,参数量1.269亿。对比YOLO11-M,mAP50-95是48.6%,参数量2010万。

推理速度上,在同一块GTX 1070Ti显卡上,batch size为1的条件下,TAPe+ML v3跑一帧要10.8毫秒,比YOLO26s的12.3毫秒还快,更是把RF-DETR-Small的41毫秒甩开了将近4倍。

| 模型 | COCO mAP50 | COCO mAP50-95 | 参数量 | 单帧延迟(ms) |

|---|---|---|---|---|

| **TAPe+ML v3** | **84.7** | **65.3** | **<0.1M** | **10.8** |

| RF-DETR-2XL | 78.5 | 60.1 | 126.9M | 17.2(T4) |

| RF-DETR-M | 73.6 | 54.7 | 33.7M | 4.4(T4) |

| YOLO11-M | 64.1 | 48.6 | 20.1M | 5.1 |

在实例分割任务上,团队选用了LVIS数据集而不是COCO,原因很直接:COCO的分割标注历史上一直是简化多边形,经常出现物体边界没框对、甚至不同物体的标注互相重叠的问题,不适合用来训练精确的轮廓级分割。LVIS的标注精度要高得多,但它有个特殊之处——它是"联邦式"标注,不是每张图里的每个物体都被标注了,只标了一部分。这意味着不能简单把"没被标注的区域"当成背景,因为里面可能有真实存在但没被标的物体。

团队的解法是,拿LVIS当正样本(确定是物体边界的地方),再用全景分割版本的COCO当负样本参考(那里整张图被完整划分,没有遗漏),两者结合起来训练背景判断能力。这套操作只在预训练阶段需要做一次,之后迁移到新数据集就不用再重复了。

在LVIS的Mask IoU/AP@75指标上(预测的分割掩膜和真实物体重叠度超过75%才算合格),没有额外训练的YOLO只能覆盖44%的物体,达标率21%;TAPe+ML在没做额外训练的情况下就能覆盖97%的物体,达标率72%;做完额外训练后,覆盖率能到98.2%,达标率74%。

为什么差距这么大?论文解释说这是架构上的根本区别。YOLO-Seg生成掩膜的方式,是先在一个比原图小得多的分辨率上(比如640×640的图缩到160×160)用32个固定的"原型"组合出一个粗糙的掩膜,再放大插值回原始分辨率。这就好比你想画一幅精细的人像画,却先在一张邮票大小的纸上打草稿,再把这张邮票等比例放大贴到画布上——放大过程里损失的细节是补不回来的。TAPe+ML v3走的是另一条路,直接在像素级别构建轮廓边界,不需要"先缩小画草稿再放大"这一步,自然精度更高。

在COCO实例分割的标准指标上(Mask AP50和Mask AP50-95),对比同样定位为实时高精度分割模型的RF-DETR-Seg-2XL和YOLO26-X-Seg:

| 模型 | Mask mAP50 | Mask mAP50-95 |

|---|---|---|

| RF-DETR-Seg-2XL | 73.1 | 49.9 |

| YOLO26-X-Seg | 71.6 | 46.8 |

| **TAPe+ML(实例分割)** | **80.7** | **58.4** |

整个流程从输入图片到输出最终的检测框和分割掩膜,包括骨干网络、检测、分类、分割、后处理全流程,平均耗时约12毫秒。

换个陌生场景,小模型还能打吗

前面所有的数据都是在COCO、ImageNet这类学术圈里"熟悉"的数据集上跑出来的。但真实世界的应用场景往往完全不是这个样子。

团队专门做了一个工业场景的试点,任务是识别传送带上的矿石堵塞情况。这是个只有两个类别的问题,但难点在于矿石堵塞这种东西,长得完全不像COCO里的人、车、动物、日常物品。摄像角度不标准,光照条件多变,材质纹理也不统一,而且能用来训练的标注数据极少。

这就是所谓的分布外场景。

> OOD(Out-of-Distribution,分布外):指模型面对的数据,其视觉特征、场景分布明显偏离了原始训练数据的分布范围,是检验模型泛化能力的关键场景。

结果很戳人。用标准的YOLO26s,只给30张图训练,准确率只有约28%。这不是模型不行,而是它内部积累的那套"世界知识"本来就是照着COCO那批常见物体的样子长出来的,遇到完全不同的视觉世界,这套知识反而成了拖后腿的偏见。论文里管这个叫"COCO偏见"。

只调整最上层输出层、保持骨干网络冻结的"head-only"模式,在同样30张图的情况下,准确率提升到约45%,比YOLO好,但还是不够用。

真正的转折点出现在允许骨干网络本身也参与调整的"backbone adaptation"模式:同样30张图,准确率跳到约65%,比YOLO直接翻倍还多。样本扩到85张,准确率涨到约85%。再扩到500张,配合非极大值抑制(NMS)这种后处理手段,准确率能冲到85.7%到96%的区间,其中一个类别甚至接近100%。

这组数据讲的其实是一个更普遍的道理:只调整最外面那一层,好比只换了衣服却没换思维方式,当新环境和旧环境差异够大的时候,衣服再合身也没用,得从骨子里跟着环境重新学一遍。如果不做这种彻底的重新适应,那套原本在COCO上训练出来的直觉,遇到陌生场景时反而会把它带偏。

团队还配了一套半自动标注机制,叫auto-annotate,标完10到15张图,系统就能自己在剩下的数据里找相似物体、给出初步标注,人只需要检查和修正,而不用从头标到尾。这套机制对工业场景特别有用,因为专业标注员本身就是稀缺资源。

| 方法 | 数据量 | 准确率 |

|---|---|---|

| YOLO 26s(基线) | 30张 | ≈28% |

| TAPe+ML,head-only | 30张 | ≈45% |

| TAPe+ML,骨干网络自适应 | 30张 | ≈65% |

| TAPe+ML,骨干网络自适应 | 85张 | ≈85% |

| TAPe+ML,骨干网络自适应+NMS | 500张 | 85.7-96% |

ImageNet上的分类实力

分类任务上,TAPe+ML v3在ImageNet-1k上跑出88.1%的Top-1准确率,在ImageNet-ReaL(标注质量更高的修正版)上是89.9%。

对照组是几个当下比较受关注的大模型:DINOv2 ViT-g/14,准确率约86.92%,参数量约11亿;ConvNeXt V2-Large,约87.26%,参数量约1.98亿;DINOv3 ViT-7B/16,约88.4%,参数量约70亿。

在几个专门用来测试模型抗干扰能力的分布偏移测试集上(ObjectNet、ImageNet-V2、ImageNet-R、ImageNet-Sketch等,这些数据集专门收集了和标准ImageNet风格差异很大的图片,用来检验模型是不是真的学到了通用规律,还是只是记住了训练集的样子),TAPe+ML v3的表现也相当稳:

| 数据集 | TAPe+ML v3 | ResNet-50 | ConvNeXt-B | DINOv2 ViT-B/14 | DINOv3 ViT-7B/16 |

|---|---|---|---|---|---|

| ObjectNet | **78.6** | 38.1 | 42.0 | 72.0 | 79.0 |

| ImageNet-Sketch | **75.8** | 30.9 | 55.0 | 62.5 | 71.3 |

| ImageNet-R | **90.3** | 81.3 | 69.0 | 78.8 | 91.1 |

在ImageNet-Sketch这个只用手绘线稿测试模型的数据集上,TAPe+ML v3拿下75.8%,而参数量差不多的ResNet-50只有30.9%——差了将近45个百分点。这个数据集专门考验模型是不是真正理解了"物体的结构",而不是靠颜色和纹理这些表面特征去蒙。差距这么大,某种程度上印证了TAPe这套表示方式确实抓住了更本质的东西。

写在后面

读完这篇论文,最让我意外的不是那些数字本身,而是它背后隐含的一个假设:我们过去十年在视觉领域的"扩大模型"这条路,可能有一部分投入其实是在弥补输入表示的先天缺陷,而不是在解决真正的智能问题。

论文自己也很坦诚地承认了几个限制。第一,小物体的精确边框定位仍然是短板,这大概和COCO数据集里小物体本身样本偏少有关。第二,骨干网络的COCO偏见问题真实存在,分布外场景必须做骨干适配,这个代价不小。第三,TAPe算法的具体实现细节是专有技术,没有公开,这意味着外部研究者暂时没法完全复现或验证其内部机制,只能通过论文报告的输入输出特性去理解它。第四,论文完全没碰3D视觉和视频追踪这类需要时序一致性的场景,虽然团队在视频场景检测实验里展示了TAPe的潜力,但那只是聚类层面的应用,不是真正的时序建模。

这几条限制说得很直白,也让整篇论文的可信度多了几分。一个东西如果什么都好,那通常意味着它藏了什么没说;把弱点摆出来,反而更让人愿意相信剩下的部分是扎实的。

最触动我的一点是那个无梯度分类的设计。我们习惯性地认为,更复杂的优化算法(梯度下降加各种学习率调度策略)代表更先进,但这个案例说明,有时候换一个完全不同的数学框架(聚类替代梯度优化),不仅解决了工程上的稳定性问题,还顺带把效果做得更好了。这提醒我,遇到一个棘手的工程问题时,第一反应不该是"给现有方法打更多补丁",而是该问一句:这个问题的框架本身,是不是可以换掉。

Q&A

Q1:TAPe+ML v3是什么?

A:TAPe+ML v3是一个基于TAPe结构化表示的紧凑型计算机视觉架构,参数量不到10万,却能在ImageNet分类、COCO检测、LVIS实例分割等多个任务上达到甚至超过参数量上亿的主流模型的效果。

Q2:TAPe+ML v3为什么能用这么少的参数做到高性能?

A:核心在于它不直接处理原始像素,而是先把图像转换成结构化的TAPe表示,让相似和关联关系在输入阶段就已经明确,模型不用再自己去猜结构,再配合背景筛选、物体定位、类别原型聚类三个专门小模型协作分工,避免了单一大模型什么都要自己学的负担。

Q3:TAPe+ML v3能不能用在工业场景这类和COCO差别很大的领域?

A:可以,但需要对骨干网络做适配训练。论文里的工业矿石检测试点显示,只调整输出层的效果有限(约45%),而允许骨干网络一起调整后,同样只用30张图训练准确率就能到约65%,扩大到500张图配合后处理能到85.7%到96%。

文章标签应用落地
资讯来源:由AI资讯编辑整理自互联网公开内容,版权归原作者所有,未经许可,不得转载。

继续浏览更多资讯

返回资讯目录

相关资讯

更多