설명
SpanBERT는 Facebook Research에서 개발한 고급 AI 모델로, 텍스트 스팬의 표현 및 예측에 중점을 두어 사전 학습 기법을 향상시키도록 설계되었습니다. 이 리포지토리는 연구원 및 개발자가 SpanBERT의 기능을 활용하고 평가하는 데 필요한 코드와 모델을 제공합니다.
더 읽기
이 프로젝트는 기본 및 대형 cased 버전 모두에서 사전 학습된 모델을 제공합니다. 이 모델들은 BERT와 동일한 구성을 공유하지만, 관련 논문에 자세히 설명된 대로 고유한 마스킹 방식과 학습 목표를 통합합니다. SpanBERT (기본 & cased) 모델은 12개의 레이어, 768개의 은닉 유닛, 12개의 헤드로 구성된 1억 1천만 개의 매개변수를 특징으로 합니다. 더 큰 SpanBERT (대형 & cased) 모델은 24개의 레이어, 1024개의 은닉 유닛, 16개의 헤드, 3억 4천만 개의 매개변수를 자랑합니다. 이 모델들은 HuggingFace BERT 모델의 형식을 채택하여 쉽게 대체할 수 있도록 설계되어 원활한 통합을 지원합니다.
이 리포지토리에는 다양한 자연어 처리(NLP) 작업에 SpanBERT를 미세 조정하기 위한 포괄적인 코드가 포함되어 있습니다. SQuAD 1.1 및 SQuAD 2.0 질문 답변 데이터셋, 관계 추출을 위한 TACRED, 그리고 NewsQA, TriviaQA, SearchQA, HotpotQA, NaturalQuestions와 같은 데이터셋을 포함하는 MRQA 벤치마크에 대한 미세 조정을 위한 특정 스크립트가 제공됩니다. 또한 RTE와 같은 GLUE 작업에 대한 미세 조정 코드도 사용할 수 있습니다. 공통 참조 해결 작업의 경우 별도의 TensorFlow 구현이 연결되어 있습니다.
성능 벤치마크가 제시되어 SQuAD 1.1, SQuAD 2.0, Coref 및 TACRED 데이터셋에서 BERT에 비해 SpanBERT의 우수한 결과를 보여줍니다. 예를 들어, SpanBERT (대형)은 SQuAD 1.1에서 94.6, SQuAD 2.0에서 88.7의 F1 점수를 달성하여 BERT (대형)을 능가합니다. 이 프로젝트는 또한 다운스트림 작업을 위한 미세 조정된 모델 다운로드를 용이하게 하여 사용자의 채택 프로세스를 단순화합니다.
SpanBERT는 코드와 사전 학습된 모델 모두에 적용되는 CC-BY-NC 4.0 라이선스 하에 공개됩니다. 이를 통해 비상업적 연구 및 개발에 접근할 수 있습니다. 리포지토리는 기여를 장려하며 문의 및 지원을 위한 연락처 정보를 제공합니다.
- 카테고리AI 모델 및 LLM
- 요금제오픈소스
- 배포클라우드, 셀프 호스팅
- 플랫폼웹
SpanBERT의 핵심 기능
사전 학습된 SpanBERT 모델 (기본 및 대형, cased)
SpanBERT 사용 및 평가를 위한 코드
SQuAD 1.1 및 SQuAD 2.0 미세 조정 스크립트
TACRED 관계 추출 미세 조정 스크립트
MRQA 데이터셋 (NewsQA, TriviaQA 등) 미세 조정 스크립트
GLUE 작업 (RTE) 미세 조정 스크립트
HuggingFace BERT 모델 형식과의 호환성
공통 참조 해결 미세 조정 코드 사용 가능
BERT 대비 성능 벤치마크
다운스트림 작업을 위한 다운로드 가능한 미세 조정 모델
SpanBERT: 스팬 표현 및 예측을 통한 사전 학습 개선 논문 구현
SpanBERT 시작하기
코드 액세스: SpanBERT GitHub 리포지토리를 클론합니다.
환경 설정: Apex (특정 커밋 권장)를 포함한 필수 종속성을 설치합니다.
사전 학습된 모델 로드: 제공된 스크립트를 사용하여 기본 또는 대형 cased SpanBERT 모델을 로드합니다.
모델 미세 조정: SQuAD, TACRED, MRQA 또는 GLUE와 같은 작업에 대해 제공된 Python 스크립트를 실행합니다.
API를 통한 통합: HuggingFace 호환성을 활용하여 사용자 지정 애플리케이션에 모델 사용을 적용합니다.
성능 평가: 평가 스크립트를 실행하여 특정 NLP 작업에 대한 모델 성능을 평가합니다.
미세 조정된 모델 다운로드: 제공된 스크립트를 사용하여 다운스트림 작업을 위한 사전 학습된 모델을 얻습니다.
SpanBERT의 사용 사례
- 질문 답변
- 관계 추출
- 공통 참조 해결
- 자연어 이해
- 텍스트 스팬 예측
- 연구 및 개발
- 벤치마크 평가













