TAPEX:通过神经SQL执行器学习的表格预训练

摘要

近年来,语言模型预训练的进展通过利用大规模非结构化文本数据取得了巨大成功。然而,由于缺乏大规模高质量的表格数据,在结构化表格数据上应用预训练仍然是一个挑战。本文提出了TAPEX,通过在一个合成语料库上学习神经SQL执行器来实现表格预训练,该语料库通过自动合成可执行的SQL查询及其执行结果获得。TAPEX通过引导语言模型在多样化、大规模且高质量的合成语料库上模仿SQL执行器,解决了数据稀缺的挑战。我们在四个基准数据集上评估了TAPEX。实验结果表明,TAPEX显著优于以往的表格预训练方法,并在所有数据集上取得了新的最先进结果。具体来说,弱监督的WikiSQL表示准确率提升至89.5%(+2.3%),WikiTableQuestions表示准确率提升至57.5%(+4.8%),SQA表示准确率提升至74.5%(+3.5%),TabFact准确率提升至84.2%(+3.2%)。据我们所知,这是首次通过合成可执行程序进行表格预训练并在各种下游任务中取得新最先进成果的工作。我们的代码可在https://github.com/microsoft/Table-Pretraining找到。

1 引言

预训练语言模型(LMs),如BERT(Devlin等,2019)和BART(Lewis等,2020),在一系列自由形式的自然语言(NL)任务中取得了成功。通过学习大量非结构化文本数据,这些模型展示了理解自然语言句子的惊人能力。受这一巨大成功的启发,研究人员尝试将预训练扩展到结构化表格数据(Herzig等,2020;Yin等,2020;Yu等,2021a;Wang等,2021b;Deng等,2020;2021;Shi等,2021a)。然而,与自由形式的自然语言句子不同,表格数据通常包含丰富且有意义的结构化信息,现有的针对非结构化数据设计的预训练方法并不完全适用。

要在结构化表格数据上应用预训练技术,存在两个关键挑战:(1)如何获得大规模高质量的预训练语料库;(2)如何设计高效的预训练任务以进行表格预训练。对于第一个挑战,现有工作通常收集包含自然语言句子和表格的并行数据作为预训练语料库,因为下游任务通常涉及对自由形式的自然语言句子和表格的联合推理。他们要么从网络上爬取表格及其周围的自然语言句子(Herzig等,2020;Yin等,2020;Deng等,2021),要么在现有表格上合成自然语言句子(Yu等,2021a;Shi等,2021a)。然而,正如Yin等(2020)所指出的,从网络上挖掘的原始数据极其嘈杂,需要复杂的启发式方法进行清理。相反,合成方法更容易控制数据质量,但通常需要专家编写数百个模板,这既昂贵又缺乏多样性。关于预训练任务,现有工作通常采用不同变体的掩码语言建模(MLM)(Devlin等,2019)来引导语言模型学习更好的表格数据表示。例如,TAPAS(Herzig等,2020)使用了全词掩码的MLM,而TABERT(Yin等,2020)提出了掩码列预测(MCP)以鼓励模型恢复掩码列的名称和数据类型。尽管取得了成功,它们仍然主要将表格数据视为文本的结构化格式,这导致其表格预训练需要极大的语料库。所有这些都阻碍了表格预训练的进展。

在这里插入图片描述

本文提出了一种新颖的以执行为中心的表格预训练方法TAPEX(通过执行进行表格预训练)。它通过近似形式语言在表格上的结构化推理过程,解决了上述挑战并实现了高效的表格预训练。结构化推理过程与表格的可执行性相关,即表格本质上能够支持各种推理操作(例如对表格中的某一列求和)。特别是,TAPEX通过预训练语言模型模仿SQL执行引擎在表格上的行为,近似SQL查询的结构化推理过程。如图1所示,通过对表格上的可执行SQL查询进行采样,TAPEX首先合成了一个大规模的预训练语料库。然后,它继续预训练语言模型以输出这些SQL查询的执行结果,这些结果是从SQL执行引擎中获得的。由于SQL查询的多样性可以系统地保证,我们可以轻松合成多样化、大规模且高质量的预训练语料库。

我们的关键见解是,如果一个语言模型能够被预训练以忠实地“执行”SQL查询并生成正确结果,那么它应该对表格有深刻的理解。因此,执行预训练任务在理解表格和推理表格方面可能更高效。据我们所知,TAPEX是第一个通过合成可执行程序探索表格预训练的工作。TAPEX概念简单且易于实现。在本文中,我们将预训练视为序列生成任务,并采用编码器-解码器模型。具体来说,我们使用预训练的编码器-解码器语言模型BART(Lewis等,2020)作为骨干。此外,我们通过两个基本的下游任务评估TAPEX的有效性:基于表格的问答(TableQA)和基于表格的事实验证(TableFV)。为了使下游任务的微调充分利用TAPEX,我们使用编码器-解码器序列生成范式重新表述这些任务。我们在四个知名基准数据集上评估TAPEX。实验结果清楚地表明,TAPEX能够为这些数据集带来显著且一致的改进。例如,TAPEX在WIKITABLEQUESTIONS数据集上比BART绝对提升了19.5%。此外,即使在小规模的预训练语料库下,TAPEX也表现出强大的效果,展示了其高效性。最后,TAPEX在所有实验基准上取得了新的最先进结果,大幅超越了以往的方法,包括在数据处理中使用多种启发式方法的复杂表格预训练方法。我们将公开代码、模型和数据,以促进未来研究。

