《Non-local Neural Networks》论文笔记

墨蓝 2023-02-28 04:08 14阅读 0赞

参考代码：

1.  [video-nonlocal-net][]
2.  [Non-local\_pytorch][Non-local_pytorch]

# 1. 概述 #

> 导读：现有的CNN与RNN的网络都有很好的局部相关性（local），但是在一些任务中就需要网络具有较大的感受野范围，因而就需要一种non-local的表达形式来捕获较大范围的信息。这篇文章不同于之前的non-local的方法，它计算的是当前位置在全局中的加权和的形式。而且文章的结构是比较灵活的可以被移植到很多的网络结构中，或多或少地会带来性能的提升（其对于网络量化的影响就还需要进一步试验观察了），这篇文章的方法可以说是比较好的Attention方法了。

# 2. 方法设计 #

## 2.1 问题描述 ##

这篇文章中将non-local的学习过程描述为下面的过程：  
 y i = 1 C ( x ) ∑ ∀ j f ( x i , x j ) g ( x j ) y\_i=\\frac\{1\}\{C(x)\}\\sum\_\{\\forall j\}f(x\_i,x\_j)g(x\_j) yi=C(x)1∀j∑f(xi,xj)g(xj)  
其中，函数 f f f是两个输入的相关性计算，函数 g g g是输入的映射表达，之后 C ( x ) = ∑ ∀ j f ( x i , x j ) C(x)=\\sum\_\{\\forall j\}f(x\_i,x\_j) C(x)=∑∀jf(xi,xj)是归一化的参数。这里的操作带有全局属性的，并不同于卷积中局部的特性。

在计算两个输入相关性的时候有时会采用FC层的形式，这篇文章的方法与FC层的比较归纳为：

*  1）文章的non-local的方法是基于不同位置的，而FC层是使用可学习的参数去映射两者之间的关系；
 *  2）文章的方法输出的是与输入尺寸相同的数据，且对输入输出的维度并没有限制。而FC中由于参数固化了，其对输入输出的维度是有限制的且对位置这样的关系并不敏感；

## 2.2 相关性函数 ##

在上面的共识中存在两个函数 f , g f,g f,g，这两个是文章中的核心函数，分别用于计算两个输入的相似性和对输入进行映射。对此这里对两个函数的采样进行了实验并筛选出适合的相似性计算函数。

**g g g函数：**  
在文章中将这个函数固定为（核心就是相似性的计算函数）： g ( x j ) = W g x j g(x\_j)=W\_gx\_j g(xj)=Wgxj，这里对 W g W\_g Wg使用 1 ∗ 1 1\*1 1∗1的卷积进行替换。

**f f f函数：**  
这个函数就是这篇文章的核心了，对此文章实验了很多相似性计算的方式如下：

*  1）**Gaussian**：这里是使用高斯函数进行计算，相应的可以表示为： f ( x i , x j ) = e x i T x j f(x\_i,x\_j)=e^\{x\_i^Tx\_j\} f(xi,xj)=exiTxj；
 *  2）**Embedded Gaussian**：这里就是在上面的计算过程上给输入添加映射权重 θ ( x i ) = W θ x i \\theta(x\_i)=W\_\{\\theta\}x\_i θ(xi)=Wθxi，因而可以描述为： f ( x i , x j ) = e θ ( x i ) T θ ( x j ) f(x\_i,x\_j)=e^\{\\theta(x\_i)^T\\theta(x\_j)\} f(xi,xj)=eθ(xi)Tθ(xj)；
 *  3）**Dot product**：这里与前面的方法的区别就是去掉了底数，则表示为： f ( x i , x j ) = θ ( x i ) T θ ( x j ) f(x\_i,x\_j)=\\theta(x\_i)^T\\theta(x\_j) f(xi,xj)=θ(xi)Tθ(xj)；
 *  4）**Concatenation**：这里就是将两个输入进行concat起来之后进行映射和激活操作，其可以描述为： f ( x i , x j ) = R e L U ( w f T \[ θ ( x i ) T θ ( x j ) \] ) f(x\_i,x\_j)=ReLU(w\_f^T\[\\theta(x\_i)^T\\theta(x\_j)\]) f(xi,xj)=ReLU(wfT\[θ(xi)Tθ(xj)\])

至于这上面的几种相似性计算方式的性能可以参考下表所示：  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L21fYnVkZHk_size_16_color_FFFFFF_t_70]

## 2.3 Non-local模块 ##

在完成输入样本的相似度计算之后，整体的non-local的框架可以描述为：  
 z i = W z y i + x i z\_i=W\_zy\_i+x\_i zi=Wzyi\+xi  
将其使用常用的操作进行表示可以描述为下图中的形式：  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L21fYnVkZHk_size_16_color_FFFFFF_t_70 1]

# 3. 实验结果 #

**Kinetics性能：**  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L21fYnVkZHk_size_16_color_FFFFFF_t_70 2]  
**其它任务上的性能：**  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L21fYnVkZHk_size_16_color_FFFFFF_t_70 3]

[video-nonlocal-net]: https://github.com/facebookresearch/video-nonlocal-net
[Non-local_pytorch]: https://github.com/AlexHex7/Non-local_pytorch
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L21fYnVkZHk_size_16_color_FFFFFF_t_70]: /images/20230209/c9bbf883f52b4474926537c92e658c33.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L21fYnVkZHk_size_16_color_FFFFFF_t_70 1]: /images/20230209/25f80929be2b499f8d6fbc5bf1a6b9a0.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L21fYnVkZHk_size_16_color_FFFFFF_t_70 2]: /images/20230209/e832fb75ea2d422ab127cebce563c83b.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L21fYnVkZHk_size_16_color_FFFFFF_t_70 3]: /images/20230209/447a6da2e8be4b5ebc26939219c3741c.png