File size: 2,950 Bytes
80f79cd
b4bd8ba
f3109a2
 
a3d73b3
 
f3109a2
342f365
 
 
f3109a2
 
 
 
 
639d146
 
f3109a2
 
 
 
 
 
 
342f365
 
 
f3109a2
 
9f98b7a
f3109a2
 
 
 
7cc0876
f3109a2
 
11aae96
f3109a2
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
9f98b7a
f3109a2
 
 
 
 
 
 
 
 
 
 
 
 
 
342f365
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
---
license: cc-by-4.0
language:
- en
datasets:
- osunlp/TableInstruct
---

---
# TableLlama: Towards Open Large Generalist Models for Tables

Project Page: [https://osu-nlp-group.github.io/TableLlama/](https://osu-nlp-group.github.io/TableLlama/)

Paper: [https://arxiv.org/abs/2311.09206](https://arxiv.org/abs/2311.09206)

Dataset: [https://huggingface.co./datasets/osunlp/TableInstruct/](https://huggingface.co./datasets/osunlp/TableInstruct/)

Code: [https://osu-nlp-group.github.io/TableLlama/](https://osu-nlp-group.github.io/TableLlama/)


## Introduction
We introduce TableLlama, an open-source large generalist model specifically tailored for various table-based tasks. The TableLlama model is trained on 🤗 [TableInstruct Dataset](https://huggingface.co./datasets/osunlp/TableInstruct), a meticulously curated instruction tuning dataset for tables. TableLlama is tuned on 2.6 million table-based task data, and can handle up to 8K context!


## Model 

[TableLlama-7B](https://huggingface.co./osunlp/TableLlama/)


## Data
The models are trained on the 🤗 [TableInstruct Dataset](https://huggingface.co./datasets/osunlp/TableInstruct), which includes a comprehensive table-based instruction tuning dataset that covers a variety of real-world tables and realistic tasks. We include 14 datasets of 11 tasks in total. Check out the dataset card for more details.


## Training Procedure
The models are fine-tuned with the TableInstruct dataset using LongLoRA (7B), fully fine-tuning version as the base model, which replaces the vanilla attention mechanism of the original Llama-2 (7B) with shift short attention. The training takes 9 days on a 48 80*A100 cluster. Check out our paper for more details.

## Evaluation
The models are evaluated on 8 in-domain datasets of 8 tasks and 6 out-of-domain datasets of 4 tasks. 


## Usage
You can use the models through Huggingface's Transformers library. 
Check our Github repo for more advanced use: [https://osu-nlp-group.github.io/TableLlama/](https://osu-nlp-group.github.io/TableLlama/)

## Prompt Format
```
Below is an instruction that describes a task, paired with an input that provides further context. Write a response that
appropriately completes the request.

### Instruction:
{instruction}

### Input:
{input}

### Question:
{question}

### Response:
```

## Limitations
We've tried our best to build table generalist models. However, we acknowledge that the models' performance may vary based on the complexity and specifics of the table tasks and datasets. Still not all table-based tasks can be covered comprehensively.


## Citation
If you use the models, data, or code from this project, please cite the original paper:

```
@misc{zhang2023tablellama,
  title={TableLlama: Towards Open Large Generalist Models for Tables}, 
  author={Tianshu Zhang and Xiang Yue and Yifei Li and Huan Sun},
  year={2023},
  eprint={2311.09206},
  archivePrefix={arXiv},
  primaryClass={cs.CL}
}
```