A lightweight CNN-based knowledge graph embedding model with channel attention for link prediction.
Math Biosci Eng
; 20(6): 9607-9624, 2023 Mar 21.
Article
in En
| MEDLINE
| ID: mdl-37322903
ABSTRACT
Knowledge graph (KG) embedding is to embed the entities and relations of a KG into a low-dimensional continuous vector space while preserving the intrinsic semantic associations between entities and relations. One of the most important applications of knowledge graph embedding (KGE) is link prediction (LP), which aims to predict the missing fact triples in the KG. A promising approach to improving the performance of KGE for the task of LP is to increase the feature interactions between entities and relations so as to express richer semantics between them. Convolutional neural networks (CNNs) have thus become one of the most popular KGE models due to their strong expression and generalization abilities. To further enhance favorable features from increased feature interactions, we propose a lightweight CNN-based KGE model called IntSE in this paper. Specifically, IntSE not only increases the feature interactions between the components of entity and relationship embeddings with more efficient CNN components but also incorporates the channel attention mechanism that can adaptively recalibrate channel-wise feature responses by modeling the interdependencies between channels to enhance the useful features while suppressing the useless ones for improving its performance for LP. The experimental results on public datasets confirm that IntSE is superior to state-of-the-art CNN-based KGE models for link prediction in KGs.
Key words
Full text:
1
Collection:
01-internacional
Database:
MEDLINE
Main subject:
Pattern Recognition, Automated
/
Neural Networks, Computer
Type of study:
Prognostic_studies
/
Risk_factors_studies
Language:
En
Journal:
Math Biosci Eng
Year:
2023
Document type:
Article
Affiliation country:
China