leo-pekelis-gradient
commited on
Commit
•
a8e2c90
1
Parent(s):
1ea7aae
Update README.md
Browse files
README.md
CHANGED
@@ -1,40 +1,47 @@
|
|
1 |
---
|
2 |
tags:
|
3 |
- generated_from_trainer
|
|
|
|
|
4 |
model-index:
|
5 |
- name: completed-model
|
6 |
results: []
|
|
|
|
|
7 |
---
|
8 |
|
9 |
-
<!-- This model card has been generated automatically according to the information the Trainer had access to. You
|
10 |
-
should probably proofread and complete it, then remove this comment. -->
|
11 |
-
|
12 |
[<img src="https://raw.githubusercontent.com/OpenAccess-AI-Collective/axolotl/main/image/axolotl-badge-web.png" alt="Built with Axolotl" width="200" height="32"/>](https://github.com/OpenAccess-AI-Collective/axolotl)
|
13 |
-
|
14 |
-
|
15 |
-
|
16 |
-
|
17 |
-
|
18 |
-
- Rewards/chosen: -0.6296
|
19 |
-
- Rewards/rejected: -2.5591
|
20 |
-
- Rewards/accuracies: 0.8571
|
21 |
-
- Rewards/margins: 1.9295
|
22 |
-
- Logps/rejected: -296.3221
|
23 |
-
- Logps/chosen: -425.5087
|
24 |
-
- Logits/rejected: -2.2481
|
25 |
-
- Logits/chosen: -1.7413
|
26 |
|
27 |
## Model description
|
28 |
|
29 |
-
|
|
|
|
|
|
|
30 |
|
31 |
## Intended uses & limitations
|
32 |
|
33 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
34 |
|
35 |
-
|
36 |
|
37 |
-
More information needed
|
38 |
|
39 |
## Training procedure
|
40 |
|
@@ -75,4 +82,4 @@ The following hyperparameters were used during training:
|
|
75 |
- Transformers 4.35.1
|
76 |
- Pytorch 2.0.1+cu118
|
77 |
- Datasets 2.14.7
|
78 |
-
- Tokenizers 0.14.1
|
|
|
1 |
---
|
2 |
tags:
|
3 |
- generated_from_trainer
|
4 |
+
- finance
|
5 |
+
- text-generation-inference
|
6 |
model-index:
|
7 |
- name: completed-model
|
8 |
results: []
|
9 |
+
language:
|
10 |
+
- en
|
11 |
---
|
12 |
|
|
|
|
|
|
|
13 |
[<img src="https://raw.githubusercontent.com/OpenAccess-AI-Collective/axolotl/main/image/axolotl-badge-web.png" alt="Built with Axolotl" width="200" height="32"/>](https://github.com/OpenAccess-AI-Collective/axolotl)
|
14 |
+
|
15 |
+
Albatross is a collection of domain-specific language models for finance applications developed by [Gradient AI](https://gradient.ai/).
|
16 |
+
|
17 |
+
This is the repository for an initial, demonstration version, the `v-alpha-tross`.
|
18 |
+
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
19 |
|
20 |
## Model description
|
21 |
|
22 |
+
The `v-alpha-tross` is based on [meta-llama/Llama-2-70b-hf](https://huggingface.co/meta-llama/Llama-2-70b-hf), with additional, finance specific, pre-training, fine-tuning and instruction tuning.
|
23 |
+
|
24 |
+
This model outperforms all other tuned Llama2-70B models on the Open LLM Leaderboard on Average score, and GSM8K. It is also skilled at extracting information from tabular data like those found in SEC filings.
|
25 |
+
|
26 |
|
27 |
## Intended uses & limitations
|
28 |
|
29 |
+
The `v-alpha-tross` is intended as a demonstration of Gradient AI's Albatross framework for developing large language models specific to the finance domain. We welcome additional research and development, but do not plan on continued internal development on this legacy model.
|
30 |
+
|
31 |
+
To get the expected performance, follow specific formatting requirements, including `INST` and `<<SYS>>` tags, and `<s>` tokens.
|
32 |
+
|
33 |
+
## Training Strategy
|
34 |
+
|
35 |
+
We aim to overcome deficiencies in general-purpose language models when faced with solving finance specific tasks.
|
36 |
+
|
37 |
+
### Pre-Training
|
38 |
+
|
39 |
+
A base Llama2-70B is further pre-trained on finance specific data since LLMs are poor at answering questions when their internal relevant document store is sparse [1].
|
40 |
+
|
41 |
+
To curate enough quality training data with low operational overhead we use a novel data gathering approach. First, we crawl public repositories of text data. Second we adapt a LiRA membership inference technique to filter our crawled corpus to datasets not likely to be already in the base model's training. And third, we consult human professionals to review the (much smaller) filtered corpus to further remove low quality results. Lastly we build synthetic data pipelines to better represent under-sampled data, and train the model to better cope with variation in data representation and formatting.
|
42 |
|
43 |
+
<span style="color:red">[TODO: ADD MARKDOWN TABLE ]</span>
|
44 |
|
|
|
45 |
|
46 |
## Training procedure
|
47 |
|
|
|
82 |
- Transformers 4.35.1
|
83 |
- Pytorch 2.0.1+cu118
|
84 |
- Datasets 2.14.7
|
85 |
+
- Tokenizers 0.14.1
|