tapas tiny finetuned wtq
简介
核心亮点
- 原生支持表格问答,无需将表格转换为纯文本
- 轻量化参数设计,推理速度快,部署成本极低
- 擅长处理 WTQ 数据集风格的表格分析任务
- Apache-2.0 协议,适合快速集成到商业项目中
使用方法
# 安装 Hugging Face transformers
pip install transformers torch
# 使用 transformers 加载模型
from transformers import AutoModel, AutoTokenizer
model = AutoModel.from_pretrained("google/tapas-tiny-finetuned-wtq")
tokenizer = AutoTokenizer.from_pretrained("google/tapas-tiny-finetuned-wtq")
Hugging Face 下载
我们推荐使用命令行或者 Hugging Face Hub SDK 来进行模型的下载。
操作指引:在下载前,请先通过如下命令安装 huggingface_hub:
pip install -U huggingface_hub
命令行下载
下载完整模型库
huggingface-cli download google/tapas-tiny-finetuned-wtq
下载单个文件到指定本地文件夹(以下载 config.json 到当前路径下 ./dir 目录为例)
huggingface-cli download google/tapas-tiny-finetuned-wtq config.json --local-dir ./dir
SDK 下载
# 模型下载
from huggingface_hub import snapshot_download
model_dir = snapshot_download('google/tapas-tiny-finetuned-wtq')
Git 下载
请确保 lfs 已经被正确安装
git lfs install
git clone https://huggingface.co/google/tapas-tiny-finetuned-wtq
如果您希望跳过 lfs 大文件下载,可以使用如下命令
GIT_LFS_SKIP_SMUDGE=1 git clone https://huggingface.co/google/tapas-tiny-finetuned-wtq
模型文件托管在 Hugging Face Hub,使用 HF CLI / SDK / Git 直接下载,不经过本站。
PyTorch / Transformers 使用
安装 Transformers
pip install -U transformers torch
模型加载和推理
from transformers import AutoModelForCausalLM, AutoTokenizer
model = AutoModelForCausalLM.from_pretrained('google/tapas-tiny-finetuned-wtq')
tokenizer = AutoTokenizer.from_pretrained('google/tapas-tiny-finetuned-wtq')
模型下载
我们推荐使用命令行或者 ModelScope SDK 来进行模型的下载。
操作指引:在下载前,请先通过如下命令安装 ModelScope:
pip install modelscope
命令行下载
下载完整模型库
modelscope download --model google/tapas-tiny-finetuned-wtq
下载单个文件到指定本地文件夹(以下载 README.md 到当前路径下 dir 目录为例)
modelscope download --model google/tapas-tiny-finetuned-wtq README.md --local_dir ./dir
SDK 下载
# 模型下载
from modelscope import snapshot_download
model_dir = snapshot_download('google/tapas-tiny-finetuned-wtq')
Git 下载
请确保 lfs 已经被正确安装
git lfs install
git clone https://www.modelscope.cn/google/tapas-tiny-finetuned-wtq.git
如果您希望跳过 lfs 大文件下载,可以使用如下命令
GIT_LFS_SKIP_SMUDGE=1 git clone https://www.modelscope.cn/google/tapas-tiny-finetuned-wtq.git
ModelScope 模型页直接下载模型文件;无需将模型文件放在本站服务器。
Notebook 快速开发
下载并安装 ModelScope library
pip install "modelscope[audio,cv,nlp,multi-modal,science]" -f https://modelscope.oss-cn-beijing.aliyuncs.com/releases/repo.html
模型加载和推理
from modelscope.pipelines import pipeline
from modelscope.utils.constant import Tasks
p = pipeline('text-generation', 'google/tapas-tiny-finetuned-wtq')
完整文档
---
language: en
tags:
- tapas
- table-question-answering
license: apache-2.0
datasets:
- wtq
---
TAPAS tiny model fine-tuned on WikiTable Questions (WTQ)
This model has 2 versions which can be used. The default version corresponds to the tapas_wtq_wikisql_sqa_inter_masklm_tiny_reset checkpoint of the original Github repository.
This model was pre-trained on MLM and an additional step which the authors call intermediate pre-training, and then fine-tuned in a chain on SQA, WikiSQL and finally WTQ. It uses relative position embeddings (i.e. resetting the position index at every cell of the table).
The other (non-default) version which can be used is:
no_reset, which corresponds totapas_wtq_wikisql_sqa_inter_masklm_tiny(intermediate pre-training, absolute position embeddings).
Disclaimer: The team releasing TAPAS did not write a model card for this model so this model card has been written by
the Hugging Face team and contributors.
Results
Size | Reset | Dev Accuracy | Link
-------- | --------| -------- | ----
LARGE | noreset | 0.5062 | tapas-large-finetuned-wtq (with absolute pos embeddings)
LARGE | reset | 0.5097 | tapas-large-finetuned-wtq
BASE | noreset | 0.4525 | tapas-base-finetuned-wtq (with absolute pos embeddings)
BASE | reset | 0.4638 | tapas-base-finetuned-wtq
MEDIUM | noreset | 0.4324 | tapas-medium-finetuned-wtq (with absolute pos embeddings)
MEDIUM | reset | 0.4324 | tapas-medium-finetuned-wtq
SMALL | noreset | 0.3681 | tapas-small-finetuned-wtq (with absolute pos embeddings)
SMALL | reset | 0.3762 | tapas-small-finetuned-wtq
MINI | noreset | 0.2783 | tapas-mini-finetuned-wtq (with absolute pos embeddings)
MINI | reset | 0.2854 | tapas-mini-finetuned-wtq
TINY | noreset | 0.0823 | tapas-tiny-finetuned-wtq (with absolute pos embeddings)
TINY | reset | 0.1039 | tapas-tiny-finetuned-wtq
Model description
TAPAS is a BERT-like transformers model pretrained on a large corpus of English data from Wikipedia in a self-supervised fashion.
This means it was pretrained on the raw tables and associated texts only, with no humans labelling them in any way (which is why it
can use lots of publicly available data) with an automatic process to generate inputs and labels from those texts. More precisely, it
was pretrained with two objectives:
- Masked language modeling (MLM): taking a (flattened) table and associated context, the model randomly masks 15% of the words in
- Intermediate pre-training: to encourage numerical reasoning on tables, the authors additionally pre-trained the model by creating
This way, the model learns an inner representation of the English language used in tables and associated texts, which can then be used
to extract features useful for downstream tasks such as answering questions about a table, or determining whether a sentence is entailed
or refuted by the contents of a table. Fine-tuning is done by adding a cell selection head and aggregation head on top of the pre-trained model, and then jointly train these randomly initialized classification heads with the base model on SQa, WikiSQL and finally WTQ.
Intended uses & limitations
You can use this model for answering questions related to a table.
For code examples, we refer to the documentation of TAPAS on the HuggingFace website.
Training procedure
Preprocessing
The texts are lowercased and tokenized using WordPiece and a vocabulary size of 30,000. The inputs of the model are
then of the form:
[CLS] Question [SEP] Flattened table [SEP]The authors did first convert the WTQ dataset into the format of SQA using automatic conversion scripts.
Fine-tuning
The model was fine-tuned on 32 Cloud TPU v3 cores for 50,000 steps with maximum sequence length 512 and batch size of 512.
In this setup, fine-tuning takes around 10 hours. The optimizer used is Adam with a learning rate of 1.93581e-5, and a warmup
ratio of 0.128960. An inductive bias is added such that the model only selects cells of the same column. This is reflected by the select_one_column parameter of TapasConfig. See the paper for more details (tables 11 and
12).
BibTeX entry and citation info
@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/PasupatL15,
author = {Panupong Pasupat and
Percy Liang},
title = {Compositional Semantic Parsing on Semi-Structured Tables},
journal = {CoRR},
volume = {abs/1508.00305},
year = {2015},
url = {http://arxiv.org/abs/1508.00305},
archivePrefix = {arXiv},
eprint = {1508.00305},
timestamp = {Mon, 13 Aug 2018 16:47:37 +0200},
biburl = {https://dblp.org/rec/journals/corr/PasupatL15.bib},
bibsource = {dblp computer science bibliography, https://dblp.org}
}