README.md

metadata

language:
  - ko
tags:
  - pytorch
  - causal-lm
  - medical
license: apache-2.0
pipeline_tag: text-generation

https://github.com/jwj7140/ko-medical-chat

Polyglot-Ko-Medical-5.8b

polyglot-ko-medical은 polyglot-ko를 기반으로 의료 분야의 한글 raw 데이터를 학습시킨 기반 모델입니다.

학습 데이터

polyglot-ko-medical은 약 420MB의 의료 분야 한글 말뭉치로 학습되었습니다. 주요 데이터셋은 다음과 같습니다.

Source	Size (MB)	Link
AIHub 의료, 법률 전문 서적 말뭉치	351.0	aihub.or.kr
AIHub 전문분야 한영 말뭉치	63.4	aihub.or.kr
질병관리청 국가건강정보포털	8.33	health.kdca.go.kr
보건복지부 국가정신건강정보포털	< 1.0	mentalhealth.go.kr

학습

polyglot-ko-medical-5.8b는 EleutherAI/polyglot-ko-5.8b에서 qlora로 추가 학습되었습니다.

• lora_alpha: 32
• lora_dropout: 0.05
• lora_r: 8
• target_modules: query_key_value
• epoch: 3
• learning_rate: 3e-4