42dot_LLM-PLM-1.3B / README.md
ykhwang's picture
Update README.md
297af45
|
raw
history blame
7.37 kB
metadata
language:
  - en
  - ko
pipeline_tag: text-generation
tags:
  - pytorch
  - llama
  - causal-lm
  - 42dot-llm
license: cc-by-nc-4.0

42dot-PLM 1.3B

42dot-PLM is a pre-trained language model (PLM) developed by 42dot which is trained using Korean and English text corpus. This repository contains a 1.3B-parameter version of the model.

Model Description

Hyperparameters

42dot-PLM is built upon a Transformer decoder architecture similar to the LLaMA 2 and its hyperparameters are listed below.

Params Layers Attention heads Hidden size FFN size
1.3B 24 32 2,048 5,632

Pre-training

Pre-training took 9 days using 256 * NVIDIA A100 GPUs. Related settings are listed below.

Params Global batch size* Initial learning rate Train iter.* Max length* Weight decay
1.3B 4.0M 4E-4 1.4T 8,192 0.1

(* unit: tokens)

Pre-training datasets

We used a set of publicly available text corpus, including:

Tokenizer

The tokenizer is based on the Byte-level BPE algorithm. We trained its vocabulary from scratch using a subset of the pre-training corpus. For constructing a subset, 10M and 10M documents are sampled from Korean and English corpus respectively. The resulting vocabulary sizes about 50K.

Zero-shot evaluations

We evaluate 42dot-PLM on a variety of academic benchmarks both in Korean and English. All the results are obtained using lm-eval-harness and models released on the Hugging Face Hub.

Korean (KOBEST)

Tasks / Macro-F1 KoGPT2
1.2B
Polyglot-Ko
1.3B
XGLM
1.7B
PolyLM
1.7B
42dot-PLM
1.3B
boolq 0.337 0.355 0.502 0.334 0.351
copa 0.67 0.721 0.616 0.513 0.711
hellaswag 0.404 0.401 0.374 0.321 0.437
sentineg 0.606 0.679 0.46 0.382 0.711
average 0.504 0.539 0.488 0.388 0.553

English

Tasks / Metric MPT
1B
OPT
1.3B
XGLM
1.7B
PolyLM
1.7B
42dot-PLM
1.3B
anli_r1/acc 0.309 0.341 0.334 0.336 0.328
anli_r2/acc 0.334 0.339 0.331 0.314 0.334
anli_r3/acc 0.33 0.336 0.333 0.339 0.333
arc_challenge/acc 0.268 0.234 0.21 0.198 0.282
arc_challenge/acc_norm 0.291 0.295 0.243 0.256 0.314
arc_easy/acc 0.608 0.571 0.537 0.461 0.623
arc_easy/acc_norm 0.555 0.51 0.479 0.404 0.561
boolq/acc 0.517 0.578 0.585 0.617 0.628
hellaswag/acc 0.415 0.415 0.362 0.322 0.419
hellaswag/acc_norm 0.532 0.537 0.458 0.372 0.538
openbookqa/acc 0.238 0.234 0.17 0.166 0.234
openbookqa/acc_norm 0.334 0.334 0.298 0.334 0.332
piqa/acc 0.714 0.718 0.697 0.667 0.718
piqa/acc_norm 0.72 0.724 0.703 0.649 0.724
record/f1 0.84 0.857 0.775 0.681 0.85
record/em 0.832 0.849 0.769 0.674 0.841
rte/acc 0.541 0.523 0.559 0.513 0.516
truthfulqa_mc/mc1 0.224 0.237 0.215 0.251 0.234
truthfulqa_mc/mc2 0.387 0.386 0.373 0.428 0.382
wic/acc 0.498 0.509 0.503 0.5 0.503
winogrande/acc 0.574 0.595 0.55 0.519 0.575
avearge 0.479 0.482 0.452 0.429 0.489

Limitations and Ethical Considerations

42dot-PLM shares a number of well-known limitations of other large language models (LLMs). For example, it may generate false and misinformative content since 42dot-PLM is also subject to hallucination. In addition, 42dot-PLM may generate toxic, harmful, and biased content due to the use of web-available training data. We strongly suggest that 42dot-PLM users should be aware of those limitations and take necessary steps to mitigate those issues.

Disclaimer

The contents generated by 42dot LLM series ("42dot LLMs") do not necessarily reflect the views or opinions of 42dot Inc. ("42dot"). 42dot disclaims any and all liability to any part for any direct, indirect, implied, punitive, special, incidental or other consequential damages arising any use of the 42dot LLMs and theirs generated contents.

License

The 42dot-PLM is licensed under the Creative Commons Attribution-NonCommercial 4.0 (CC BY-NC 4.0) license.

Citation

@misc{42dot2023lm,
      title={42dot LM: Instruction Tuned Large Language Model of 42dot},
      author={Woo-Jong Ryu and Sang-Kil Park and Jinwoo Park and Sungmin Lee and Yongkeun Hwang},
      year={2023},
      url = {https://github.com/42dot/42dot_LLM},
      version = {pre-release},
}