arXiv · 2601.22107
Prior-Informed Flow Matching for Graph Reconstruction
Abstract
We introduce \textit{Prior-Informed Flow Matching (PIFM)}, a conditional flow model for graph reconstruction. Reconstructing graphs from partial observations remains a key challenge; classical embedding methods often lack global consistency, while modern generative models struggle to incorporate structural priors. PIFM bridges this gap by integrating embedding-based priors with continuous-time flow matching. Grounded in a permutation equivariant version of the distortion-perception theory, our method first uses a prior, such as GraphSAGE or node2vec, to form an informed initial estimate of the adjacency matrix based on local information. It then applies rectified flow matching to refine this estimate, transporting it toward the true distribution of clean graphs and learning a global coupling. Experiments on different datasets demonstrate that PIFM consistently enhances classical embeddings, outperforming them and state-of-the-art generative baselines in reconstruction accuracy.
Explore related subjects
Keep this discovery
Explore connections, maps & timelines
Harvey Chen, Nicolas Zilberstein, Santiago Segarra. 2026-01-29. Prior-Informed Flow Matching for Graph Reconstruction. https://arxiv.org/abs/2601.22107
Cite the original work for its findings. Save a collection to share your selection of sources.