CVPR'21 | RMNet：又快又好！基于局部特征记忆网络的视频物体分割

原创谢浩哲将门创投今天

收录于话题

#CVPR 2021

14个

CVPR 2021 文章专题

第·15·期

视频物体分割（Video Object Segmentation）的目标是估计视频中未知类别物体的掩膜，它被广泛应用于视频后期处理等场景中。现有基于时空记忆（Space-Time Memory）的方法通过全局特征匹配计算当前帧和过去帧像素级的相似性，并根据该相似性将过去的物体掩膜“迁移”至当前帧。然而这种全局的相似性度量无法很好地区分相似物体，并且造成了较大的计算开销。

为了解决这个问题，我们提出了RMNet，仅提取目标物体所在区域的特征，并在该区域内进行特征匹配，从而显著减少了对相似物体的错配，并提高了计算效率。

Paper：
https://arxiv.org/pdf/2103.12934
Project Page：
https://haozhexie.com/project/rmnet
GitHub：
https://github.com/hzxie/RMNet

一、引言

现有的视频物体分割使用了掩膜传播（Mask Propagation）或者特征匹配的策略估计物体在视频序列中的掩膜。早期基于掩膜传播的方法[1-3]使用光流将物体的掩膜从上一帧传递至下一帧，并使用一个全卷积网络（FullyConvolutional Network）完善估计的物体掩膜。然而基于掩膜传播的方法很容易造成误差累积，特别是当目标物体被遮挡或是漂移时。

近几年，基于特征匹配的方法[4-7]使用全局-全局的特征匹配计算过去帧和当前帧特征相似度和匹配关系。在这些方法中，基于STM（Space-Time Memory）的方法[4-6]将过去帧记忆在网络中，从而更好地应对物体的遮挡和漂移。然而，这些方法也记忆了目标物体之外区域的特征，从而导致了对相似物体的错配，也造成了更高的计算复杂度。

本质上，全局特征所造成的错误匹配可以被分为2种类型，如图1所示：

(I) 在当前帧中的目标物体被错误的匹配至过去帧中的物体（红色实线）

(II) 在过去帧中的目标物体被错误的匹配至当前帧中的物体（红色虚线）

造成这两种类型的错误主要是由于在不必要的区域（即目标物体未出现的区域）进行了特征匹配。事实上，对于每一帧而言，目标物体仅在较小的一个区域内出现，因此在局部区域进行特征匹配是合理的也是必要的。

为了解决这个问题，我们提出了RMNet，仅记忆过去帧中目标物体所在区域的特征，从而有效避免了(I)所描述的错误匹配。为了跟踪和预测当前帧中的目标物体，我们估计了相邻两帧的光流，并将上一帧物体的掩膜通过光流warp至当前帧。通过warp得到的物体掩膜提供了一个大致的目标物体出现区域，从而有效避免了(II)所描述的错误匹配。

图1 全局-全局匹配和局部-局部匹配的正确和错误的特征匹配

二、解决方案

图2 RMNet的总体框架图

所提出的RMNet的总体框架图如图2所示。和STM[4]一样，当前帧被视作查询帧（Query Frame），过去帧以及估计的物体掩膜被用于记忆帧（Memory Frame）存储于记忆网络中。对于STM中的记忆帧和查询帧，整张图像的特征都被存储下来；而在RMNet中，我们仅使用目标物体所在区域的特征，并为记忆帧和查询帧中的目标物体分别生成对应的局部记忆特征和局部查询特征。局部记忆特征和局部查询特征是通过从记忆编码器（Memory Encoder）和查询编码器（Query Encoder）抽取的特征表示（Feature Embedding）和局部注意力图（Regional AttentionMap）点乘得到的。局部记忆特征和局部查询特征都包含一个局部键（Regional Key）和局部值（Regional Value）。

在STM中，时空记忆阅读器（Space-Time Memory Reader）被用于在记忆帧和查询帧中进行全局的特征匹配。而在RMNet中，局部记忆阅读器（Regional Memory Reader）仅在包含目标物体的局部记忆特征和局部查询特征进行局部特征匹配，不仅减少了对相似物体的错配，而且提高了特征匹配效率。局部记忆阅读器的输出被输入至解码器（Decoder）生成目标物体在查询帧的掩膜。

