nielsr HF staff commited on
Commit
bf726de
1 Parent(s): ebbdcf4

Add wikisql dataset

Browse files
Files changed (1) hide show
  1. README.md +114 -0
README.md ADDED
@@ -0,0 +1,114 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ language: en
3
+ tags:
4
+ - tapas
5
+ - question-answering
6
+ license: apache-2.0
7
+ datasets:
8
+ - wikisql
9
+ ---
10
+
11
+ # TAPAS base model fine-tuned on WikiSQL (in a supervised fashion)
12
+
13
+ This model has 4 versions which can be used. The latest version, which is the default one, corresponds to the `tapas_wikisql_sqa_inter_masklm_base_reset` checkpoint of the [original Github repository](https://github.com/google-research/tapas).
14
+ This model was pre-trained on MLM and an additional step which the authors call intermediate pre-training, and then fine-tuned on [SQA](https://www.microsoft.com/en-us/download/details.aspx?id=54253) and
15
+ [WikiSQL](https://github.com/salesforce/WikiSQL). It uses relative position embeddings by default (i.e. resetting the position index at every cell of the table).
16
+
17
+ The other (non-default) versions which can be used are:
18
+ - `revision="v3"`, which corresponds to `tapas_wikisql_sqa_inter_masklm_base` (intermediate pre-training, absolute position embeddings)
19
+ - `revision="v2"`, which corresponds to `tapas_wikisql_sqa_masklm_base_reset` (no intermediate pre-training, relative position embeddings)
20
+ - `revision="v1"`, which corresponds to `tapas_wikisql_sqa_masklm_base` (no intermediate pre-training, absolute position embeddings)
21
+
22
+ Disclaimer: The team releasing TAPAS did not write a model card for this model so this model card has been written by
23
+ the Hugging Face team and contributors.
24
+
25
+ ## Model description
26
+
27
+ TAPAS is a BERT-like transformers model pretrained on a large corpus of English data from Wikipedia in a self-supervised fashion.
28
+ 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
29
+ can use lots of publicly available data) with an automatic process to generate inputs and labels from those texts. More precisely, it
30
+ was pretrained with two objectives:
31
+
32
+ - Masked language modeling (MLM): taking a (flattened) table and associated context, the model randomly masks 15% of the words in
33
+ the input, then runs the entire (partially masked) sequence through the model. The model then has to predict the masked words.
34
+ This is different from traditional recurrent neural networks (RNNs) that usually see the words one after the other,
35
+ or from autoregressive models like GPT which internally mask the future tokens. It allows the model to learn a bidirectional
36
+ representation of a table and associated text.
37
+ - Intermediate pre-training: to encourage numerical reasoning on tables, the authors additionally pre-trained the model by creating
38
+ a balanced dataset of millions of syntactically created training examples. Here, the model must predict (classify) whether a sentence
39
+ is supported or refuted by the contents of a table. The training examples are created based on synthetic as well as counterfactual statements.
40
+
41
+ This way, the model learns an inner representation of the English language used in tables and associated texts, which can then be used
42
+ to extract features useful for downstream tasks such as answering questions about a table, or determining whether a sentence is entailed
43
+ 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.
44
+
45
+
46
+ ## Intended uses & limitations
47
+
48
+ You can use this model for answering questions related to a table.
49
+
50
+ For code examples, we refer to the documentation of TAPAS on the HuggingFace website.
51
+
52
+
53
+ ## Training procedure
54
+
55
+ ### Preprocessing
56
+
57
+ The texts are lowercased and tokenized using WordPiece and a vocabulary size of 30,000. The inputs of the model are
58
+ then of the form:
59
+
60
+ ```
61
+ [CLS] Question [SEP] Flattened table [SEP]
62
+ ```
63
+
64
+ The authors did first convert the WikiSQL dataset into the format of SQA using automatic conversion scripts.
65
+
66
+ ### Fine-tuning
67
+
68
+ 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.
69
+ 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
70
+ ratio of 0.1424. See the [paper](https://arxiv.org/abs/2004.02349) for more details (tables 11 and 12).
71
+
72
+
73
+ ### BibTeX entry and citation info
74
+
75
+ ```bibtex
76
+ @misc{herzig2020tapas,
77
+ title={TAPAS: Weakly Supervised Table Parsing via Pre-training},
78
+ author={Jonathan Herzig and Paweł Krzysztof Nowak and Thomas Müller and Francesco Piccinno and Julian Martin Eisenschlos},
79
+ year={2020},
80
+ eprint={2004.02349},
81
+ archivePrefix={arXiv},
82
+ primaryClass={cs.IR}
83
+ }
84
+ ```
85
+
86
+ ```bibtex
87
+ @misc{eisenschlos2020understanding,
88
+ title={Understanding tables with intermediate pre-training},
89
+ author={Julian Martin Eisenschlos and Syrine Krichene and Thomas Müller},
90
+ year={2020},
91
+ eprint={2010.00571},
92
+ archivePrefix={arXiv},
93
+ primaryClass={cs.CL}
94
+ }
95
+ ```
96
+
97
+ ```bibtex
98
+ @article{DBLP:journals/corr/abs-1709-00103,
99
+ author = {Victor Zhong and
100
+ Caiming Xiong and
101
+ Richard Socher},
102
+ title = {Seq2SQL: Generating Structured Queries from Natural Language using
103
+ Reinforcement Learning},
104
+ journal = {CoRR},
105
+ volume = {abs/1709.00103},
106
+ year = {2017},
107
+ url = {http://arxiv.org/abs/1709.00103},
108
+ archivePrefix = {arXiv},
109
+ eprint = {1709.00103},
110
+ timestamp = {Mon, 13 Aug 2018 16:48:41 +0200},
111
+ biburl = {https://dblp.org/rec/journals/corr/abs-1709-00103.bib},
112
+ bibsource = {dblp computer science bibliography, https://dblp.org}
113
+ }
114
+ ```