模型:
google/tapas-small-finetuned-wikisql-supervised
此模型有两个版本可供使用。默认版本对应于 original Github repository 中的tapas_wikisql_sqa_inter_masklm_small_reset检查点。该模型在MLM上进行了预训练,并进行了作者称为中间预训练的额外步骤,然后通过 SQA 和 WikiSQL 以链式进行微调。它使用相对位置嵌入(即在每个表格单元格中重置位置索引)。
另一个(非默认)可用的版本是:
免责声明:发布TAPAS的团队没有为这个模型编写模型卡片,所以这个模型卡片是由Hugging Face团队和贡献者编写的。
TAPAS是一种类似BERT的transformers模型,在大规模英文数据(来自维基百科)上以自监督方式进行预训练。这意味着它仅在未经人工标注的原始表格和相关文本上进行了预训练,没有以任何方式人工标注它们(这也是为什么它可以使用大量可用的公共数据),并使用自动化过程从这些文本中生成输入和标签。更准确地说,它以两个目标进行了预训练:
这样,模型学习了用于表格和相关文本的英语的内部表示,然后可以用于提取用于下游任务的有用特征,如回答关于表格的问题,或确定一个句子是否由表格的内容支持或否定。微调是通过在预训练模型之上添加单元格选择头和聚合头,然后将这些随机初始化的分类头与基本模型一起进行联合训练,在SQA和WikiSQL上进行的。
您可以使用此模型来回答与表格相关的问题。
有关代码示例,请参阅HuggingFace网站上有关TAPAS的文档。
文本转换为小写,并使用WordPiece进行分词,词汇表大小为30,000。模型的输入采用以下形式:
[CLS] Question [SEP] Flattened table [SEP]
作者首先使用自动转换脚本将WikiSQL数据集转换为SQA格式。
该模型在32个Cloud TPU v3核上进行了50,000个步骤的微调,最大序列长度为512,批量大小为512。在此设置中,微调大约需要10小时。使用的优化器是Adam,学习率为6.17164e-5,预热比率为0.1424。有关更多细节,请参阅 paper (表11和12)。
@misc{herzig2020tapas, title={TAPAS: Weakly Supervised Table Parsing via Pre-training}, author={Jonathan Herzig and Paweł Krzysztof Nowak and Thomas Müller and Francesco Piccinno and Julian Martin Eisenschlos}, year={2020}, eprint={2004.02349}, archivePrefix={arXiv}, primaryClass={cs.IR} }
@misc{eisenschlos2020understanding, title={Understanding tables with intermediate pre-training}, author={Julian Martin Eisenschlos and Syrine Krichene and Thomas Müller}, year={2020}, eprint={2010.00571}, archivePrefix={arXiv}, primaryClass={cs.CL} }
@article{DBLP:journals/corr/abs-1709-00103, author = {Victor Zhong and Caiming Xiong and Richard Socher}, title = {Seq2SQL: Generating Structured Queries from Natural Language using Reinforcement Learning}, journal = {CoRR}, volume = {abs/1709.00103}, year = {2017}, url = {http://arxiv.org/abs/1709.00103}, archivePrefix = {arXiv}, eprint = {1709.00103}, timestamp = {Mon, 13 Aug 2018 16:48:41 +0200}, biburl = {https://dblp.org/rec/journals/corr/abs-1709-00103.bib}, bibsource = {dblp computer science bibliography, https://dblp.org} }