2 下游任务的微调

在深入探讨我们提出的表格预训练细节之前,我们首先描述如何利用编码器-解码器序列生成范式来解决下游任务的微调问题。在本节中,我们首先介绍两个与表格相关的基本下游任务的背景:基于表格的问答(TableQA)和基于表格的事实验证(TableFV)。然后,我们将详细阐述我们的生成式微调方法。

在这里插入图片描述

##2.1 下游任务的形式化描述

如第1节所述,下游任务通常涉及对自由形式的自然语言(NL)句子和表格的联合推理。因此,下游任务的示例通常包含一个自然语言句子 x x x 和一个(半)结构化表格 T T T 作为模型输入。每个自然语言句子由 K K K 个词元组成,记为 x = x 1 , x 2 , ⋯   , x K \mathbf{x} = x_1, x_2, \cdots, x_K x=x1,x2,,xK,而每个表格 T T T M M M { r i } i = 1 M \{r_i\}_{i=1}^M {ri}i=1M 组成,其中每一行 r i r_i ri 包含 N N N 个单元格值 { s ( i , j ) } j = 1 N \{s_{(i,j)}\}_{j=1}^{N} {s(i,j)}j=1N。每个单元格 s ⟨ i , j ⟩ s_{\langle i,j\rangle} si,j 包含一系列词元,并对应一个表头 c j c_j cj。至于输出,不同任务之间存在差异。在本文中,我们主要关注基于表格的问答(TableQA)和基于表格的事实验证(TableFV)。

TableQA 的目标是从表格内容中检索信息以回答用户的问题,因此其输出要么是一组单元格值,要么是通过聚合函数(例如求和 SUM)在选定表格区域上计算出的数值。值得注意的是,对于半结构化表格,答案可能并非精确的表格单元格值,而是其规范化形式(例如,将“2k”转换为“2000”),这使得下游任务更具挑战性(Oguz 等,2020)。

对于 TableFV,输出是一个二元决策,即“支持”(entailed)或“拒绝”(refused),用于指示自然语言句子是否符合表格所表明的事实。

2.2 生成式微调

在本节中,我们提出了一种用于下游任务微调的生成式方法。与以往的工作不同,我们将 TableQA 和 TableFV 都建模为序列生成任务,并利用生成式语言模型(LMs)自回归地生成输出。以 TableQA 为例,给定一个自然语言问题,我们的方法通过逐词解码的方式生成答案。

架构
我们的方法理论上适用于任何能够生成序列的语言模型,例如 GPT-3(Brown 等,2020)和 UniLM(Bao 等,2020)。在我们的实验中,我们基于 BART(Lewis 等,2020)实现了我们的方法,BART 是一种广泛使用的预训练编码器-解码器模型。BART 遵循标准的序列到序列 Transformer 架构(Vaswani 等,2017),并将 ReLU 激活函数修改为 GeLU。它通过破坏句子(即随机采样长度可变的片段并用单个 [MASK] 标记替换每个片段)进行预训练,然后优化重建损失。关于层数,我们在实验中采用了 BART L a r g e _\mathrm{Large} Large 配置,即编码器和解码器各使用 12 层。

模型输入
如图 2 所示,输入包含一个自然语言句子及其对应的表格。编码自然语言句子相对简单,而编码表格则较为复杂,因为表格具有潜在的结构。在实践中,我们将表格展平为一个序列,以便直接输入模型。通过插入几个特殊标记来指示表格的边界,展平的表格可以表示为:
T ∗ = [HEAD] , c 1 , ⋯   , c N , [ROW] , 1 , r 1 , [ROW] , 2 , r 2 , ⋯   , r M . T^* = \text{[HEAD]}, c_1, \cdots, c_N, \text{[ROW]}, 1, r_1, \text{[ROW]}, 2, r_2, \cdots, r_M. T=[HEAD],c1,,cN,[ROW],1,r1,[ROW],2,r2,,rM.
其中,[HEAD] 和 [ROW] 是分别指示表头和行区域的特殊标记,[ROW] 后的数字用于指示行索引。值得注意的是,我们还使用竖线 | 分隔不同列的表头或单元格。最后,我们将展平的表格 T ∗ T^* T 前缀自然语言句子 x x x,并将它们输入模型编码器。

模型输出
在编码器的注意力机制基础上,解码器负责建模 TableQA 和 TableFV 的输出。

