IT数码 购物 网址 头条 软件 日历 阅读 图书馆
TxT小说阅读器
↓语音阅读,小说下载,古典文学↓
图片批量下载器
↓批量下载图片,美女图库↓
图片自动播放器
↓图片自动播放器↓
一键清除垃圾
↓轻轻一点,清除系统垃圾↓
开发: C++知识库 Java知识库 JavaScript Python PHP知识库 人工智能 区块链 大数据 移动开发 嵌入式 开发工具 数据结构与算法 开发测试 游戏开发 网络协议 系统运维
教程: HTML教程 CSS教程 JavaScript教程 Go语言教程 JQuery教程 VUE教程 VUE3教程 Bootstrap教程 SQL数据库教程 C语言教程 C++教程 Java教程 Python教程 Python3教程 C#教程
数码: 电脑 笔记本 显卡 显示器 固态硬盘 硬盘 耳机 手机 iphone vivo oppo 小米 华为 单反 装机 图拉丁
 
   -> 人工智能 -> 论文阅读《Dense Relation Distillation with Context-aware Aggregation for Few-Shot Object Detection》 -> 正文阅读

[人工智能]论文阅读《Dense Relation Distillation with Context-aware Aggregation for Few-Shot Object Detection》

Background & Motivation

文中认为在此之前的 Few-shot 目标检测方法中的两类存在着不足:

  • 采用元学习的训练策略,同时 reweight 每一类的表征。

  • fine-tune 的方法,用 Novel 类数据微调模型的 Head 或者采用辅助分支来 refine 模型的 RPN 和 Head。

首先,support set 和 query set 之间的关系很难完全的表征出来,很多此前的方法采用的 global pooling 的方法会导致局部细节信息的丢失,模型无法学习各类和边界预测的关键特征。其次,Few-shot 目标检测领域的多尺度问题,也一直没有好的办法来应对。文章的 Motivation 就是为了解决上述不足。

Dense Relation Distillation with Context-aware Aggregation(DCNet)

文章提出了两个模块?Dense relation distillation 和?Context-aware Feature Aggregation,分别来应对上述问题。

第一个模块判断 query image 中的物体与 support set 中的物体是否属于同一个类别,可以看作是《Non-local Neural Networks》的扩展,只不过这篇文章里的方法是 self-attention。文章认为当发生遮挡等现象时,对于细节信息的?distillation 就变得尤为重要。经过这个模块,query 和 support 中相同的特征会被?further activated。

第二个模块用来应对多尺度问题,同时关注局部和全局特征可以保留不同尺度物体的上下文信息。采用了三个尺度,并且通过?attention mechanism 来聚合这三个尺度的特征。

模型的输入是一个 support set 及对应的 mask 和一张 query image 及其标注。跟 Matching Net 的设定一样,目标是找到 query image 在?support set 与之对应的类别并定位。具体的架构基于 Meta-Rcnn,包含两个共享权重的特征提取模块,输出 query image 和 support set 对应的特征图。

Dense relation distillation(DRD)

query 和 support 的每一张特征图都各自经过一个3*3的卷积被编码为 key 和 value,这两个部分的卷积不共享权重(key 的通道域变为特征图的1/8,value 的通道域变为特征图的 1/2)。这两个部分的 key 用来度量 query 和 support 特征的相似度,value 则存放了具体的信息。

Specifically, key and value maps are produced from features, which serve as encoding visual semantics for matching and containing detailed appearance information for decoding respectively.

之后将 key 和 value 输入到 DRD 中,度量相似度的具体做法是将 query 和 support 的?key 逐像素的输入经过两个不同的线性变换,这两个线性变换的参数通过梯度下降更新。

再将算出的每个像素的相似度输入到 softmax 中,将各个位置的相似度归一化,得到最后的相似度度量 W。最后的特征图输出为

对于 support set 中输入的 N 张图像,将得出的 y 直接相加,得到最后的?refined query feature。

Context-aware Feature Aggregation(CFA)

