如何用张量分解加速深层神经网络？（附代码）

本文为雷锋字幕组编译的技术博客，原标题Accelerating deep neural networks with tensor decompositions，作者为Jacob。

翻译 |  林立宏    整理 |  凡江

背景

在这篇文章中，我将介绍几种低秩张量分解方法，用于在现有的深度学习模型中进行分层并使其更紧凑。我也将分享PyTorch代码，它使用Tensorly来进行在卷积层上的CP分解和Tucker分解。

尽管希望大部分帖子都是可以独立阅读的，关于张量分解的回顾可以在这里找到。Tensorly的作者也写了于Tensor的基础内容非常棒的notebook。这帮助我很好的开始学习这块内容，建议你阅读一下这些内容。

加上裁剪（pruning），张量分解是加快现有深度神经网络的实用工具，我希望这篇文章能让这些内容更加容易理解。

这些方法需要将一个层分解成几个更小的层。尽管在分解后会有更多的层，但是浮点运算次数和权重的总数会变小。一些报告的结果是整个网络的x8倍的速度提升（不针对像imagenet这样的大型任务），或者imagenet中的特定层中x4倍的提升。我的结论是用这些分解方式，我能够获得x2到x4倍的加速，这取决于我愿意牺牲多少的精度。

在这篇文章中我介绍了一些称为裁剪（pruning）的技术以减少模型中的参数数量。在一个数据集上正向传递（有时是反向传递）裁剪（pruning），然后根据网络中激活的一些标准对神经元进行排序。

完全不同的是，张量分解的办法只用到层的权重，假设网络层是参数化的，它的权重能够用一个矩阵或者是一个低秩的张量来表示。这意味这个它们在参数化的网络下效果最佳。像VGG神经网络设计为完全参数化的。另外一个关于参数化模型的例子是使用更少的类别对网络进行微调以实现更简单的任务。

和裁剪（pruning）相似，分解之后通过模型需要微调来恢复准确性。

在我们会深入讨论细节之前，最后一件要说明的事是，虽然这些方法是实用的，并给出了很好的结果，但它们有一些缺点：

• 它们能够在一个线性权重上执行（比如一个卷积或者一个全连接的层），忽略了任何非线性的内容。

• 它们是贪婪，自认为聪明地分解层，忽略了不同层之间的相互作用。

目前还要试图解决这些问题，而且它仍然是一个活跃的研究领域。

截断SVD用于分解完全连接的层

第一份我能找到的使用这个来加速深度神经网络的是在Fast-RNN论文中，Ross Girshick使用它来加速用于检测的全连接层。代码可以在这里找到：pyfaster-rcnn implementation。

SVD概况

奇异值分解使我们能够分解任何具有n行和m列的矩阵A：

S是一个对角矩阵，其对角线上有非负值（奇异值），并且通常被构造成奇异值按降序排列的。U和V是正交矩阵：

如果我们取最大的奇异值并将其余的归零，我们得到A的近似值：

具有作为Frobenius范数最接近于A的秩t矩阵的性质，所以如果t足够大，是A的良好近似。

在全连接层上的SVD

一个全连接层通常是做了矩阵乘法，输入一个矩阵A然后增加一个偏差b：

我们可以取A的SVD，只保留第一个奇异值。

这不是一个完全连接的层，而是指导我们如何实现它作为两个较小的：

• 第一个将有一个mxt的形状，将没有偏差，其权重将取自。

• 第二个将有一个txn的形状，将有一个等于b的偏差，其权重将取自。

权重总数从nxm下降到t（n + m）。

在卷积层上张量分解

二维卷积层是一个多维矩阵（后面用-张量），有四个维度：

cols x rows x input_channels x output_channels.

遵循SVD的例子，我们想要以某种方式将张量分解成几个更小的张量。卷积层转换为几个较小近似的卷积层。

为此，我们将使用两种流行的（至少在Tensor算法的世界中）张量分解：CP分解和Tucker分解（也称为高阶SVD或其他名称）。

1412.6553 使用微调CP分解加速卷积神经网络

1412.6553 Speeding-up Convolutional Neural Networks Using Fine-tuned CP-Decomposition 这篇论文说明了如果CP分解能够用于卷积层的加速，正如我们会看到的，这将卷积层纳入类似移动网络的东西。

他们使用它来加速网络的速度，而不会明显降低精度。在我自己的实验中，我可以使用这个在基于VGG16的网络上获得x2加速，而不会降低准确度。

我使用这种方法的经验是，需要非常仔细地选择学习率，微调以使其工作，学习率通常应该非常小（大约）。

一个秩R矩阵可以被视为R秩和1矩阵的和，每个秩1矩阵是一个列向量乘以一个行向量：

SVD为我们提供了使用SVD中的U和V列来写矩阵和的方法：