对于 TableQA,输出是由逗号分隔的答案的拼接,解码器以自回归的方式生成这些答案。通过这种方式,我们的模型可以轻松支持 TableQA 中(几乎)所有的操作符及其组合。 对于 TableFV,与 BART 处理序列分类任务的方式类似(Lewis 等,2020),相同的输入被同时送入编码器和解码器,并在解码器最后一个词元的隐藏状态上使用一个二元分类器来生成输出。 值得注意的是,我们的方法可以轻松扩展到其他与表格相关的任务,方法类似。

在这里插入图片描述

微调策略
由于我们的方法可以在同一架构上执行各种下游任务,因此可以轻松实现多任务学习。为此,我们探索了两种微调方式:一种是常规微调,另一种是多任务微调。前者是在每个单独的下游任务上对模型进行微调。后者受到 TAPAS(Herzig 等,2020)和 T5(Raffel 等,2020)的启发,首先在相关或类似的中间下游任务上对模型进行微调,然后继续在目标下游任务上进行微调。

讨论
我们的方法具有以下几个优点:

  1. 灵活性:由于编码器-解码器模型强大的表达能力,我们的方法可以轻松适应(几乎)任何类型的输出。
  2. 便捷性:我们的方法不需要对预训练语言模型进行任何修改(例如,特定于表格的掩码),并且可以以端到端的方式进行训练。
  3. 可迁移性:由于我们将下游任务形式化为序列生成任务,这使得不同任务可以共享相同的训练协议,因此我们的方法可以轻松实现多任务微调。

3 通过执行进行表格预训练

如第 1 节所述,TAPEX 通过训练语言模型模仿 SQL 执行引擎的行为来实现高效的表格预训练。在本节中,我们从预训练任务和预训练语料库两个方面说明如何进行表格预训练。

3.1 预训练任务

在自然语言预训练中,MLM(掩码语言建模)任务被广泛使用,现有工作通常采用重建任务进行表格预训练。它们通常以损坏的表格和自然语言句子作为输入,并尝试恢复损坏的部分,以加强自然语言句子与表格之间的联系。尽管这些预训练任务表现良好,但由于通常需要极大的预训练语料库,它们的效率往往较低。

为了设计高效的表格预训练任务,我们认为关键在于表格的可执行性。也就是说,结构化表格使我们能够通过编程语言(如 SQL 查询)对它们执行离散操作,而非结构化文本则不具备这种特性。基于此,TAPEX 采用 SQL 执行作为唯一的预训练任务。如图 3 所示,TAPEX 的预训练过程与上述生成式微调的过程类似。给定一个可执行的 SQL 查询和一个表格 T T T,TAPEX 首先将 SQL 查询和展平的表格 T ∗ T^* T 拼接并输入模型编码器。然后,它通过现成的 SQL 执行器(例如 MySQL)获取查询的执行结果,作为模型解码器的监督信号。直观地说,预训练过程是鼓励语言模型成为一个神经 SQL 执行器。我们相信,如果一个语言模型能够被训练为忠实地“执行”SQL 查询并生成正确的结果,那么它应该对表格有深刻的理解。

3.2 预训练语料库

合成预训练语料库对于表格预训练非常重要。通常,有两个关键因素:表格来源和 SQL 查询采样策略。

在这里插入图片描述

表格来源
根据 Yin 等人(2020)的先前工作,我们选择公开可用的半结构化表格作为表格来源。然而,与 Yin 等人(2020)需要数百万个原始表格不同,TAPEX 即使在仅有几千个表格的情况下也能表现良好。因此,我们没有从网络上获取嘈杂的表格并通过启发式方法过滤它们,而是直接从现有的公共数据集中挑选高质量的表格。具体来说,我们从 WIKITABLEQUESTIONS(Pasupat & Liang,2015)的训练集中随机选择了近 1,500 个表格作为预训练语料库的表格来源。需要注意的是,我们用于预训练的表格与所有下游任务的开发集和测试集中使用的表格没有重叠,因此不存在数据泄露问题。

查询采样
关于多样化 SQL 查询的采样,文献中有多种选择。我们可以根据概率上下文无关文法(Wang 等,2021a)采样 SQL 查询,也可以在不同表格上实例化 SQL 模板(Zhong 等,2020a)。在我们的实验中,我们采用后者,其中 SQL 模板是从 SQUALL 数据集(Shi 等,2020b)中自动提取的。一个 SQL 模板的示例是:SELECT num1 WHERE text1 = val1,其中 num1text1 分别对应数值列和文本列,val1 是指与列 text1 相关的某个单元格值。给定一个 SQL 模板,在每次实例化时,我们从采样的表格中均匀采样表头和单元格值来填充模板,形成一个具体的 SQL 查询。值得注意的是,执行结果为空的结果会被丢弃,因为空结果不能反映表格的可执行性信息。通过这种方式,我们可以获得一个高质量的大规模预训练语料库。

4 实验

在本节中,我们在不同的下游任务上评估 TAPEX 以验证其有效性。