之后 refined query feature 被送入 RPN,产生的 proposal 经过 RoIAlign 后输出尺度分别为4*4、8*8和12*12的特征图。大尺度的特征图用来检测小物体,小尺度的特征图用了来检测物体。

The RoI pooling layer uses max pooling to convert the features inside any valid region of interest into a small feature map with a fixed spatial extent of H×W (e.g., 7 × 7), where H and W are layer hyper-parameters that are independent of any particular RoI.

用下图所示的方法来整合多尺度的特征图:

图中 GAP 代表全局平均池化,Linear 代表全连接层。每个尺度中的两个分支经过 softmax 归一化到0和1之间,最后的输出是这三个尺度输出的加权求和。

Experiment

采用元学习的训练方法:

训练和测试都是单尺度的,query image 最短边是800像素,最长边是1333像素且保持这个比例。微调阶段将最后的全连接层随机初始化,并且不冻结网络的任何部分。

We train our model with a mini-batch size as 4 with 2 GPUs.? ?

PASCAL VOC 数据集上的结果:

各个模块的消融实验:

表中第一和第二行说明了 DRD 的效果。

* denotes CFA module with attention aggregation fashion.?

Based on the plain CFA module, we further propose an attention-based aggregation mechanism to adaptively fuse different RoI features.

对 RoIAlign 的尺度也进行了消融实验:

对 DRD 和 CFA 的效果进行了可视化:

Moreover, different from former meta-learning based methods which performs prediction in a class-wise manner, our proposed DRD module can model relations between query and support features in all classes at the same time.

CFA 可以用来缓解误分类和漏检。

COCO 数据集上的结果:

Conclusion

key 和 value 的区别仅仅是通道域的大小不同,就发挥了不同的作用。

Transformer 的方法整合到 Faster Rcnn 里似乎有很多种方法。

CFA 缓解了误分类和漏检,这一点特性值得关注。

附加

  • RoI Align

详解 Mask-RCNN 中的 “RoIAlign” 作用 / 双线性插值的方法_暖仔会飞的博客-CSDN博客

【温故知新】RoI Pooling、RoI Align、ROI Warping pooling、PS-ROI Pooling、PS-ROI Align、PrROI Pooling_linkstack的博客-CSDN博客

  人工智能 最新文章
2022吴恩达机器学习课程——第二课(神经网
第十五章 规则学习
FixMatch: Simplifying Semi-Supervised Le
数据挖掘Java——Kmeans算法的实现
大脑皮层的分割方法
【翻译】GPT-3是如何工作的
论文笔记:TEACHTEXT: CrossModal Generaliz
python从零学(六)
详解Python 3.x 导入(import)
【答读者问27】backtrader不支持最新版本的
上一篇文章      下一篇文章      查看所有文章
加:2021-09-19 07:59:01  更:2021-09-19 08:00:42 
 
开发: C++知识库 Java知识库 JavaScript Python PHP知识库 人工智能 区块链 大数据 移动开发 嵌入式 开发工具 数据结构与算法 开发测试 游戏开发 网络协议 系统运维
教程: HTML教程 CSS教程 JavaScript教程 Go语言教程 JQuery教程 VUE教程 VUE3教程 Bootstrap教程 SQL数据库教程 C语言教程 C++教程 Java教程 Python教程 Python3教程 C#教程
数码: 电脑 笔记本 显卡 显示器 固态硬盘 硬盘 耳机 手机 iphone vivo oppo 小米 华为 单反 装机 图拉丁

360图书馆 购物 三丰科技 阅读网 日历 万年历 2024年11日历 -2024/11/27 14:25:27-

图片自动播放器
↓图片自动播放器↓
TxT小说阅读器
↓语音阅读,小说下载,古典文学↓
一键清除垃圾
↓轻轻一点,清除系统垃圾↓
图片批量下载器
↓批量下载图片,美女图库↓
  网站联系: qq:121756557 email:121756557@qq.com  IT数码