A Three-Way Model for Collective Learning on Multi-Relational Data
Maximilian NickelVolker TrespHans‐Peter Kriegel
Introduces RESCAL, a tensor factorization model for multi-relational data that enables collective learning across interconnected entities and asymmetric relations with an efficient alternating least-squares algorithm that scales significantly faster than standard relational learning methods.
Relational data from networks and the semantic web is growing rapidly in areas such as social modeling and bioinformatics, yet standard methods struggle to capture correlations across interconnected entities. This limits performance on tasks like link prediction and entity resolution where collective learning from related items can improve accuracy.
The article set out to develop and test a tensor factorization approach called RESCAL that models multi-relational data while enabling collective learning through shared latent components.
The authors represent relations as a three-way tensor and factorize each slice as Xk ≈ A Rk A^T. They solve the model with an alternating least-squares algorithm adapted from DEDICOM methods and evaluate it on a new US presidents dataset plus established benchmarks including Cora, Kinships, Nations, and UMLS. Experiments use cross-validation and compare against CP, DEDICOM, BCTF, IRM, MRC, and SUNS variants.
RESCAL achieved AUC scores comparable or superior to state-of-the-art methods on link prediction and entity resolution while running in minutes rather than hours. It clearly outperformed CP and non-aggregated baselines on collective tasks, with gains of 0.3–0.5 AUC points in the presidents example. Runtime was often an order of magnitude faster than DEDICOM and competitive with CP despite the richer model. The method also produced meaningful clusters of countries in the Nations data.
These results indicate that RESCAL can deliver higher-quality predictions for relational tasks at lower computational cost, supporting faster decisions in domains that rely on network data. The shared latent space automatically propagates information across relations, removing the need for manual feature engineering.
The authors recommend exploring distributed and stochastic gradient versions for larger data and adding constraints such as typed relations to further improve speed and accuracy. Additional work on noisy relation aggregation would strengthen the approach before wide deployment.
The main limitations are the lack of rank reduction on the relation mode, which may hurt performance when relations are noisy, and slower scaling with model rank in regularized settings. Results rest on several standard benchmarks and controlled experiments, giving reasonable confidence for similar data but warranting caution on very large or sparse graphs without further validation.
- Paper: The link prediction problem for social networks, David Liben-Nowell et al. (2003). This foundational paper establishes the link prediction formulation and baseline structural heuristics that multi-relational tensor models aim to generalize.
- Paper: Factorization Machines, Steffen Rendle (2010). This work introduces low-rank parameter factorizations to capture higher-order variable interactions in sparse relational data.
- Paper: Probabilistic Matrix Factorization, Andriy Mnih et al. (2007). It provides the low-rank matrix decomposition principles and latent factor modeling foundations that RESCAL extends to multi-relational tensor slices.
- Paper: Markov logic networks, Matthew Richardson et al. (2006). It presents the core collective learning problem and probabilistic reasoning framework over multi-relational knowledge domains.
- Paper: Complex Embeddings for Simple Link Prediction, Théo Trouillon et al. (2016). This work directly compares against RESCAL while introducing complex embeddings to capture symmetric and antisymmetric multi-relational patterns efficiently.
- Paper: Embedding Entities and Relations for Learning and Inference in Knowledge Bases, Bishan Yang et al. (2014). It builds upon multi-relational tensor factorization like RESCAL to establish a unified linear and bilinear embedding framework for knowledge base completion.
- Paper: Reasoning With Neural Tensor Networks for Knowledge Base Completion, Richard Socher et al. (2013). It extends bilinear multi-relational tensor modeling into deep neural tensor architectures for relational reasoning.
- Paper: Translating Embeddings for Modeling Multi-relational Data, Antoine Bordes et al. (2013). It proposes translational distance models as a highly scalable alternative to full bilinear tensor factorizations for multi-relational knowledge graph completion.
- Paper: Knowledge Graph Embedding by Translating on Hyperplanes, Zhen Wang et al. (2014). It extends relational embedding methods by projecting entities onto relation-specific hyperplanes to better capture multi-relational mapping properties.
- Paper: Learning Entity and Relation Embeddings for Knowledge Graph Completion, Yankai Lin et al. (2015). It further advances relational representation learning by mapping entities into separate relation-specific vector spaces.
- Paper: RotatE: Knowledge Graph Embedding by Relational Rotation in Complex Space, Zhiqing Sun et al. (2019). It generalizes multi-relational link prediction by modeling relations as rotations in complex space to simultaneously represent symmetry, antisymmetry, and composition.
- Paper: Modeling Relational Data with Graph Convolutional Networks, Michael Schlichtkrull et al. (2018). It adopts relational graph convolutional networks as neural encoders paired with tensor-style decoders to model multi-relational knowledge graphs.
- Paper: Convolutional 2D Knowledge Graph Embeddings, Tim Dettmers et al. (2017). It advances knowledge graph link prediction beyond shallow bilinear factorizations by applying 2D convolutions over entity and relation embeddings.