数据集和评估
我们在弱监督的 WikiSQL(WIKISQL-WEAK)(Zhong 等,2017)、WIKITABLEQUESTIONS(Pasupat & Liang,2015)、SQA(Iyyer 等,2017)和 TABFACT(Chen 等,2020)上评估我们方法的性能。与仅需要对表格单元格值进行过滤和可选聚合的 WIKISQL-WEAK 相比,WIKITABLEQUESTIONS 需要更复杂的推理能力。SQA 是一个对话式基准测试,要求我们的方法能够建模对话上下文。数据集详细信息见附录 A。对于 TableQA 数据集,评估指标是表示准确率(denotation accuracy),它检查预测的答案是否与真实答案一致。值得注意的是,我们在 WIKISQL-WEAK 上使用 TAPAS(Herzig 等,2020)提供的答案注释进行评估,因为官方评估脚本中近 2% 的答案是不正确的。对于 TABFACT,评估指标是准确率,通过计算正确预测的百分比得出。

实现细节
我们基于 fairseq(Ott 等,2019)实现了我们的方法。在预训练期间,我们为 TAPEX 合成了多达 500 万对 SQL 查询及其执行结果。在下文中,除非明确说明,所有实验结果默认在 500 万对的设置下进行评估。我们的预训练过程最多运行 50,000 步,批量大小为 256。在 8 个 Tesla V100 GPU 上完成预训练大约需要 36 小时。最佳预训练检查点是根据验证集上的损失选择的。对于所有下游数据集,微调过程最多运行 20,000 步,批量大小为 128。对于预训练和微调,学习率均为 3 × 1 0 − 5 3×10^−5 3×105

在这里插入图片描述

4.1 主要结果

表1、表2、表3和表4分别总结了各种模型在WIKISQL-WEAK、WIKITABLEQUESTIONS、SQA和TABFACT数据集上的实验结果。对于所有数据集的开发集和测试集,我们报告了我们方法在五次随机运行中的中位性能。

WIKISQL-WEAK如表1所示,TAPEX在所有基线模型中表现最佳,领先优势显著。在WIKISQL-WEAK的测试集上,TAPEX的表征准确率为89.5%,比BART高出3.7%,比之前的最佳表现高出2.3%。这一结果具有重要意义,因为之前的最佳模型已经使用了执行引导解码。简而言之,TAPEX在著名的WIKISQL-WEAK基准测试中实现了新的最先进的结果。

WIKITABLEQUESTIONS在更具挑战性的WIKITABLEQUESTIONS上,TAPEX也取得了新的最先进的表征准确率,达到了57.5%,比之前的最佳系统高出4.8%(见表2)。与此同时,我们发现单独使用BART仅能达到38.0%的表征准确率,远低于之前的预训练模型的表现。我们推测,这一性能下降可能是由于WIKITABLEQUESTIONS的训练数据量相对较小,导致BART在适应表格结构时遇到更多挑战。

在这里插入图片描述

然而,TAPEX相比BART在低数据量情况下带来了19.5%的显著提升,表明在数据较少的情况下,TAPEX带来的改进通常更为显著。

SQA 表3展示了各种模型在SQA测试集上的表现,其中TAPEX再次在对话层面(48.4%)和句子层面(74.5%)获得了新的最先进的表征准确率。这一改进令我们感到惊讶,因为SQA是一个对话数据集,而我们的预训练任务是无上下文的。同时,TAPEX在SQA上相对于BART的显著改进再次验证了TAPEX缓解低资源问题的观察。

TABFACT
除了TableQA,TAPEX在TableFV上的表现也非常出色。如表4所示,TAPEX在TABFACT的所有子集上均取得了新的最先进的结果。例如,它在Testcomplex上超越了之前的最佳系统4.0%。这一结果表明,TAPEX赋予了BART通用的表格理解能力,这些能力可以适应不同的下游任务,无论这些任务是否与TAPEX的预训练任务高度相似。

总体结果 在四个数据集上的实验结果表明,TAPEX能够广泛提升模型在理解表格方面的能力,特别是在低数据量情况下。

4.2 多任务结果

如§2.2所讨论的,我们的方法可以轻松进行多任务学习,从而为下游任务带来好处。为了验证这一点,我们进行了多任务微调实验,得出了以下发现:(1)当以BART初始化时,多任务微调显著提高了目标任务的性能;(2)当以TAPEX初始化时,多任务微调的提升趋于边际化,表明多任务学习获得的大部分“技能”(宽泛来说)可以通过我们的表格预训练获得。详细结果可见附录B。

5 分析

在本节中,我们将从多个方面仔细分析我们的方法。此外,我们还进行了探索性分析,以为未来的工作提供更多见解,相关内容可以在附录C中找到。

通过预训练执行SQL为了了解TAPEX在预训练后执行SQL的表现,我们分析了其在近20,000个未见表格上的SQL查询执行情况。总体来说,SQL执行的准确率相对较高,TAPEX正确“执行”了89.6%的SQL查询。特别地,TAPEX在Filter、Aggregate和Superlative操作符上的表现更好,表明它在表格单元选择和表格聚合方面具有较高的准确性。对于Arithmetic和Comparative操作符,TAPEX也表现得很好,展示了其在表格中的数值推理能力。总之,TAPEX已经学会了成为一个具备良好选择、聚合和数值能力的神经SQL执行器。

