Daily Arxiv

전 세계에서 발간되는 인공지능 관련 논문을 정리하는 페이지 입니다.
본 페이지는 Google Gemini를 활용해 요약 정리하며, 비영리로 운영 됩니다.
논문에 대한 저작권은 저자 및 해당 기관에 있으며, 공유 시 출처만 명기하면 됩니다.

Bridging Generative and Discriminative Learning: Few-Shot Relation Extraction via Two-Stage Knowledge-Guided Pre-training

Created by
  • Haebom

저자

Quanjiang Guo, Jinchuan Zhang, Sijie Wang, Ling Tian, Zhao Kang, Bin Yan, Weidong Xiao

개요

TKRE(Two-Stage Knowledge-Guided Pre-training for Relation Extraction)는 소량의 데이터로 관계 추출을 수행하는 Few-Shot Relation Extraction (FSRE) 문제를 해결하기 위한 새로운 프레임워크입니다. 기존 모델들의 일반화 능력의 한계와 주석 데이터 부족 문제를 해결하기 위해, LLM(Large Language Model)과 기존 관계 추출 모델을 결합하여 생성적 학습과 판별적 학습 패러다임을 연결합니다. LLM을 활용하여 설명 기반 지식과 스키마 제약을 가진 합성 데이터를 생성하고, Masked Span Language Modeling (MSLM)과 Span-Level Contrastive Learning (SCL)을 결합한 2단계 사전 학습 전략을 통해 관계 추론 및 일반화 능력을 향상시킵니다. 벤치마크 데이터셋 실험 결과, 최첨단 성능을 달성하여 저자원 환경에서의 활용 가능성을 보여줍니다.

시사점, 한계점

시사점:
LLM과 기존 관계 추출 모델의 시너지 효과를 통해 FSRE 성능 향상.
LLM을 활용한 데이터 증강 및 지식 기반 학습의 효과 입증.
2단계 사전 학습 전략(MSLM 및 SCL)을 통한 관계 추론 및 일반화 능력 향상.
저자원 환경에서의 FSRE 문제 해결에 대한 새로운 접근 방식 제시.
최첨단 성능 달성.
한계점:
LLM 기반 데이터 생성의 품질 및 신뢰도에 대한 추가적인 연구 필요.
특정 도메인이나 관계 유형에 대한 일반화 성능 평가 필요.
다양한 저자원 환경에 대한 로버스트니스(robustness) 평가 필요.
LLM 사용으로 인한 계산 비용 및 환경적 영향 고려 필요.
👍