DOI QR코드

DOI QR Code

Topic-based Knowledge Graph-BERT

토픽 기반의 지식그래프를 이용한 BERT 모델

  • Min, Chan-Wook (Dept. of Applied Artificial Intelligence, Kwang-Woon University) ;
  • Ahn, Jin-Hyun (Dept. of Management Information System, Jeju University) ;
  • Im, Dong-Hyuk (School of Information Convergence, Kwang-Woon University)
  • 민찬욱 (광운대학교 인공지능융합학과) ;
  • 안진현 (제주대학교 경영정보학과) ;
  • 임동혁 (광운대학교 정보융합학부)
  • Published : 2022.05.17

Abstract

최근 딥러닝의 기술발전으로 자연어 처리 분야에서 Q&A, 문장추천, 개체명 인식 등 다양한 연구가 진행 되고 있다. 딥러닝 기반 자연어 처리에서 좋은 성능을 보이는 트랜스포머 기반 BERT 모델의 성능향상에 대한 다양한 연구도 함께 진행되고 있다. 본 논문에서는 토픽모델인 잠재 디리클레 할당을 이용한 토픽별 지식그래프 분류와 입력문장의 토픽을 추론하는 방법으로 K-BERT 모델을 학습한다. 분류된 토픽 지식그래프와 추론된 토픽을 이용해 K-BERT 모델에서 대용량 지식그래프 사용의 효율적 방법을 제안한다.

Keywords

Acknowledgement

이 성과는 정부(과학기술정보통신부)의 재원으로 한국연구재단의 지원을 받아 수행된 연구임(No. NRF-2021R1F1A1054739). 또한, 본 연구는 과학기술정보통신부 및 정보통신기술진흥센터의 대학ICT연구센터지원사업의 연구결과로 수행되었음 (IITP-2022-2018-0-01417).