通过预训练理解表格为了了解TAPEX是否有助于下游任务更好地理解表格,我们对未经过微调的TAPEX在WIKITABLEQUESTIONS样本上的自注意力进行了可视化和分析。如图4所示,TAPEX似乎更关注与单元格对应的行和表头。从图4中的例子来看,注意力权重表明“adrian lewis”与第一列“player”以及整个第三行紧密相关,这些是“adrian lewis”在结构化表格中的位置。

通过预训练进行表格推理为了了解TAPEX是否能改善表格推理,我们将TAPEX与BART在500个随机选取的问题上的表现进行了比较,并在表5中进行了人工分析。可以发现,TAPEX在所有操作符上的表现都显著提高,这表明它确实增强了BART在文本和表格联合推理方面的能力。
在这里插入图片描述

在这里插入图片描述

预训练语料库的规模
图5展示了在不同规模的预训练语料库下的下游任务表现。可以看出,即使我们的预训练语料库是合成的,扩大预训练语料库的规模通常会带来积极的效果。这一观察类似于语言建模中的现象(Brown等,2020):预训练语料库越大,下游任务的表现越好。通过对不同数据集的比较,我们可以发现,对于像WIKISQL-WEAK这样的简单任务,扩大预训练语料库带来的提升逐渐趋于边际效应,而对于像TABFACT这样的复杂任务,提升依然显著。同时,低数据量情况下的下游数据集通过增加预训练语料库也呈现出积极的趋势。总之,当下游任务较为困难或下游数据集较小的时候,语料库的规模尤为重要。

预训练效率
如§1所述,现有的表格预训练方法的效率较低,因为它们通常需要一个极其庞大的语料库。因此,以WIKITABLEQUESTIONS为例,我们比较了TAPEX与TAPAS(Herzig等,2020)、TABERT(Yin等,2020)和GRAPPA(Yu等,2021a)的预训练效率。值得注意的是,GRAPPA的一部分预训练语料库来自人工标注的高质量平行数据。如图6所示,TAPEX在使用远小于其他方法的预训练语料库时,依然能取得非常有前景的表现,表明我们提出的SQL执行预训练任务比其他表格预训练任务更为高效。

局限性
我们方法的第一个局限性是,它无法理想地处理大表格。如上所述,我们采用了表格扁平化技术来表示表格。当表格相对较小时,这种方法效果良好,但当表格过大无法放入内存时,它就变得不可行。在实际应用中,我们可以通过移除一些不相关的行或列来压缩表格,但这会降低下游任务的表现。第二个局限性是,文本到SQL的任务无法从我们提出的表格预训练中受益。我们曾尝试将TAPEX应用于文本到SQL任务,其中输入保持不变,输出转换为SQL。然而,TAPEX在该任务上并未显示出相较BART的显著优势。我们将其归因于两个因素:首先,我们的合成预训练语料库对语义解析中至关重要的“基础对接”没有贡献(Liu等,2021);其次,TAPEX学到的表格推理能力(例如聚合)在SQL生成中可能并不必要。例如,模型即使不知道“sum”的数学含义,也能理解NL短语“total”作为聚合函数“sum”。

6. 相关工作

表格预训练
与我们最相关的工作是表格预训练,其关键因素包括预训练语料库和预训练任务。关于预训练语料库,大多数先前的工作几乎都收集了自然语言(NL)-表格数据来进行表格预训练。它们要么挖掘了大量表格及其自然语言句子上下文(Yin等,2020;Herzig等,2020),要么利用人工标注的平行NL-表格数据集进行预训练(Deng等,2021;Yu等,2021a),或者使用人工编写的模板合成了NL-表格语料库(Yu等,2021a;Eisenschlos等,2020)。我们的工作与它们不同,因为我们是首个使用纯合成的SQL-表格数据进行表格预训练的研究,这使得我们能够自动合成一个多样化、大规模且高质量的预训练语料库。关于预训练任务,现有的工作提出了几种预训练任务,例如:列预测遮蔽(Mask Column Prediction,Yin等,2020)、单元格级多选填空(Multi-choice Cloze at the Cell Level,Wang等,2021b)和结构对接(Structure Grounding,Deng等,2021)。与所有这些工作不同,我们提出了一种新的SQL执行任务来进行表格预训练。

