在上一篇的末尾,我们也提到:当 Swin 在学习“约束”的同时,CNN 也在学习“自由”。

实际上,CNN 体系内部早就已经出现了这种“让结构变得更灵活”的趋势并发展出了繁多的落地技术。
因此,在正式展开“ CNN 的 Transformer 化 ” 前,为了不让内容过于割裂,还需要展开一些前置内容,引入一些 CNN 内的一些相关改进来简单过渡
本篇的内容关于池化技术的初步改进:RoI Pooling

1. 基本池化 Pooling#

先回顾一下最基本的池化操作,以最常见的最大池化(Max Pooling)为例:

�(�,�)=max(�,�)∈��(�+�,�+�)

这里的 � 表示池化窗口,我们用最大池化来筛选“最强响应”。

此外,还可以使用平均池化来保留“整体信息”:

�(�,�)=1|�|∑(�,�)∈��(�+�,�+�)

不再展开过多细节,这里附上我们之前介绍池

总的来说,Pooling 的本质是通过下采样来压缩信息,从而减少计算量。
同时,由于其在局部窗口内进行聚合,它实际上引入了一种重要的归纳偏置:

局部不变性先验:在小范围内的变化不会显著影响输出。

也正因为这一点,在不显著改变语义的前提下,Pooling 还可以间接扩大后续网络的感受野。

当然,最基础的 Pooling 也存在明显的局限:

采样位置是完全固定的。

换句话说,它的行为是“刚性的”:窗口位置固定、采样方式固定、不随输入内容发生变化。
总结一下就是不够灵活,这在分类任务中通常问题不大,但在目标检测等任务中就会变得非常明显。
比如目标刚好落在池化窗口边界或发生轻微形变或偏移,此时的 Pooling 可能会丢失部分关键信息甚至直接削弱目标响应。

问题的关键在于:

Pooling 仍然是在“整张特征图上使用统一规则”,但有时我们真正关心的是“某一个局部目标”。

因此,Pooling 的第一步改进就出现在检测任务中,我们继续。

2. Region of Interest Pooling#

Region of Interest Pooling 简称 RoI Pooling,它起源于 15 年的论文:Fast R-CNN
Fast R-CNN 是一个目标检测模型,它首次系统性提出了 RoI Pooling,用于将任意大小的候选框映射为固定尺寸特征。

同样,如果你对检测任务本身的相关概念有些遗忘了,可以在这里回顾:目标定位与特征点检测
首先要说明的是,Fast R-CNN 并不像现在主流的 YOLO 一样是端到端模型,它的目标检测逻辑是先使用物理方法得到候选框,再把候选框映射输入网络进行分类,是一个两阶段模型。
我们具体展开其逻辑如下:

2.1 候选框映射#

image.png


如图所示,第一步就是把物理方法生成的候选框在原图中的相对位置映射到原图卷积后得到的特征图上。

这个映射本质上依赖于一个非常关键的参数: 空间缩放比例(spatial scale)
如果原图尺寸为 �×�,特征图尺寸为 �′×�′,那么:

scale�=�′�,scale�=�′�

假设原图上的 RoI 为:

(�1,�1,�2,�2)

(�1,�1)是左上角坐标,(�2,�2)是右下角坐标
那么映射后就是:

�1′=�1⋅scale�,�1′=�1⋅scale�

�2′=�2⋅scale�,�2′=�2⋅scale�

我们再看看图里的实例:

  • 原图:128×128
  • feature map:6×6
    则:

scale=6128≈0.046875

假设原图中有一个候选框:

(32, 32, 96, 96)

映射到 feature map:

坐标 原图值 计算公式 feature map 值
�1 32 32×6128 1.5
�1 32 32×6128 1.5
�2 96 96×6128 4.5
�2 96 96×6128 4.5

所以得到:

(1.5, 1.5, 4.5, 4.5)

这里就有一个问题:要不要取整?
在 RoI Pooling(Fast R-CNN 原始做法) 中,会进行直接取整,具体如下:

  • 左上角:向下取整 ⌊�1′⌋。
  • 右下角:向上取整 ⌈�2′⌉。

逻辑还是比较清晰的:扩大一点范围,保证不漏。 但同样,这也带来了误差。
由此得到最终结果:

(1.5,1.5,4.5,4.5)→(1,1,5,5)

