【NLP】向量检索模型落地：瓶颈及解法！

作者 | Maple小七 

整理 | NewBeeNLP

今天分享一篇来自EMNLP 2021的论文，讨论了三种简单有效的压缩方法：

• 无监督PCA降维
• 有监督微调降维
• 乘积量化

实验结果表明简单的PCA降维有着很高的性价比，能够以少于3%的top-100准确度损失换取48倍的压缩比，以少于4%的top-100准确度损失换取96倍的压缩比。

• Simple and Effective Unsupervised Redundancy Elimination to Compress Dense Vectors for Passage Retrieval

Introduction

近两年来，以DPR为代表的稠密向量检索模型在开放域问答等领域得到了广泛的应用。虽然DPR能够提供更准确的检索结果，但DPR所生成的向量索引的内存占用是很大的。

比如当我们对维基百科建立索引时，基于倒排索引的BM25仅占用2.4GB的内存，而DPR生成的768维稠密向量则需要占用61GB的内存，这比BM25足足多出了24倍，而这多出的24倍的内存在多个数据集上仅换取了平均2.5%的指标提升(Top-100 Accuracy)。

可以猜测的是，DPR生成的768维稠密向量或许太大了，可能是存在大量冗余的，我们可以尝试用少量的精度损失换取内存占用的大幅下降。针对这个问题，本文探索了三种简单有效的稠密向量压缩方法，包括主成分分析(Principal Component Analysis, PCA)、乘积量化(Product Quantization, PQ)和有监督降维(Supervised Dimensionality Reduction)。

Quantifying Redundancy

首先，我们先确认DPR稠密向量是否存在冗余，作者给出了两个常用指标：PCA的「解释方差比」(explained variance ratio)，以及和的「互信息」(mutual information)。

PCA利用特征分解将一组可能存在相关性的向量变换到线性无关的特征向量构成的的坐标系下，并保留方差较大的方向，丢弃方差较小的方向，而解释方差比则是度量PCA降维效果的指标：

其中是从大到小第个的特征值对应的方差，和分别表示PCA降维前和降维后的维度。该比值说明了保留前个特征向量能保留原始稠密向量方差的比例。

另一种衡量向量冗余度的方法是计算和的互信息，互信息可以根据公式得到：

其中可以利用DPR的优化目标近似计算：

互信息的上界是，为了和PCA对比，我们可以对互信息做标准化。

经过不同程度的PCA降维，解释方差比和标准互信息随向量维度的变化趋势如下图所示。我们可以发现，200维左右是一个甜区，「将768维的向量降维到200维，降维后的向量能够保留90%的方差和99%的互信息，而进一步降维会导致信息量的急速下降。」

Dense Vector Compression

接下来，我们尝试三种简单的方法来压缩稠密向量：

• 「Supervised Approach：」 我们可以简单地在双塔编码器的顶部分别增加两个线性层

  和来降维，在训练时可以冻结下层参数，仅微调线性层，同时我们还可以增加一个正交规范化损失鼓励和互相正交，这能使得降维后的点积相似度和降维前的scale是一致的。

  
  

• 「Unsupervised Approach：」 我们可以将和混在一起，然后对这一向量集拟合一个线性PCA变换，在推理阶段，使用拟合到的PCA变换对DPR生成的向量进行降维。

• 「Product Quantization：」 我们还可以使用乘积量化来进一步压缩向量大小，其基本原理是将维的向量分解成个子向量，每个子向量采用-means量化，并使用比特存储。比如，一个维的向量占用了比特，通过将其分解为个比特的子向量，该向量的大小便压缩到了比特，即原始大小的，平均来说，每个维度所占用的比特数从降低到了。

Experiment & Results

作者在NQ、TriviaQA、WQ、CuratedTREC、SQuAD上测试了DPR的top-准确率(前个召回结果中至少有一个正确的比例)，实验细节可参见原文。

Dimensionality Reduction

我们首先对比有监督降维和无监督降维的表现，其中PCA-*和Linear-*分别是无监督PCA降维和有监督微调降维(仅微调线性层)的结果，而DPR-*表示不冻结下层参数，与线性层联合微调的结果。可以发现，在向量维度较大的时候(、)，无监督PCA的表现更好，当向量维度较小的时候()，有监督微调的表现会更好，然而这时候模型性能下降得也非常明显，因此总体来说无监督PCA更有实用价值。

虽然理论上Linear-*可以学习到PCA-*拟合的线性映射，但想要让参数收敛到一个好的解并不简单。另外，冻结下层参数(Linear-*)比不冻结(DPR-*)的结果都要好，这同样是训练不充分所导致的。综上，在大多数情况下，我们仅需要做简单的线性PCA变换，就能获取很不错的压缩比了。

Product Quantization

乘积量化是一种非常有效的压缩方法，作者在上述实验结果的基础上进一步加入了乘积量化，实验结果如下表所示，其中PQ-2表示经过量化之后每个维度所占用的比特数为。从下表可以看出，PQ-1压缩过于激进了，虽然其压缩比是PQ-2的两倍，但指标下降却不止两倍，这是很不划算的。

综上，我们认为PCA降维加乘积量化是最好的压缩方式，如果我们将指标下降幅度限制在平均4%以内，我们可以使用PCA-128+PQ2将稠密向量压缩96倍，将维基百科的向量索引的内存占用从61GB降低到642MB，同时将推理时间从7570ms降低到416ms。

Hybrid Search

大量研究已表明结合稀疏向量检索(BM25)和稠密向量检索能够提升性能，其中最简单有效的方法是对分数做线性加权求和：

这里我们简单地设定，即稠密检索和稀疏检索等权。

加入混合检索可以进一步提升性能，下图展示了不同压缩方法的检索准确率和索引大小的关系，其中每条曲线从左到右依次为PQ1、PQ2和w/o PQ，图中的黑色虚线为帕累托边界，原始的768维DPR向量并没有落在帕累托边界上，表明其的确有改进的空间。具体来说，「PCA-256+PQ2+Hybrid Search的压缩策略将61GB的索引大小降低到了3.7GB，其Top-100准确率甚至比原始DPR更好(+0.2%)。」

Discussion

限制稠密向量检索模型落地的一大瓶颈就是推理时延和内存消耗的问题，这篇论文通过实验证明了简单的主成分分析加上乘积量化，在辅以稀疏向量检索，就能在保证准确度的前提下大幅减少内存占用，提升检索速度，颇具实用价值。