Metric learning with multi-relational data

Saved in:
Bibliographic Details
Title: Metric learning with multi-relational data
Authors: Pan, Jiajun, Le Capitaine, Hoel
Contributors: Data User Knowledge (LS2N - équipe DUKe), Laboratoire des Sciences du Numérique de Nantes (LS2N), Institut National de Recherche en Informatique et en Automatique (Inria)-Centre National de la Recherche Scientifique (CNRS)-IMT Atlantique (IMT Atlantique), Institut Mines-Télécom Paris (IMT)-Institut Mines-Télécom Paris (IMT)-NANTES UNIVERSITÉ - École Centrale de Nantes (Nantes Univ - ECN), Nantes Université (Nantes Univ)-Nantes Université (Nantes Univ)-Nantes université - UFR des Sciences et des Techniques (Nantes univ - UFR ST), Nantes Université - pôle Sciences et technologie, Nantes Université (Nantes Univ)-Nantes Université (Nantes Univ)-Nantes Université - pôle Sciences et technologie, Nantes Université (Nantes Univ)-Institut National de Recherche en Informatique et en Automatique (Inria)-Centre National de la Recherche Scientifique (CNRS)-IMT Atlantique (IMT Atlantique), Nantes Université (Nantes Univ)
Source: International Journal of Machine Learning and Cybernetics. 16:2957-2969
Publisher Information: Springer Science and Business Media LLC, 2024.
Publication Year: 2024
Subject Terms: Multi-Relational Learning, [INFO]Computer Science [cs], Metric Learning
Description: International audience ; Over the past decades, there has been a growing interest in metric learning, a type of representation learning that aims to learn a distance metric that can fit to the data being analyzed. Many metric learning algorithms have been designed for data lying in Euclidean spaces, where a parametric Mahalanobis metric can be learned. However, such algorithms are often unable to handle relational data, that is not independent and identically distributed (i.i.d.), or can only be used at an entity level. In contrast, relational data allows for the discovery of complex interactions between features and entities, which can lead to better models. In this paper, we introduce two novel metric learning algorithms tailored to handle relational data, that preserve the structural information of the graph and use the features of the nodes as well. The first one is supervised and makes full use of both the graph structure and node labels with a carefully designed loss function, while the second is unsupervised and only uses the graph structure. Our experimental results show that both methods outperform state-of-the-art learning algorithms. Interestingly, we also find that the proposed unsupervised method often performs better than traditional supervised metric learning approaches.
Document Type: Article
Language: English
ISSN: 1868-808X
1868-8071
DOI: 10.1007/s13042-024-02430-x
Rights: Springer Nature TDM
Accession Number: edsair.doi.dedup.....c0a43436027f858c308a1dea9f848fd3
Database: OpenAIRE
Description
Abstract:International audience ; Over the past decades, there has been a growing interest in metric learning, a type of representation learning that aims to learn a distance metric that can fit to the data being analyzed. Many metric learning algorithms have been designed for data lying in Euclidean spaces, where a parametric Mahalanobis metric can be learned. However, such algorithms are often unable to handle relational data, that is not independent and identically distributed (i.i.d.), or can only be used at an entity level. In contrast, relational data allows for the discovery of complex interactions between features and entities, which can lead to better models. In this paper, we introduce two novel metric learning algorithms tailored to handle relational data, that preserve the structural information of the graph and use the features of the nodes as well. The first one is supervised and makes full use of both the graph structure and node labels with a carefully designed loss function, while the second is unsupervised and only uses the graph structure. Our experimental results show that both methods outperform state-of-the-art learning algorithms. Interestingly, we also find that the proposed unsupervised method often performs better than traditional supervised metric learning approaches.
ISSN:1868808X
18688071
DOI:10.1007/s13042-024-02430-x