Skip to content

jason9693/Soongsil-BERT

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

22 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Soongsil-BERT(KcBERT v2)

Soongsil Univ. Community BERT : 숭실대학교 or 대학 커뮤니티 전용 언어모델

** Updates on 2021.11.06 **

Soongsil BERT 를 Google Colab에서 학습을 진행할 수 있도록 하는 코드와 튜토리얼을 제공합니다! 아래 Down Stream Task 항목을 참고해 주세요.

Down Stream Task 학습


공개된 한국어 BERT는 대부분 한국어 위키, 뉴스 기사, 책 등 잘 정제된 데이터를 기반으로 학습한 언어모델입니다. 한편, 실제로 NSMC와 같은 댓글형 데이터셋은 정제되지 않았고 구어체 특징에 신조어가 많으며, 오탈자 등 공식적인 글쓰기에서 나타나지 않는 표현들이 빈번하게 등장합니다.

Beomi 님의 KcBERT는 위와 같은 특성의 데이터셋에 적용하기 위해, 네이버 뉴스의 댓글과 대댓글을 수집해 토크나이저와 BERT모델을 처음부터 학습한 Pretrained BERT 모델입니다.

Soongsil BERT(KcBERTv2)는 대학 커뮤니티 특성에 맞는 데이터셋에 적용하기 위해, KcBERT의 데이터에 더해 대학 커뮤니티, 대학 공지사항 등의 데이터를 추가하여 처음부터 학습한 RoBERTa모델 기반의 Pretrained BERT 모델입니다.

Soongsil BERT는 Huggingface의 Transformers 라이브러리를 통해 간편히 불러와 사용할 수 있습니다. (별도의 파일 다운로드가 필요하지 않습니다.)

How to use

Requirements

  • pytorch ~= 1.6.0

  • transformers ~= 3.0.1

  • emoji ~= 0.6.0

  • soynlp ~= 0.0.493

from transformers import AutoTokenizer, AutoModelForMaskedLM

# Small Model (220MB)

tokenizer = AutoTokenizer.from_pretrained("jason9693/soongsil-bert-small")

model = AutoModelForMaskedLM.from_pretrained("jason9693/soongsil-bert-small")

# Base Model (370M)

tokenizer = AutoTokenizer.from_pretrained("jason9693/soongsil-bert-base")

model = AutoModelForMaskedLM.from_pretrained("jason9693/soongsil-bert-base")

Train Data & Preprocessing

Raw Data

학습 데이터는 KcBERT에서 사용한 데이터에 아래 항목들을 추가하여 학습을 진행하였습니다.

  1. 숭실대학교 커뮤니티 유어슈
  2. 숭실대학교 dcinside
  3. 숭실대학교 각 학과 공지사항
  4. 국립국어원(2020). 국립국어원 구어 말뭉치(버전 1.0).
  5. 국립국어원(2020). 국립국어원 웹 말뭉치(버전 1.0).

데이터 사이즈는 텍스트만 추출시 약 20GB이며, 2억개 이상의 문장으로 이뤄져 있습니다.

Preprocessing

PLM 학습을 위해서 전처리를 진행한 과정은 다음과 같습니다.

  1. 한글 및 영어, 특수문자, 그리고 이모지(🥳)까지!

    정규표현식을 통해 한글, 영어, 특수문자를 포함해 Emoji까지 학습 대상에 포함했습니다.

    한편, 한글 범위를 ㄱ-ㅎ가-힣 으로 지정해 ㄱ-힣 내의 한자를 제외했습니다.

  2. 댓글 내 중복 문자열 축약

    ㅋㅋㅋㅋㅋ와 같이 중복된 글자를 ㅋㅋ와 같은 것으로 합쳤습니다.

  3. Cased Model

    KcBERT는 영문에 대해서는 대소문자를 유지하는 Cased model입니다.

  4. 글자 단위 10글자 이하 제거

    10글자 미만의 텍스트는 단일 단어로 이뤄진 경우가 많아 해당 부분을 제외했습니다.

  5. 중복 제거

    중복적으로 쓰인 댓글을 제거하기 위해 중복 댓글을 하나로 합쳤습니다.

이를 통해 만든 최종 학습 데이터는 15GB, 1억 4천만개 문장입니다.

아래 명령어로 pip로 설치한 뒤, 아래 clean함수로 클리닝을 하면 Downstream task에서 보다 성능이 좋아집니다. (<unk> 감소)

pip install soynlp emoji

아래 clean 함수를 Text data에 사용해주세요.

import re
import emoji
from soynlp.normalizer import repeat_normalize

emojis = ''.join(emoji.UNICODE_EMOJI.keys())
pattern = re.compile(f'[^ .,?!/@$%~%·∼()\x00-\x7Fㄱ-힣{emojis}]+')
url_pattern = re.compile(
    r'https?:\/\/(www\.)?[-a-zA-Z0-9@:%._\+~#=]{1,256}\.[a-zA-Z0-9()]{1,6}\b([-a-zA-Z0-9()@:%_\+.~#?&//=]*)')

def clean(x):
    x = pattern.sub(' ', x)
    x = url_pattern.sub('', x)
    x = x.strip()
    x = repeat_normalize(x, num_repeats=2)
    return x

