Skip to content

Latest commit

 

History

History
240 lines (188 loc) · 14.5 KB

README.md

File metadata and controls

240 lines (188 loc) · 14.5 KB
Raw

42dot LLM

42dot LLM은 포티투닷 (42dot)에서 자체 개발한 언어 모델들의 시리즈로, 다음의 특징을 가지고 있습니다.

  • 국내 최초의 한영통합 언어 모델 (=42dot LLM-PLM) 공개 more
  • 42dot LLM-PLM 기반의 생성형 언어 모델 (=42dot LLM-SFT) 공개 more
  • 직접 구축한 (수집, 정제) 데이터, 자체 학습 인프라 사용

뿐만 아니라, 🤗42dot LLM-PLM과 🤗42dot LLM-SFT를 공개했습니다.

42dot LLM-SFT 실행 예제 (example_cli.py 실행)

42dot LLM-PLM (사전 학습 모델)

아키텍처

42dot LLM-PLM은 LLaMA 2 와 유사한 Transformer decoder 아키텍처를 사용했고, 모델 하이퍼파라미터는 아래와 같습니다.

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

학습

42dot LLM-PLM의 학습은 A100 80G 256장을 사용했고, 약 8일 (49,152 GPU hours) 소요되었습니다. 학습 세팅은 아래와 같습니다.

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

(* 단위: tokens)

학습 데이터셋

42dot LLM-PLM의 학습 데이터는 모두 웹 상에 공개된 데이터를 이용해 진행했고, 그 구성은 아래와 같습니다.

토크나이저

Byte-level BPE 토크나이저를 사용했고, 학습 데이터셋에서 1,000만 건의 문서를 샘플링해 학습했습니다. Vocabulary 크기는 약 50K입니다.

Zero-shot 성능 평가

42dot LLM-PLM 및 비슷한 파라미터 크기의 타 PLM과의 성능을 비교하기 위해 한국어 및 영어 Zero-shot 벤치마크를 진행했고, 아래의 평가결과는 lm-eval-harness를 이용해 도출했습니다.

한국어

  • 비교대상:
    • Polyglot-Ko 1.3B: GPT-NeoX 아키텍처로 한국어 213B 토큰 (863 GB)을 학습한 모델
    • KoGPT2 1.2B: GPT 아키텍처로 40GB 이상의 한국어 데이터셋을 학습한 모델
    • XGLM 1.7B: GPT-3 아키텍처로 한국어를 포함한 30개 언어, 500B 토큰을 학습한 모델
    • PolyLM 1.7B: LLaMA 아키텍처로 한국어를 포함한 18개 언어, 640B 토큰을 학습한 모델
  • 평가 데이터셋:
    • KoBEST: BoolQ, COPA, HellaSwag, SentiNeg 의 4개 태스크

42dot LLM-PLM 한국어 성능

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

영어

  • 비교대상:
    • OPT 1.3B: GPT-3 아키텍처로 영어 300B 토큰을 학습한 모델
    • MPT 1B: MPT 아키텍처로 RedPajama 데이터의 200B 토큰을 학습한 모델
    • XGLM 1.7B
    • PolyLM 1.7B

42dot LLM-PLM 영어 성능

Tasks / Metric MPT
1B		 OPT
1.3B		 XGLM
1.7B		 PolyLM
1.7B		 42dot LLM-PLM
1.3B
anli_r1/acc 0.309 0.341 0.334 0.336 0.325
anli_r2/acc 0.334 0.339 0.331 0.314 0.34
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.288
arc_challenge/acc_norm 0.291 0.295 0.243 0.256 0.317
arc_easy/acc 0.608 0.571 0.537 0.461 0.628
arc_easy/acc_norm 0.555 0.51 0.479 0.404 0.564
boolq/acc 0.517 0.578 0.585 0.617 0.624
hellaswag/acc 0.415 0.415 0.362 0.322 0.422
hellaswag/acc_norm 0.532 0.537 0.458 0.372 0.544
openbookqa/acc 0.238 0.234 0.17 0.166 0.222
openbookqa/acc_norm 0.334 0.334 0.298 0.334 0.34
piqa/acc 0.714 0.718 0.697 0.667 0.725
piqa/acc_norm 0.72 0.724 0.703 0.649 0.727
record/f1 0.84 0.857 0.775 0.681 0.848
record/em 0.832 0.849 0.769 0.674 0.839
rte/acc 0.541 0.523 0.559 0.513 0.542
truthfulqa_mc/mc1 0.224 0.237 0.215 0.251 0.236
truthfulqa_mc/mc2 0.387 0.386 0.373 0.428 0.387
wic/acc 0.498 0.509 0.503 0.5 0.502
winogrande/acc 0.574 0.595 0.55 0.519 0.583
average 0.479 0.482 0.452 0.429 0.492

42dot LLM-SFT (생성형 언어 모델)

42dot LLM-SFT는 42dot LLM-PLM에 SFT (Supervised Fine-Tuning)를 수행한 모델로, 학습을 위한 파라미터는 아래와 같습니다.

Model Global Batch Size Learning rate Epochs Max length Weight decay Warmup ratio
1.3B 16 2e-5 3 4,096 0 0.03

학습은 A100 80G GPU 16장을 사용했고 약 7시간 (112 GPU hours) 소요되었습니다.

학습 데이터셋

질문/요청 및 이에 대한 응답으로 이루어진 Single/Multi-turn 형태의 대화 데이터를 학습에 사용했습니다.

  • 42dot LLM-SFT의 학습 데이터 관련 내용은 공개하지 않습니다. 대신, 이미 공개되어 있는 다양한 한국어 (evolve-instruct, ko-lima-vicuna, 등) 및 영어 (ShareGPT, OpenAssistant, etc.)의 데이터를 참고할 수 있습니다.

