본문으로 건너뛰기

트렌딩 - GitHub 인기 레포 & HuggingFace 모델

sentence-transformers/all-MiniLM-L6-v2

문장 임베딩 및 문장 유사도256K 컨텍스트apache-2.0

문장과 짧은 문단을 384차원 벡터로 바꿔 의미 검색과 유사도 비교에 활용하는 영어 임베딩 모델

학습 방식 · nreimers/MiniLM-L6-H384-uncased 기반 Fine-tuning + contrastive learning, 1,170,060,424개 문장 쌍 학습

TL;DR

이 모델은 nreimers/MiniLM-L6-H384-uncased를 기반으로 1B 문장 쌍 이상에 Fine-tuning한 sentence-transformers 임베딩 모델입니다. 입력 문장이나 짧은 문단을 384차원 dense vector로 변환하며, 배치 안의 모든 문장 쌍에 cosine similarity를 계산한 뒤 실제 연결된 쌍을 맞히도록 contrastive learning을 적용합니다. 의미 검색, 문장 유사도 비교, 클러스터링에 적합하지만 256 word pieces를 넘는 입력은 잘립니다.

핵심 포인트

  • MiniLM 기반 모델을 1,170,060,424개 문장 쌍에 Fine-tuning해 대규모 의미 표현을 학습했습니다.
  • 배치의 문장 쌍마다 cosine similarity를 계산하고 실제 쌍을 분류하는 contrastive objective를 사용합니다.
  • SentenceTransformer.encode가 입력을 바로 384차원 벡터로 변환하며 의미 검색과 클러스터링에 활용됩니다.
  • Transformers만 사용할 때는 attention mask를 반영한 mean pooling과 L2 정규화가 추가로 필요합니다.

제한사항

  • 문장과 짧은 문단 인코더를 목표로 하며, 256 word pieces를 초과하는 입력은 잘립니다.
  • README의 메타데이터에서 language가 en으로 지정되어 있어 한국어 등 다른 언어의 성능 수치는 확인되지 않습니다.

5.3k

LIKES

255.1M

DOWNLOADS

0 / 0

조회수

관련 토론

아직 관련 토론이 없습니다.

댓글

댓글을 작성하려면 로그인이 필요합니다.