English

Task-Agnostic Contrastive Pretraining for Relational Deep Learning

Machine Learning 2025-07-01 v1 Databases

Abstract

Relational Deep Learning (RDL) is an emerging paradigm that leverages Graph Neural Network principles to learn directly from relational databases by representing them as heterogeneous graphs. However, existing RDL models typically rely on task-specific supervised learning, requiring training separate models for each predictive task, which may hamper scalability and reuse. In this work, we propose a novel task-agnostic contrastive pretraining approach for RDL that enables database-wide representation learning. For that aim, we introduce three levels of contrastive objectives-row-level, link-level, and context-level-designed to capture the structural and semantic heterogeneity inherent to relational data. We implement the respective pretraining approach through a modular RDL architecture and an efficient sampling strategy tailored to the heterogeneous database setting. Our preliminary results on standard RDL benchmarks demonstrate that fine-tuning the pretrained models measurably outperforms training from scratch, validating the promise of the proposed methodology in learning transferable representations for relational data.

Keywords

Cite

@article{arxiv.2506.22530,
  title  = {Task-Agnostic Contrastive Pretraining for Relational Deep Learning},
  author = {Jakub Peleška and Gustav Šír},
  journal= {arXiv preprint arXiv:2506.22530},
  year   = {2025}
}

Comments

arXiv admin note: text overlap with arXiv:2506.22199

R2 v1 2026-07-01T03:37:08.008Z