使用机器学习加速对非结构化数据的查询-第1部分（使用BlazeIt加速聚合和限制查询）

随着强大的 深度神经网络 （DNN）和人工标记服务（我们统称为“Oracle方法”）的广泛使用，我们可以越来越多地对非结构化数据记录（例如，视频、文本）进行自动化 查询 。例如，城市规划人员通过 查询 路边摄像头的视频对车辆进行计数，以了解交通状况。律师可以提取包含雇员/雇主信息的电子邮件（ 关系提取 ）来发现有效信息。

执行此类 查询 的一种简单方法是使用Oracle方法将非结构化数据记录完全转化为结构化信息。例如，对象检测DNN可以从视频的帧中提取对象类型和对象位置，或者基于BERT的DNN可以提取员工/雇主信息。

然而这种传统 查询 方法的运行成本可能非常高：对象检测DNN的执行速度比实时慢十倍，而人工标注可能要花费数十万美元。为了减少此类 查询 的成本，NoScope、概率预测等使用了代理模型（proxy model）的方法，它通过训练类似oracle方法的廉价模型得到代理分数，主要是针对二元预测的即席（ad-hoc）方式。但是要对非结构化数据执行 查询 还有很多工作要做。下面开始介绍我们小组中针对这一问题的几个新项目。

我们将发布一系列博客文章，描述我们最近在对非结构化数据 查询 进行加速优化方面的工作：

• 本文将描述即将在VLDB 2020上发表的最新工作BlazeIt。我们将描述如何加快聚合和限制查询 。
• 在第2部分中，我们将介绍一类新的查询 ：具有统计保证的近似选择查询 （SUPG查询 ）。我们将描述为什么我们需要统计保证，其语义以及这些查询 的有效算法。SUPG也将在VLDB 2020上展出！
• 第3部分将描述基于DNN的可视数据查询 中的系统瓶颈。我们将展示视觉数据的预处理是当前一个主要瓶颈，以及如何解决这一瓶颈。关于这项工作的论文将在VLDB 2021中发布。
• 第4部分将介绍如何为相同数据上的各种查询 建立索引。我们将展示如何使用索引来有效地回答以前的博客文章及更多文章中的所有查询 。

代理分数（Proxy Score）

此前在近似二元预测（approximating binary predicates）的语境 查询 中已经使用了代理模型。这些算法遵循相同的通用策略：使用oracle方法中的标签训练更小更便宜的代理模型。然后代理模型会为每个数据记录生成一个分数，该分数会估计oracle预测的可能性。例如，我们可以训练一个小的DNN来估计汽车是否在视频帧中。

但是许多需求不止是简单地执行二元分类 。如查询 每帧视频是否有汽车存在，并不能统计每帧视频的汽车数量。

为了纠正这个问题，我们引入了二元分类之外的代理模型。本文将重点介绍代理模型，这些代理模型用于将oracle方法产生的任意值近似于非结构化数据记录。在摄取时，我们的系统使用oracle方法处理一小部分记录：然后在 查询 时使用这些记录来训练代理模型以估计oracle的结果。

在查询 中使用代理分数

现在我们可以生成代理分数来 近似计算 统计信息的oracle方法结果，我们如何使用这些分数来回答 查询 ？我们将简要描述如何完成近似聚合和基数限制的选择 查询 。

系统总览

BlazeIt具有两个关键组件： 摄取（离线）组件和查询 处理组件。在离线组件中，BlazeIt将使用oracle方法注释一个非结构化数据记录的示例：这些注释用于训练代理模型。BlazeIt的 查询 处理组件将执行 查询 ，并为每个 查询 训练新的代理模型。下图展示了Blazelt的系统。

系统总览，BlazeIt尝试在受限的情况下尽可能有效地回答 查询 。

近似汇总

我们描述的第一种 查询 类型是加速聚合 查询 ，该 查询 对数据集中的每条记录统计数据进行近似处理（如对每帧视频的汽车数量进行计数）。我们侧重于近似聚合，因为要提供准确的 查询 答案需要穷举执行oracle方法，而这是非常昂贵的。为了避免详尽实现，我们提供了两种 查询 处理算法。

