nielsr HF staff commited on
Commit
81b3272
1 Parent(s): 1652e4c

First draft of model card

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