表格与文本的联合理解
由于我们的实验主要集中在TableQA和TableFV任务上,我们的方法与这些任务的先前方法也有紧密的关系。对于TableQA,先前的工作几乎将其表述为一个弱语义解析任务(Liang等,2018;Wang等,2019a;Guo等,2021),这些工作通常使用强化学习来优化表格上的语义解析器。虽然这些解析器可以生成逻辑形式(例如SQL),但由于庞大的搜索空间和存在虚假的程序(Goldman等,2018),它们在训练时会遇到困难。此外,最近有一种有前景的工作方向(Mueller等,2019;Herzig等,2020),该方向旨在回答自然语言句子而不依赖于逻辑形式。这些工作通过选择单元格值并可选地应用聚合操作符来预测答案。它们容易训练,但建模能力有限。例如,它们难以支持复合聚合操作符,如max(Year) - min(Year)。与这些工作不同,我们的方法使用生成模型来处理TableQA,并且能够同时享受端到端训练和灵活性。对于TableFV,先前的工作通常使用有限扩展性的专用架构(Shi等,2020a;Yang等,2020;Shi等,2021b)。例如,Zhong等(2020b)利用图构建机制、语义解析器和语义组合模型来捕捉自然语言句子与表格之间的连接。尽管该方法在TableFV上效果良好,但它不易应用于其他表格相关任务。与这些方法相比,我们的方法能够在相同架构下很好地适用于各种下游任务。

7. 结论

在本文中,我们提出了TAPEX,一种以执行为中心的表格预训练方法,其语料库通过抽样SQL查询及其执行结果自动合成。TAPEX通过在一个多样化、大规模且高质量的合成语料库上学习神经SQL执行器,解决了表格预训练中的数据稀缺问题。四个下游数据集的实验结果表明,TAPEX大幅度超越了先前的表格预训练方法,并在所有数据集上都达到了新的最先进结果。我们的工作为通过在合成可执行程序上进行预训练来开发结构化数据开辟了道路,这一方法在概念上简单,并且具有很大的潜力,能够扩展到其他研究领域(例如知识库)。

A. 下游数据集

数据集的统计信息见表6,表7展示了我们模型的输入和输出示例。需要注意的是,SQA是一个对话基准,我们直接将历史记录和第i个问题拼接在一起作为输入中的“句子”部分(x),正如Liu等(2020)所做的那样。

B. 多任务结果

表8展示了在§2.2中提到的多任务微调的完整实验结果。需要注意的是,我们选择了WIKISQL-WEAK和TABFACT作为迁移源,因为它们的训练数据相对丰富。

C. 探索性分析

在本节中,我们进行了一项探索性分析,以提供更多对未来工作的洞察。具体而言,我们探索了两个有趣的研究问题:(1)预训练中SQL查询的难度如何影响下游任务的表现?(2)在预训练中使用自然语言句子是否比使用SQL查询更好?

C.1 预训练中SQL查询难度的影响

SQL难度标准
受到Yu等(2018)的启发,我们假设SQL查询的难度可以通过SQL元素的数量来衡量。一个元素可以是SQL关键字(例如SELECT),也可以是表格模式(即标题或单元格值)。在实践中,我们通过一个现成的SQL解析器来获取SQL查询的元素,该解析器为每个SQL查询返回一系列SQL元素。根据经验,我们将包含 ≤ 6 \leq 6 6个元素的SQL查询归类为Easy,包含 > 6 > 6 >6 ≤ 14 \leq 14 14个元素的SQL查询归类为Medium,包含 > 14 > 14 >14 ≤ 20 \leq 20 20个元素的SQL查询归类为Hard,其余的归类为ExtraHard。不同难度级别的SQL查询示例可以在表9中找到。基于SQL难度标准,我们将SQUALL(Shi等,2020b)中的模板分为四个难度级别,并从Easy级别( ≤ \leq Easy)到ExtraHard级别( ≤ \leq ExtraHard)逐步添加它们来构建预训练语料库。值得注意的是,为了避免预训练规模的影响,我们保持上述预训练语料库中的示例数量相同。

下游性能
实验结果如图7所示。可以看出,在大多数情况下,将更难的SQL查询添加到预训练语料库中是有帮助的。例如,与 ≤ \leq Easy级别的查询相比, ≤ \leq Medium级别的查询在下游任务的表现上带来了持续的改善(例如,在WIKITABLEQUESTIONS数据集上提高了10.6%)。同时,我们也注意到,在Medium级别之后,SQL查询难度的影响变得不那么显著。在TABFACT数据集中,涉及Extra-Hard级别的SQL查询进行预训练甚至略微损害了性能。
在这里插入图片描述

细粒度分析
为了从细粒度的角度理解影响,我们将WIKITABLEQUESTIONS开发集中的问题分为四个难度级别,并借助SQUALL提供的WIKITABLEQUESTIONS问题的SQL查询注释。所有细粒度的实验结果如图8所示。我们可以看到,随着更难的SQL查询的加入,针对相同难度级别问题的性能得到了大幅提升。例如,加入Medium级别的SQL查询将Medium级别问题的性能从 38.2 38.2% 38.2 ≤ \leq Easy)提升至 56.2 56.2% 56.2 ≤ \leq Medium),这与预期一致。更令人鼓舞的是,加入更简单的SQL查询甚至可以提高更难问题的表现。例如,与BART相比, ≤ \leq Medium级别的预训练使Hard级别问题的性能提高了最多13.1%。

C.2 预训练中自然语言的影响