평가

  • 비교대상:
    • ChatGPT: OpenAI가 공개한 생성형 언어 모델 서비스 (GPT-3.5 및 GPT-4)
    • Bard: Google이 공개한 생성형 언어 모델 서비스
    • KORani-v2-13B: LLaMA 13B을 한국어 데이터셋으로 파인튜닝한 모델
Model GPT-3.5 GPT-4 Bard KORani 42dot LLM-SFT
Params Unknown Unknown Unknown 13B 1.3B
  • 평가 데이터셋:
    • 10가지의 카테고리에 대해 총 121개의 프롬프트로 구성했습니다. [Dataset Link]
  • 평가 방법:
    • GPT-4를 평가 수단으로 사용했고, <질문, 응답>을 포함하는 프롬프트를 입력으로 6가지 항목에 대해 각 모델의 응답을 평가했습니다.

평가 데이터셋에 대한 응답 품질 평가

상용 서비스와 42dot LLM-SFT의 응답 비교

모델 공개

사용법

본 리포지토리에는 간단한 생성 코드를 함께 제공하며, 아래 명령을 통해 관련 패키지를 설치하고 직접 모델을 구동해 볼 수 있습니다.

$ pip install -r requirements.txt
$ python example_cli.py

기본적으로 디바이스 설정을 자동으로 찾으며, CPU 또는 메모리 여유가 충분한 GPU를 자동으로 찾아 최적의 디바이스에서 동작하도록 설계되어 있습니다. 또한 --device=cpu 옵션으로 항상 CPU에서 구동할 수 있으며, 별도의 Quantization 옵션 없이도 M1 맥북 프로에서는 CPU 옵션으로 로컬 구동(약 4GB 정도 여유 메모리 필요)이 가능합니다. 이외에도 생성과 관련한 다양한 옵션을 지원하며 --help로 도움말을 확인할 수 있습니다.

$ python example_cli.py --help

llama.cpp

허깅페이스에 올려주신 피드백을 참고하여 llama.cpp에서도 구동할 수 있도록 Special Tokens를 패치하였습니다.

이제 아래 가이드를 참고하여 llama.cpp에서 42dot LLM-SFT 모델을 실행할 수 있습니다.

  1. 42dot LLM-SFT 모델을 ggml FP32 포맷으로 변환합니다.
$ python convert.py ./42dot_LLM-SFT-1.3B/ --vocabtype bpe
  1. 모델을 4비트로 Quantization 합니다(선택사항).
$ ./quantize ./42dot_LLM-SFT-1.3B/ggml-model-f32.gguf ./42dot_LLM-SFT-1.3B/ggml-model-q4_0.gguf q4_0
  1. 인퍼런스를 실행합니다. 아래 저희가 제시하는 옵션을 그대로 사용하는 것을 권장합니다.
$ ./main -m ./42dot_LLM-SFT-1.3B/ggml-model-f32.gguf \
--temp 0.5 \
--top_p 0.95 \
--top_k 20 \
--n-predict 512 \
--repeat-penalty 1.2 \
--color \
--prompt "호기심 많은 인간 (human)과 인공지능 봇 (AI bot)의 대화입니다. \n봇의 이름은 42dot LLM이고 포티투닷 (42dot)에서 개발했습니다. \n봇은 인간의 질문에 대해 친절하게 유용하고 상세한 답변을 제공합니다. \n" \
--in-prefix "<human>: " \
--in-suffix "<bot>:" \
--interactive-first

한계점

다른 LLM과 마찬가지로 42dot LLM도 여러 한계를 가지고 있습니다. 42dot LLM를 활용할 때 이러한 한계점들을 감안하기 바랍니다.

  • 언어 모델은 환각 (Hallucination)이라는 근본적인 문제가 있습니다. 마찬가지로 언어 모델인 42dot LLM도 이러한 환각 문제를 가지고 있으며, 생성하는 내용이 사실과 일치하지 않을 수 있습니다.
  • 자체적으로 42dot LLM-SFT의 학습 데이터를 구축하면서 케이스를 최대한 다양화했지만, 미처 포함하지 못한 질문-응답 케이스가 존재할 수 있기 때문에 기대하는 형태의 응답을 생성하지 못할 수 있습니다.
  • 동일한 입력에 대해 매번 다른 응답을 생성할 수 있습니다. 또한, 사용자가 입력한 질문/요청인 프롬프트에 민감합니다. 예를 들어, 주어진 질문에 정확한 답변을 생성했더라도, 동일한 내용에 표현 방식만 다른 질문/요청에 전혀 다른 응답을 생성할 수 있습니다.
  • 생성 결과에 별도의 필터링을 적용하지 않았습니다. 따라서 도덕, 인종, 문화, 성별, 나이, 지역, 종교, 정치성향 등에 대해 편향적이거나 부적절한 응답을 생성할 수 있습니다.

라이센스

공개한 모델은 42dot의 R&D 결과물로서, CC BY-NC 4.0을 따릅니다.

유의사항

42dot LLM을 통해 생성한 내용은 42dot의 입장과 무관하며, 42dot은 생성 내용 및 이로 인해 발생하는 문제에 대해 책임지지 않습니다.

채용안내

저희는 42dot LLM이 언어 모델 기술 및 생태계, 나아가 서비스 확산에 기여하길 기대합니다. 아울러 미래 자동차의 핵심인 SDV (Software-Defined Vehicle)를 실현하기 위한 42dot의 여정을 함께할 분들을 기다리고 있습니다.

Citation

@misc{42dot2023llm,
      title={42dot LLM: A Series of Large Language Model by 42dot},
      author={42dot Inc.},
      year={2023},
      url = {https://github.com/42dot/42dot_LLM},
      version = {1.0.0},
}