3 个月前

超越最大间隔:少样本目标检测中的类别间隔均衡

超越最大间隔:少样本目标检测中的类别间隔均衡

摘要

少样本目标检测近年来取得了显著进展,其核心思想是利用基础类别对象所学习到的特征表示来表征新类别对象。然而,这一方法中隐含着一个被忽视的关键矛盾:新类别分类与特征表示之间存在内在冲突。一方面,为了实现精确的新类别分类,不同基础类别之间的特征分布应尽可能分离(最大化类间间隔,max-margin);另一方面,为了精确重建新类别,基础类别之间的特征分布应尽可能接近,以减小新类别内部的类内距离(最小化类内间隔,min-margin)。针对这一矛盾,本文提出了一种类别间隔均衡(Class Margin Equilibrium, CME)方法,旨在系统性地优化特征空间划分与新类别重构过程。CME 首先通过全连接层将定位特征解耦,将少样本目标检测问题转化为少样本分类问题。随后,在特征学习过程中引入一种简单而有效的类别间隔损失(class margin loss),为新类别保留充足的间隔空间。最后,CME 通过对抗性最小-最大(adversarial min-max)方式扰动新类别实例的特征,以实现类间间隔的动态均衡。在 Pascal VOC 与 MS-COCO 数据集上的实验结果表明,CME 显著优于两种基线检测器(平均性能提升达 3%~5%),达到了当前最优水平。代码已开源,地址为:https://github.com/Bohao-Lee/CME。

代码仓库

Bohao-Lee/CME
官方
pytorch
GitHub 中提及
yangyucheng000/CME
GitHub 中提及

基准测试

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
超越最大间隔:少样本目标检测中的类别间隔均衡 | 论文 | HyperAI超神经