RelDiff: Relational Data Generative Modeling with Graph-Based Diffusion Models

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Hudovernik, Valter, Xu, Minkai, Shi, Juntong, Šubelj, Lovro, Ermon, Stefano, Štrumbelj, Erik, Leskovec, Jure
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866912406367633408
author Hudovernik, Valter
Xu, Minkai
Shi, Juntong
Šubelj, Lovro
Ermon, Stefano
Štrumbelj, Erik
Leskovec, Jure
author_facet Hudovernik, Valter
Xu, Minkai
Shi, Juntong
Šubelj, Lovro
Ermon, Stefano
Štrumbelj, Erik
Leskovec, Jure
contents Real-world databases are predominantly relational, comprising multiple interlinked tables that contain complex structural and statistical dependencies. Learning generative models on relational data has shown great promise in generating synthetic data and imputing missing values. However, existing methods often struggle to capture this complexity, typically reducing relational data to conditionally generated flat tables and imposing limiting structural assumptions. To address these limitations, we introduce RelDiff, a novel diffusion generative model that synthesizes complete relational databases by explicitly modeling their foreign key graph structure. RelDiff combines a joint graph-conditioned diffusion process across all tables for attribute synthesis, and a $2K+$SBM graph generator based on the Stochastic Block Model for structure generation. The decomposition of graph structure and relational attributes ensures both high fidelity and referential integrity, both of which are crucial aspects of synthetic relational database generation. Experiments on 11 benchmark datasets demonstrate that RelDiff consistently outperforms prior methods in producing realistic and coherent synthetic relational databases. Code is available at https://github.com/ValterH/RelDiff.
format Preprint
id arxiv_https___arxiv_org_abs_2506_00710
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle RelDiff: Relational Data Generative Modeling with Graph-Based Diffusion Models
Hudovernik, Valter
Xu, Minkai
Shi, Juntong
Šubelj, Lovro
Ermon, Stefano
Štrumbelj, Erik
Leskovec, Jure
Machine Learning
Real-world databases are predominantly relational, comprising multiple interlinked tables that contain complex structural and statistical dependencies. Learning generative models on relational data has shown great promise in generating synthetic data and imputing missing values. However, existing methods often struggle to capture this complexity, typically reducing relational data to conditionally generated flat tables and imposing limiting structural assumptions. To address these limitations, we introduce RelDiff, a novel diffusion generative model that synthesizes complete relational databases by explicitly modeling their foreign key graph structure. RelDiff combines a joint graph-conditioned diffusion process across all tables for attribute synthesis, and a $2K+$SBM graph generator based on the Stochastic Block Model for structure generation. The decomposition of graph structure and relational attributes ensures both high fidelity and referential integrity, both of which are crucial aspects of synthetic relational database generation. Experiments on 11 benchmark datasets demonstrate that RelDiff consistently outperforms prior methods in producing realistic and coherent synthetic relational databases. Code is available at https://github.com/ValterH/RelDiff.
title RelDiff: Relational Data Generative Modeling with Graph-Based Diffusion Models
topic Machine Learning
url https://arxiv.org/abs/2506.00710