PAC-Bayesian Generalization Bounds for Knowledge Graph Representation Learning

arXiv:2405.06418v1 Announce Type: new
Abstract: While a number of knowledge graph representation learning (KGRL) methods have been proposed over the past decade, very few theoretical analyses have been conducted on them. In this paper, we present the first PAC-Bayesian generalization bounds for KGRL methods. To analyze a broad class of KGRL models, we propose a generic framework named ReED (Relation-aware Encoder-Decoder), which consists of a relation-aware message passing encoder and a triplet classification decoder. Our ReED framework can express at …

abstract analyze arxiv bayesian class cs.ai cs.lg graph graph representation knowledge knowledge graph paper representation representation learning stat.ml them type while