我们证明了可以直接使用代理分数来回答近似聚合 查询 。由于代理分数和基本事实接近，因此我们可以直接汇总分数。例如要计算每条记录的平均值，我们可以对代理分数求和，然后除以记录总数。由于代理模型是通过oracle方法训练的，所以代理和oracle之间的误差将理想地平均化。经证实，直接使用代理分数比回答聚合 查询 的替代方法要有效得多。

虽然直接使用代理分数可能是有效的，但某些应用程序需要 查询 准确性的统计保证。为了满足这种需求，我们可以在近似 查询 处理（AQP）技术的启发下，使用采样技术来加速近似聚合 查询 。通过适当地使用 置信区间 ，我们可以实现 查询 的统计保证。但是标准的AQP技术在采样中不使用代理分数，这是有价值的信息来源。为了利用代理评分，我们将它们用作控制变量，这是一种统计方法，可以减少抽样方差。最后我们将控制变量与始终有效的停止算法结合在一起，该算法使用较少样本且方差较小的样本，称为EBS停止。综合讲，这可以使我们的系统在给定的误差水平下使用更少的样本。下图展示了控制变量和算法概述-算法的关键部分是算法始终有效，并根据样本方差终止。

控制变量示意图。m（t）是真实值，a（t）是代理分数。虽然并不总是精确的，但a（t）可以近似为m（t）。

EBS停止的 伪代码 ，如果满足差异条件，它将 提前停止 。

为了展示我们算法的效用，我们展示了它们在 近似计算 每帧视频的汽车数量上的性能。关于每帧视频是否有汽车的问题，我们将原始方法与使用代理模型的方法进行比较。如下图所示，我们的方 法大大 优于 基准 方法。尤其是已知某汽车在视频帧中出现，并不能了解该汽车是否在视频中普遍存在。

BlazeIts 与详尽注释，二进制检测工作和随机抽样相比，聚合 查询 的性能更高。如图所示，BlazeIt优于所有 基准

基数限制选择

我们描述如何加速的第二种 查询 类型是基数有限的选择 查询 ，用于找到满足某些条件的少量记录。这些 查询 通常用于手动研究异常事件。

为了加快这些 查询 的速度，我们使用代理分数对感兴趣的记录进行排名。尤其是，我们训练一个代理模型来估算感兴趣的数量（例如，一帧中的汽车数量）并根据这些分数进行排名。我们发现，即使此类事件很少发生，代理模型在排名最高的数据记录中也可以具有很高的精度。

下图中显示了算法的性能（有和没有代理模型的效果）和基线。与近似聚合一样，对于异常事件的基数限制选择，我们的算 法大大 胜过传统方法和随机抽样。

与详尽的注释，先前的二元分类工作和随机采样相比，BlazeIt在极限 查询 上的性能更高。如图所示，BlazeIt优于所有基线。

结论

由于 机器学习 的发展，非结构化数据 查询 变得越来越可行。但是部署oracle方法的成本很高，因此执行此类 查询 的费用可能会过高。我们本文中介绍了使用代理得分来加速汇总和限制 查询 的方法，我们希望这些方法可以开始对非结构化数据进行 查询 。在下一篇博文中，我们将介绍如何通过统计保证执行近似选择 查询 。

相关阅读

• Accelerating Queries over Unstructured Data with ML, Part 2 (Approximate Selection Queries with Statistical Guarantees) 31 Aug 2020（https://dawn.cs.stanford.edu/2020/08/31/supg/）
• How do MLPerf v0.7 entries compare on cost? 17 Aug 2020（https://dawn.cs.stanford.edu/2020/08/17/mlperf-v0.7-cost/）
• Selection via Proxy: Efficient Data Selection for Deep Learning 23 Apr 2020（https://dawn.cs.stanford.edu/2020/04/23/selection-via-proxy/）