自然语言生成
直观而言,与SQL查询相比,在预训练中使用自然语言句子对下游任务可能更有利,因为预训练目标几乎与微调目标相同。然而,获取一个流畅的自然语言句子以忠实反映SQL查询的语义并非易事。在这个实验中,我们遵循Zhong等(2020a)的方法,训练了一个SQL到自然语言的模型,并使用该模型将预训练语料库中的SQL查询翻译成自然语言句子。具体而言,我们的SQL到自然语言模型基于BART-Large(Lewis等,2020),并在SQUALL数据集(Shi等,2020b)上进行训练,该数据集包含近9,000个SQL-NL对。然后,我们将训练好的SQL到自然语言模型应用于TAPEX的预训练语料库(50万条),并获得了大小相同的自然语言预训练语料库。通过手动分析100个翻译后的自然语言句子,我们惊讶地发现所有自然语言句子都很流畅,且近68%的句子忠实于相应SQL查询的语义。表10展示了一些采样的SQL查询及其对应的翻译后的自然语言句子。获得自然语言预训练语料库后,我们遵循与TAPEX相同的预训练和微调流程来利用它。
在这里插入图片描述

性能比较
我们在表11中比较了TAPEX使用SQL与使用自然语言(NL)在所有下游任务上的表现。令人惊讶的是,TAPEX使用自然语言的表现与使用SQL的表现相当,甚至在某些情况下更差。例如,与使用SQL查询进行预训练相比,使用自然语言句子在WIKITABLEQUESTIONS上的表现下降了1.4%。我们将这种下降归因于翻译后的自然语言句子中包含了一些噪音。以表11中的第二行为例,翻译后的自然语言句子包含了额外的信息,比如“在1989年美国职业棒球大联盟选秀中”,这些可能会干扰预训练过程。

D SQL执行的细粒度分析

图9提供了针对每种操作符类型的SQL执行准确率的细粒度分析。

在这里插入图片描述

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:/a/967102.html

如若内容造成侵权/违法违规/事实不符,请联系我们进行投诉反馈qq邮箱809451989@qq.com,一经查实,立即删除!

相关文章

Matlab机械手碰撞检测应用

本文包含三个部分: Matlab碰撞检测的实现URDF文件的制作机械手STL文件添加夹爪 一.Matlab碰撞检测的实现 首先上代码 %% 检测在结构环境中机器人是否与物体之间发生碰撞情况,如何避免? % https://www.mathworks.com/help/robotics/ug/che…

激活函数篇 04 —— softmax函数

将模型的输出转换为概率分布,使得模型能够输出每个类别的概率值。 Softmax ( a i ) e a i ∑ j 1 n e a j \text{Softmax}(a_i)\frac{e^{a_i}}{\sum_{j1}^n e^{a_j}} Softmax(ai​)∑j1n​eaj​eai​​ 其中, a i a_i ai​ 是输入向量中的第 i i i 个…

软件工程的熵减:AI如何降低系统复杂度

软件开发的世界,如同一个不断膨胀的宇宙。随着功能的增加和时间的推移,代码库越来越庞大,系统复杂度也随之水涨船高。代码膨胀、维护困难、开发效率低下等问题困扰着无数开发者。这不禁让人联想到物理学中的“熵增”原理——一个孤立系统的熵…

springboot008房屋租赁系统

版权声明 所有作品均为本人原创,提供参考学习使用,如需要源码数据库配套文档请移步 www.taobysj.com 搜索获取 技术实现 开发语言:Javavue。 框架:后端spingboot前端vue。 模式:B/S。 数据库:mysql。 开…

DeepSeek训练成本与技术揭秘

引言:在当今人工智能蓬勃发展的时代,DeepSeek 宛如一颗耀眼的新星,突然闯入大众视野,引发了全球范围内的热烈讨论。从其惊人的低成本训练模式,到高性能的模型表现,无一不让业界为之侧目。它打破了传统认知&…

计算机视觉语义分割——Attention U-Net(Learning Where to Look for the Pancreas)

计算机视觉语义分割——Attention U-Net(Learning Where to Look for the Pancreas) 文章目录 计算机视觉语义分割——Attention U-Net(Learning Where to Look for the Pancreas)摘要Abstract一、Attention U-Net1. 基本思想2. Attention Gate模块3. 软注意力与硬注意力4. 实验…

<论文>DeepSeek-R1:通过强化学习激励大语言模型的推理能力(深度思考)

一、摘要 本文跟大家来一起阅读DeepSeek团队发表于2025年1月的一篇论文《DeepSeek-R1: Incentivizing Reasoning Capability in LLMs via Reinforcement Learning | Papers With Code》,新鲜的DeepSeek-R1推理模型,作者规模属实庞大。如果你正在使用Deep…

【PDF提取内容】如何批量提取PDF里面的文字内容,把内容到处表格或者批量给PDF文件改名,基于C++的实现方案和步骤

