Enhancing Knowledge Graph Completion with GNN Distillation and Probabilistic Interaction Modeling
Created by
Haebom
저자
Lingzhi Wang, Pengcheng Huang, Haotian Li, Yuliang Wei, Guodong Xin, Rui Zhang, Donglin Zhang, Zhenzhou Ji, Wei Wang
개요
본 논문은 지식 그래프 완성(KGC) 문제를 해결하기 위해 GNN 증류와 추상 확률적 상호작용 모델링(APIM)을 통합한 통합 프레임워크를 제안합니다. GNN 증류는 반복적인 메시지-피처 필터링 과정을 통해 과다 평활화 문제를 완화하고, APIM은 확률적 시그니처와 전이 행렬을 통해 구조화된 추상적 상호작용 패턴을 학습하여 엔티티와 관계 상호작용을 더 풍부하고 유연하게 표현합니다. WN18RR과 FB15K-237 데이터셋을 사용한 실험 결과, 제안된 기법이 기존 모델보다 성능이 크게 향상됨을 보여줍니다. 이는 정보 전파 제어와 구조화된 확률적 모델링의 중요성을 강조하며, 지식 그래프 완성 발전을 위한 새로운 방향을 제시합니다. 소스 코드는 https://anonymous.4open.science/r/APIM_and_GNN-Distillation-461C 에서 확인할 수 있습니다.