Logo image
Generative Flow-Matching Modeling of the Neurodevelopmental Connectome via Dynamic Hypergraphs
Preprint   Open access

Generative Flow-Matching Modeling of the Neurodevelopmental Connectome via Dynamic Hypergraphs

Katherine Emily Birch, Alberto Durán-López, Daniel Bolaños-Martinez, Maria Bermudez-Edo, Roman Bauer and Suparna De
KDD '26: Proceedings of the 30th ACM SIGKDD Conference on Knowledge Discovery and Data Mining
Association for Computing Machinery (ACM)
32nd ACM SIGKDD Conference on Knowledge Discovery and Data Mining (KDD 2026) (Jeju, Korea, 09/08/2026–13/08/2026)
2026

Abstract

Generative AI Hypergraph Flow matching Pretermbirth Health informatics Markov processes Learning latent representations Applied computing → Biological networks; Health informatics Mathematics of computing → Hypergraphs; Markov processes Computing methodologies → Learning latent representations Neuroscience

Human brain structure datasets are regularly small and lack a representative sample of target phenotypes. Augmenting these is often challenging due to highly complex patterns of connectivity, and this issue is particularly emphasized when there are significant target class imbalances. We introduce a model that enables the generation of novel data instances and data exploration. Specifically, we consider the case of preterm birth, where datasets include very few preterm individuals. We present a diffusion-style flow-matching framework, whereby conditioning on continuous gestational age (GA), the model learns the underlying geometry of the brain and can reproduce differences in connectivities for infants born at varying numbers of weeks. This approach is inspired by the brain's fundamental capacity for self-organization. Moreover, to understand the real implications of varying GA on the organization of the developing brain, we integrate a dynamic hypergraph layer. This allows the model to dynamically learn the higher-order dependencies that evolve in the brain, facilitating the generation of biologically plausible structural topologies. To ensure appropriateness and sufficiency in the generated networks, we utilize biologically informed losses. Additionally, we compare the generation with key graph metrics commonly employed in neurobiological studies of brain structure. Our model results demonstrate biologically realistic connectivity patterns discovered through the dynamic hypergraph approach. The code is publicly available at: https://github.com/Katherine-Birch/Generative-Flow-Matching-Modeling-of-the-Neurodevelopmental-Connectome-via-Dynamic-Hypergraphs.

This is the author's version of the work. It is posted here for your personal use. Not for redistribution. The definitive Version of Record will be published in the Proceedings of the 32nd ACM SIGKDD International Conference on Knowledge Discovery and Data Mining (KDD '26), doi: https://doi.org/10.1145/3770855.3819022

pdf
KDD_AI_for_Sciences_camera_ready1.77 MBDownloadView
Author's Accepted Manuscript Open Access
url
https://kdd2026.kdd.org/View
Event Website Conference website

Metrics

2 File views/ downloads
3 Record Views

Details

Logo image

Usage Policy