tapas base finetuned wikisql supervised

Providergoogle
Categorytable-question-answering
Licenseapache-2.0
Downloads430
Stars0

Overview

The TAPAS Base model, fine-tuned on the WikiSQL dataset, is a specialized transformer designed for table-based question answering. Unlike standard LLMs that treat tables as flattened text, TAPAS leverages a table-aware architecture to perform joint reasoning over tabular data and natural language queries. It is particularly effective for structured data retrieval tasks where the goal is to extract specific cells or perform basic aggregations based on a user's question. For developers, this provides a more precise, deterministic alternative to general-purpose RAG pipelines when dealing with structured grids. It integrates well into data analysis pipelines where SQL-like querying is required but the input is natural language.

Highlights

  • Optimized for precise table-based question answering
  • Fine-tuned on WikiSQL for high retrieval accuracy
  • Jointly processes tabular data and natural language
  • Apache-2.0 license for flexible commercial deployment
  • Efficient alternative to complex text-to-SQL pipelines

Usage

Install
# Install Hugging Face transformers
pip install transformers torch
SDK Usage
# Load model with transformers
from transformers import AutoModel, AutoTokenizer

model = AutoModel.from_pretrained("google/tapas-base-finetuned-wikisql-supervised")
tokenizer = AutoTokenizer.from_pretrained("google/tapas-base-finetuned-wikisql-supervised")

Hugging Face Download

We recommend downloading the model via the Hugging Face CLI or Hub SDK.

Guidance:Before downloading, install huggingface_hub with:

Guidance
pip install -U huggingface_hub

CLI Download

Download the full repository

Download the full repository
huggingface-cli download google/tapas-base-finetuned-wikisql-supervised

Download a single file to a local folder (e.g. config.json into ./dir)

Download a single file to a local folder (e.g. config.json into ./dir)
huggingface-cli download google/tapas-base-finetuned-wikisql-supervised config.json --local-dir ./dir

See the official docs for more CLI options

SDK Download

SDK Download
# 模型下载
from huggingface_hub import snapshot_download
model_dir = snapshot_download('google/tapas-base-finetuned-wikisql-supervised')

Git Download

Make sure git-lfs is installed first

Git Download
git lfs install
git clone https://huggingface.co/google/tapas-base-finetuned-wikisql-supervised

To skip LFS large-file downloads, use:

Skip LFS
GIT_LFS_SKIP_SMUDGE=1 git clone https://huggingface.co/google/tapas-base-finetuned-wikisql-supervised

Model files are hosted on the Hugging Face Hub — download directly via HF CLI / SDK / Git, not through this site.

PyTorch / Transformers Usage

Install Transformers

Install Transformers
pip install -U transformers torch

Load the model and run inference

Load the model and run inference
from transformers import AutoModelForCausalLM, AutoTokenizer

model = AutoModelForCausalLM.from_pretrained('google/tapas-base-finetuned-wikisql-supervised')
tokenizer = AutoTokenizer.from_pretrained('google/tapas-base-finetuned-wikisql-supervised')

Model Download

We recommend downloading the model via the ModelScope CLI or SDK.

Guidance:Before downloading, install ModelScope with:

Guidance
pip install modelscope

CLI Download

Download the full repository

Download the full repository
modelscope download --model google/tapas-base-finetuned-wikisql-supervised

Download a single file to a local folder (e.g. README.md into ./dir)

Download a single file to a local folder (e.g. README.md into ./dir)
modelscope download --model google/tapas-base-finetuned-wikisql-supervised README.md --local_dir ./dir

See the docs for more CLI options

SDK Download

SDK Download
# 模型下载
from modelscope import snapshot_download
model_dir = snapshot_download('google/tapas-base-finetuned-wikisql-supervised')

Git Download

Make sure git-lfs is installed first

Git Download
git lfs install
git clone https://www.modelscope.cn/google/tapas-base-finetuned-wikisql-supervised.git

To skip LFS large-file downloads, use:

Skip LFS
GIT_LFS_SKIP_SMUDGE=1 git clone https://www.modelscope.cn/google/tapas-base-finetuned-wikisql-supervised.git

ModelScope 模型页直接下载模型文件;无需将模型文件放在本站服务器。

Notebook Quickstart

Install the ModelScope library

Install the ModelScope library
pip install "modelscope[audio,cv,nlp,multi-modal,science]" -f https://modelscope.oss-cn-beijing.aliyuncs.com/releases/repo.html

Load the model and run inference

Load the model and run inference
from modelscope.pipelines import pipeline
from modelscope.utils.constant import Tasks

p = pipeline('text-generation', 'google/tapas-base-finetuned-wikisql-supervised')

Full Documentation

来源: HuggingFace

---
language: en
tags:

  • tapas

license: apache-2.0
datasets:
  • wikisql

---

TAPAS base model fine-tuned on WikiSQL (in a supervised fashion)

his model has 2 versions which can be used. The default version corresponds to the tapas_wikisql_sqa_inter_masklm_base_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, and WikiSQL. 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 to tapas_wikisql_sqa_inter_masklm_base (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.

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
the input, then runs the entire (partially masked) sequence through the model. The model then has to predict the masked words. This is different from traditional recurrent neural networks (RNNs) that usually see the words one after the other, or from autoregressive models like GPT which internally mask the future tokens. It allows the model to learn a bidirectional representation of a table and associated text.
  • Intermediate pre-training: to encourage numerical reasoning on tables, the authors additionally pre-trained the model by creating
a balanced dataset of millions of syntactically created training examples. Here, the model must predict (classify) whether a sentence is supported or refuted by the contents of a table. The training examples are created based on synthetic as well as counterfactual statements.

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 and WikiSQL.

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:

code
[CLS] Question [SEP] Flattened table [SEP]

The authors did first convert the WikiSQL 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 6.17164e-5, and a warmup
ratio of 0.1424. See the paper for more details (tables 11 and 12).

BibTeX entry and citation info

bibtex
@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}
}
bibtex
@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}
}
bibtex
@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}
}
Join our Telegram