Cleaned Data (KcBERT 데이터)

원본 데이터를 위 clean함수로 정제한 12GB분량의 txt 파일을 아래 Kaggle Dataset에서 다운받으실 수 있습니다 :)

https://www.kaggle.com/junbumlee/kcbert-pretraining-corpus-korean-news-comments

Tokenizer Train

Tokenizer는 Huggingface의 Tokenizers 라이브러리를 통해 학습을 진행했습니다.

그 중 ByteLevelBPETokenizer 를 이용해 학습을 진행했고, Vocab Size는 `16000'으로 진행했습니다.

Tokenizer를 학습하는 것에는 KcBERT데이터는 1/10로 샘플링한 데이터로 학습을 진행했고, 그외 데이터는 모두 사용했습니다.

BERT Model Pretrain

  • Soongsil BERT Small config
{
  "architectures": [
    "RobertaForMaskedLM"
  ],
  "attention_probs_dropout_prob": 0.1,
  "bos_token_id": 0,
  "eos_token_id": 2,
  "gradient_checkpointing": false,
  "hidden_act": "gelu",
  "hidden_dropout_prob": 0.1,
  "hidden_size": 768,
  "initializer_range": 0.02,
  "intermediate_size": 3072,
  "layer_norm_eps": 1e-12,
  "max_position_embeddings": 514,
  "model_type": "roberta",
  "num_attention_heads": 12,
  "num_hidden_layers": 6,
  "pad_token_id": 1,
  "total_flos": 4.897939032867707e+19,
  "type_vocab_size": 1,
  "vocab_size": 16000
}
  • Soongsil BERT Base config
{
  "architectures": [
    "RobertaForMaskedLM"
  ],
  "attention_probs_dropout_prob": 0.1,
  "bos_token_id": 0,
  "eos_token_id": 2,
  "gradient_checkpointing": false,
  "hidden_act": "gelu",
  "hidden_dropout_prob": 0.1,
  "hidden_size": 768,
  "initializer_range": 0.02,
  "intermediate_size": 3072,
  "layer_norm_eps": 1e-05,
  "max_position_embeddings": 514,
  "model_type": "roberta",
  "num_attention_heads": 12,
  "num_hidden_layers": 12,
  "pad_token_id": 1,
  "total_flos": 8.217624121867606e+19,
  "type_vocab_size": 1,
  "vocab_size": 16000
}

Soongsil BERT Model Config는 RoBERTa Base, Large 기본 세팅값을 그대로 사용했습니다. (MLM 15%, No NSP 등)

V100 GPU를 이용해 각각 30일정도 진행했고,

모델 학습 Loss는 Step에 따라 초기 200k에 가장 빠르게 Loss가 줄어들다 400k이후로는 조금씩 감소하는 것을 볼 수 있습니다.

  • Small, Base Model Loss

KcBERT-Base Pretraining Loss

학습은 V100 1대를 이용해 학습을 진행했고, 학습 시간은 Small Model 기준 30일 정도 진행했습니다. Base Model은 약 50일정도 진행한 뒤 가장 낮은 loss를 가진 체크포인트로 정했습니다.

Pretraining Example

HuggingFace MASK LM

HuggingFace soongsil-roberta-small 모델 에서 아래와 같이 테스트 해 볼 수 있습니다.

글로벌미디어, SoongsilBERT-Small

물론 base 모델 에서도 테스트 할 수 있습니다.

image-20200806160624340

추가 MLM 샘플(Base Model)

숭실대학교 커뮤니티의 특성에 최적화된 MLM태스크 샘플을 공유합니다.

글로벌미디어, SoongsilBERT-Small

글로벌미디어, SoongsilBERT-Small

글로벌미디어, SoongsilBERT-Small

Finetuning Example

NSMC Binary Classification

네이버 영화평 코퍼스 데이터셋을 대상으로 Fine Tuning을 진행해 예제학습을 진행해 보았습니다.

Soongsil-BERT 모델을 Fine tuning한 코드는 아래 링크를 통해 직접 학습을 진행하실 수 있습니다.

  • base 모델 : base
  • small 모델 : base

NSMC모델의 경우, Huggingface SoongsilBERT-nsmc-base를 통해 직접 모델을 테스트 해보실 수 있습니다.

Korean Hate Speech Detection

한국어 혐오분류 발화 데이터셋을 대상으로 Fine Tuning을 진행해 예제학습을 진행해 보았습니다.

Soongsil-BERT 모델을 Fine tuning한 코드는 아래 링크를 통해 직접 학습을 진행하실 수 있습니다.

  • base 모델 : base

예제학습 성능 및 결과확인

저희가 진행한 예제학습 태스크들에 대한 결과는 이 링크의 결과표를 참고해 주세요.

Acknowledgement

Soongsil BERT Model을 학습하는 NIA의 AI Hub 고성능 컴퓨팅 지원사업 프로그램의 지원을 받았습니다.

모델 학습 과정에서 많은 조언을 주신 Beomi님, Monologg 님 감사합니다 :)

Reference

Github Repos

Papers

Blogs

About

숭실대학교 커뮤니티용 언어모델

Resources

License

Stars

Watchers

Forks

Packages

No packages published