这就是 RoI Pooling 的第一步。

2.2 划分子区域#

现在,我们得到了将候选框映射到特征图中的区域,下一步是把该区域划分为 �×� 个子区域。

image.png

要说明的是,这里并不是把区域切成 �×� 个正方形,而是按宽和高分别等分,形成网格
对最终划分好的每个子网格,我们就叫它一个 bin 。

展开来说,设当前 RoI 为:

(�1′,�1′,�2′,�2′)

则:

�=�2′−�1′,ℎ=�2′−�1′

划分为 �×� 后,一个 bin 的尺寸就是:

�������ℎ=��,���ℎ���ℎ�=ℎ�

继续上一节的例子:

(1, 1, 5, 5)

所以:

�=4,ℎ=4

设 �=2 则:

�������ℎ=2,���ℎ���ℎ�=2

但实际上,这只是一种最理想的请况,新的问题是:

候选框不是正方形怎么办?最终 bin 的尺寸不是整数怎么办?

第一个问题只要跳出思维定视就好,答案是:一样切
关键在于第二个问题:
 

image.png


如图所示,在真实情况下:

��, ℎ�

很可能不是整数。
例如图中的:

ℎ=5, �=2⇒���ℎ���ℎ�=2.5

而原始 RoI Pooling 给出的答案是和上部分是一样的:

每个 bin 的边界都会再次取整。

其实现方式是,对于每一个方向

  • 起点:向下取整 ⌊⋅⌋ 。
  • 终点:向上取整 ⌈⋅⌉ 。

例如两 bin 的高度方向

  • bin1:

[0, 2.5]→[0, 3]

  • bin2:

[2.5, 5.0]→[2, 5]

这里的关键点是:RoI Pooling 的 bin 划分是在连续空间完成的,而取整是在每个 bin 内独立进行的。

但这种取整策略也会带来相应的问题:
 

image.png


因此,这种取整策略便让 RoI Pooling 的逻辑中出现了局限。
我们先继续整体逻辑,再专门来阐述这个问题。

2.3 区域池化#

最后一步就较为简单了:在每个划分好的 bin 内进行最大或平均池化得到最终结果

image.png

这便是 RoI Pooling 的完整逻辑和其在检测任务中发挥的作用:

任意大小的候选框都可以通过 bin 的划分映射为固定尺寸特征,从而统一维度用于判定。

到这里,你可能会有这样一个问题:

这不就是在图像里挑了一块进行普通池化吗?并没有改变池化的逻辑吧?归根结底只是池化在检测任务中的特化,在哪里体现出“更灵活”了?

我们就这个问题来总述一下这段逻辑:

3. RoI Pooling 的意义和局限#

3.1 什么是“更灵活”的池化?#

如果从计算流程上看,RoI Pooling 似乎只是做了一件很简单的事情:

从 feature map 中截取一个区域,然后做一次固定尺寸的池化。

但它真正的意义并不在“操作复杂度”,而在于:

它第一次在 CNN 中引入了“基于目标的特征对齐机制”。

展开来说,在传统 Pooling 中,我们默认:池化窗口是预先固定的,与输入内容无关。
这时的池化本质就是一个“空间无关的压缩操作”。

而 RoI Pooling 做了一件关键变化:

池化窗口不再固定,而是由“候选目标(RoI)”决定。

从这个角度看,“更灵活”体现在:Pooling 的位置由输入决定、 Pooling 的范围随目标变化 。

因此,它的“灵活性”本质是:从“固定操作”变成了“针对目标的操作”。

当然,这只是针对池化的初步改进,就现在的眼光来看,它的逻辑并不复杂,在所有 CV 任务中的泛用性也并不强。
但这是一个“起点”,我们在其后续的改进中就能越发体会到“灵活”二字。
以此,我们来展开 RoI Pooling 的局限问题。

3.2 RoI Pooling 的对齐问题#

RoI Pooling 的逻辑中存在一个非常根本的问题:它并没有做到完善的几何对齐。

首先,RoI Pooling 至少经历了两次对齐问题:RoI 映射时的取整和 bin 划分时的取整。

image.png

如图所示,原本连续的目标区域,在进入特征图后被“折断”了两次。最终采样的位置并不一定与真实 RoI 对齐。

Logo

汇聚全球AI编程工具,助力开发者即刻编程。

更多推荐