Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Can We Build Scene Graphs, Not Classify Them? FlowSG: Progressive Image-Conditioned Scene Graph Generation with Flow Matching

About

Scene Graph Generation (SGG) unifies object localization and visual relationship reasoning by predicting boxes and subject-predicate-object triples. Yet most pipelines treat SGG as a one-shot, deterministic classification problem rather than a genuinely progressive, generative task. We propose FlowSG, which recasts SGG as continuous-time transport on a hybrid discrete-continuous state: starting from a noised graph, the model progressively grows an image-conditioned scene graph through constraint-aware refinements that jointly synthesize nodes (objects) and edges (predicates). Specifically, we first leverage a VQ-VAE to quantize a scene graph (e.g., continuous visual features) into compact, predictable tokens; a graph Transformer then (i) predicts a conditional velocity field to transport continuous geometry (boxes) and (ii) updates discrete posteriors for categorical tokens (object features and predicate labels), coupling semantics and geometry via flow-conditioned message aggregation. Training combines flow-matching losses for geometry with a discrete-flow objective for tokens, yielding few-step inference and plug-and-play compatibility with standard detectors and segmenters. Extensive experiments on VG and PSG under closed- and open-vocabulary protocols show consistent gains in predicate R/mR and graph-level metrics, validating the mixed discrete-continuous generative formulation over one-shot classification baselines, with an average improvement of about 3 points over the state-of-the-art USG-Par.

Xin Hu, Ke Qin, Wen Yin, Yuan-Fang Li, Ming Li, Tao He• 2026

Related benchmarks

TaskDatasetResultRank
Predicate ClassificationVisual Genome
Recall@5065.7
61
Scene Graph Detection (SGDet)Visual Genome (VG)
R@5036.5
28
Predicate ClassificationPSG
Mean Recall@5054.9
7
Scene Graph Detection (SGDet)Visual Genome (VG) closed-set (test)
R@5036.5
7
Scene Graph GenerationPSG open-set
Recall@5026.7
5
Scene Graph GenerationVG open-set
R@5016.9
5
Showing 6 of 6 rows

Other info

Follow for update