以下分别介绍基于 C 批量提取 PDF 里文字内容并导出到表格,以及批量给 PDF 文件改名的实现方案、步骤和应用场景。 批量提取 PDF 文字内容并导出到表格 应用场景 文档数据整理:在处理大量学术论文、报告等 PDF 文档时,需要提取其中的关键信…

collabora online+nextcloud+mariadb在线文档协助

1、环境 龙蜥os 8.9 docker 2、安装docker dnf -y install dnf-plugins-core dnf config-manager --add-repo https://download.docker.com/linux/centos/docker-ce.repo sed -i shttps://download.docker.comhttps://mirrors.tuna.tsinghua.edu.cn/docker-ce /etc/yum.repos.…

Meta AI 最近推出了一款全新的机器学习框架ParetoQ,专门用于大型语言模型的4-bit 以下量化

每周跟踪AI热点新闻动向和震撼发展 想要探索生成式人工智能的前沿进展吗?订阅我们的简报,深入解析最新的技术突破、实际应用案例和未来的趋势。与全球数同行一同,从行业内部的深度分析和实用指南中受益。不要错过这个机会,成为AI领…

w198基于Springboot的智能家居系统

🙊作者简介:多年一线开发工作经验,原创团队,分享技术代码帮助学生学习,独立完成自己的网站项目。 代码可以查看文章末尾⬇️联系方式获取,记得注明来意哦~🌹赠送计算机毕业设计600个选题excel文…

电脑连接wifi但是浏览器打开不了网页,使用手机热点能正常使用

电脑连接wifi但是浏览器打开不了网页,使用手机热点能正常使用 打开控制面板 打开网络和Internet(查看网络状态和任务) 点击更改适配器设置 双击WLAN 点击属性并双击打开Internet 协议版本4(TCP/IPv4) 将自动…

蓝桥杯K倍区间(前缀和与差分,取模化简)

输入 5 2 1 2 3 4 5 输出 6 思路:首先由连续子串和可以想用前缀和,由于加减法总和取模和分别取模结果不受影响,所以我们前缀和之后直接取模方便观察性质,本题前缀和:1,3,6,10&#…

《Wiki.js知识库部署实践 + CNB Git数据同步方案解析》

一、wiki.js 知识库简介 基本概述 定义 :Wiki.js 是一个开源、现代、轻量且功能强大的 Wiki 应用程序,基于 Node.js 构建,旨在帮助个人和团队轻松创建、管理和共享知识。开源性质 :它遵循 AGPLv3 许可证,任何人都可以…

递增三元组(蓝桥杯18F)

暴力求解&#xff1a; #include<iostream> using namespace std; int main() {int N;cin >> N;int* A new int[N];int* B new int[N];int* C new int[N];for (int i 0; i < N;i) {cin >> A[i];}for (int i 0; i < N; i) {cin >> B[i];}for…

【抽象代数】1.2. 半群与群

群的定义 群非空集合二元运算性质 定义1. 设 为一个非空集合&#xff0c;上有二元运算&#xff0c;满足结合律&#xff0c;则称或为一个半群。 定义2. 设 为半群&#xff0c;若元素 满足 &#xff0c;则称 为 的左幺元&#xff08;右幺元&#xff1a;&#xff09;&#…

idea如何使用AI编程提升效率-在IntelliJ IDEA 中安装 GitHub Copilot 插件的步骤-卓伊凡

idea如何使用AI编程提升效率-在IntelliJ IDEA 中安装 GitHub Copilot 插件的步骤-卓伊凡 问题 idea编译器 安装copilot AI工具 实际操作 在 IntelliJ IDEA 中安装 GitHub Copilot 插件的步骤如下&#xff1a; 打开 IntelliJ IDEA&#xff1a; 打开你的 IntelliJ IDEA 应用…

机器学习之数学基础:线性代数、微积分、概率论 | PyTorch 深度学习实战

前一篇文章&#xff0c;使用线性回归模型逼近目标模型 | PyTorch 深度学习实战 本系列文章 GitHub Repo: https://github.com/hailiang-wang/pytorch-get-started 本篇文章内容来自于 强化学习必修课&#xff1a;引领人工智能新时代【梗直哥瞿炜】 线性代数、微积分、概率论 …

生成式聊天机器人 -- 基于Pytorch + Global Attention + 双向 GRU 实现的SeqToSeq模型 -- 上

生成式聊天机器人 -- 基于Pytorch Global Attention 双向 GRU 实现的SeqToSeq模型 -- 上 前言数据预处理下载并加载数据原始数据格式化数据清洗与字典映射转换为模型需要的数据格式 SeqToSeq 模型Encoder 编码器Decoder 解码器全局注意力机制解码器实现 前言 本文会介绍使用…

Maven 安装配置(完整教程)

文章目录 一、Maven 简介二、下载 Maven三、配置 Maven3.1 配置环境变量3.2 Maven 配置3.3 IDEA 配置 四、结语 一、Maven 简介 Maven 是一个基于项目对象模型&#xff08;POM&#xff09;的项目管理和自动化构建工具。它主要服务于 Java 平台&#xff0c;但也支持其他编程语言…