如果我们选择一个小于矩阵满秩的R，那么这个和就是一个近似值，就像截断SVD的情况一样。

CP分解让我们推广了张量。

使用CP分解，我们的卷积核，一个四维张量公式，可以近似为一个选定的R：

我们希望R对于有效的分解是小的，但是对保持近似高精度是足够大的。

带CP分解的卷积正向传递

为了传递图层，我们使用输入进行卷积：

这给了我们一个办法来解决这个问题：

1.首先做一个wise（1x1xS）与卷积。这减少了从S到R输入通道的数量。下一步将在较少数量的通道上完成卷积，使其更快。

2.用在空间维度上执行分离的卷积。就像在移动网络中一样，卷积是深度可分的，分别在每个通道中完成。与mobilenets不同，卷积在空间维度上也是可分的。

3.做另一个逐点卷积来改变从R到T的通道数量如果原始卷积层有一个偏差，在这一点上加上它。

注意像在移动网中的逐点和深度卷积的组合。在使用mobilenets的时候，你必须从头开始训练一个网络来获得这个结构，在这里我们可以把现有的图层分解成这种形式。

与移动网络一样，为了获得最快的速度，需要一个有效实现深度可分离卷积的平台。

用PyTorch和Tensorly卷积层CP分解

1511.06530 用于快速和低功率移动应用的深度卷积神经网络的压缩

1511.06530 Compression of Deep Convolutional Neural Networks for Fast and Low Power Mobile Applications 这一篇非常酷的论文，说明了如何使用Tucker分解来加速卷积层来得到更好的结果。我也在基于VGG的参数化网络用了这种加速，比CP分解的精度要好。作者在论文中指出，它可以让我们使用更高的学习率（我用）进行微调。

Tucker分解也称为高阶奇异值分解（HOSVD）或者其他名称，是对张量进行奇异值分解的一种推广。

它认为SVD的推广的原因是的分量通常是正交的，但这对于我们的目的并不重要。被称为核心矩阵，并定义不同的轴如何相互作用。

在上面描述的CP分解中，沿着空间维度的分解导致空间上可分离的卷积。无论如何，过滤器是非常小的，通常是3×3或5×5，所以可分离的卷积并不节省我们大量的计算，而且是一个积极的近似。

Trucker分解有用的性质是，它不必沿着所有的轴（模式）分解。我们可以沿着输入和输出通道进行分解（模式2分解）：

卷积正向传递与塔克分解

像CP分解一样，写一下卷积公式并插入内核分解：

这给了我们以下用Tucker分解进行卷积的配方：

1.与进行点对点卷积，信道从S减少到的数量。

2.用进行正则（不可分）卷积。这个卷积代替了原始层的S输入通道和T输出通道，具有输入通道和输出通道。如果这些等级小于S和T，这就是减少的原因。

3.用进行点对点卷积以回到原始卷积的T个输出通道。由于这是最后一次卷积，所以在这一点上，如果有偏差就加上偏差。

我们如何选择分解行列

一种方法是尝试不同的值并检查准确性。尝试后的启发是，效果很好。

理想情况下，选择行列应该是自动的。

作者提出使用变分贝叶斯矩阵分解（VBMF）（Nakajima等，2013）作为估计等级的方法。

VBMF很复杂，不在本文的讨论范围内，但是在一个非常高层次的总结中，他们所做的是将矩阵近似为低秩矩阵和高斯噪声之和。在找到A和B之后，H是等级的上限。

为了将其用于Tucker分解，我们可以展开原始权重张量的s和t分量来创建矩阵。然后我们可以使用VBMF估计和作为矩阵的秩。

我用这个在Python上实现的VBMF，相信它可以工作。

VBMF通常返回的秩，非常接近我之前，仔细和乏味的手动调整得到的结果。

这也可以用于估计完全连接层的截断SVD加速的等级。

用PyTorch和Tensorly卷积层Tucker分解

总结

在这篇文章中，我们讨论了几个张量分解的方法来加速深度神经网络。

• 截断的SVD可用于加速完全连接的层。

• CP分解将卷积层分解成类似移动网络的东西，尽管它更具侵略性，因为它在空间维度上也是可分的。

• Tucker分解减少了二维卷积层操作的输入和输出通道的数量，并且使用逐点卷积来切换2D卷积之前和之后的通道数量。

我觉得有趣的是网络设计中的常见模式，逐点和深度卷积，自然而然出现在这些分解中！

---

博客原址 https://jacobgil.github.io/deeplearning/tensor-decompositions-deep-learning

更多文章，关注雷锋网，添加雷锋字幕组微信号（leiphonefansub）为好友

备注「我要加入」，To be a  AI  Volunteer ！

雷锋网(公众号：雷锋网)雷锋网

。