局部特征表示

近期基于STM的方法为过去帧的每张图都构建了全局的特征表示。然而，对于记忆帧，目标物体之外的特征可能导致特征匹配过程中对于相似物体错误匹配，如图1中的红色实线所示。为了解决这个问题，我们引入了局部记忆特征，仅在包含目标物体的区域内进行特征匹配。

具体而言，对于

时刻，在特征尺度上目标物体的掩膜可表示为

，则对于第

个物体的局部注意力图

可通过如下方式计算获得：

其中

和

分别表示目标物体边界框左上角和右下角的坐标。这两个坐标是通过如下公式确定的：

其中,

表示边界框的膨胀像素值，它决定了对于所估计的物体掩膜的错误容忍度。特别地，当第

个物体在

中消失时，我们定义

。给定第

个物体在记忆帧中的局部注意力图

，局部记忆特征中的局部键

和局部值

是由记忆编码器生成的局部记忆特征表示和局部注意力图

点乘得到的。

和局部记忆特征类似，我们使用了局部查询特征缓解查询帧中对于相似物体的错误匹配，如图1中的红色虚线所示。为了尽可能准确地获取当前帧中目标物体所在的区域，我们为该物体跟踪并预测了一个粗略的掩膜

。具体而言，我们将上一帧物体的

通过所提出的TinyFlowNet估计的光流映射到当前帧，得到当前帧目标物体的掩膜

。和局部记忆特征一样，

被用于生成查询帧中第

个物体的局部注意力图

。为了更好地处理对目标物体的遮挡，当目标物体的像素个数小于

时，我们定义

，这将会在查询帧中触发对目标物体的全局搜索。如图3所示，当目标物体消失时，特征匹配区域（用红色边界框表示）将会被扩展至全图；当目标物体再次出现时，特征匹配区域重新聚焦至包含目标物体的区域。这个机制可以有利于基于光流的跟踪，使得网络可以感知目标物体的消失，使得网络可以对物体的遮挡更加鲁棒。和局部记忆特征类似，局部查询特征中的局部键

和局部值

是由查询编码器生成的全局查询特征表示与局部注意力图点乘得到的。

图3 在目标物体被遮挡前后特征匹配区域的变化

局部记忆阅读器

在STM中，时空记忆阅读器被用于度量查询帧和记忆帧的相似度。给定第

个物体记忆特征中的键

和查询特征中的键

，则

和

的相似度为：

其中表示特征键中的通道数，表示记忆帧的数量。令和分别表示在和中的索引值，则在位置上的查询值可被计算为：

其中

表示记忆特征中的值。最终，时空记忆阅读器在

位置上的输出为：

其中

代表查询特征的值；

表示拼接（Concatenation）。

基于局部特征表示，RMNet中使用了局部记忆阅读器仅在包含目标物体的区域内进行局部-局部的特征匹配，如图2所示。和之前工作中[4-6]使用的全局-全局的记忆阅读器相比，局部记忆阅读器可以减少在记忆帧和查询帧中对于相似物体的错误特征匹配。令和分别表示第个目标物体在记忆帧和查询帧中进行特征匹配的区域。在全局-全局的特征匹配中，像素之间的相似度是通过一个大矩阵乘法获得的，即和为特征表示键值中的所有位置。而在局部记忆阅读器中，和分别被定义为：

对于满足或者的位置，它们的相似度被定义为：

令和分别表示查询帧中第个物体所在区域的高度和宽度，和分别表示记忆帧中包含目标物体的区域的最大高度和最大宽度。因此，时空记忆阅读器的时间复杂度为。相比之下，局部记忆阅读器的时间复杂度被减小至。如图4所示，且。时空记忆阅读器本质上是一个非局部神经网络（Non-LocalNeural Network），而它经常因为全局-全局特征匹配所导致的高计算复杂度被人诟病；而所提出的局部记忆阅读器通过局部-局部的特征匹配使得其的时间复杂度显著降低。