Energy-Based Learning for Scene Graph Generation
Not on the map. Why?
Abstract (arXiv)
Traditional scene graph generation methods are trained using cross-entropy losses that treat objects and relationships as independent entities. Such a formulation, however, ignores the structure in the output space, in an inherently structured prediction problem. In this work, we introduce a novel energy-based learning framework for generating scene graphs. The proposed formulation allows for efficiently incorporating the structure of scene graphs in the output space. This additional constraint in the learning framework acts as an inductive bias and allows models to learn efficiently from a small number of labels. We use the proposed energy-based framework to train existing state-of-the-art models and obtain a significant performance improvement, of up to 21% and 27%, on the Visual Genome and GQA benchmark datasets, respectively. Furthermore, we showcase the learning efficiency of the proposed framework by demonstrating superior performance in the zero- and few-shot settings where data is scarce.
Key facts. Cited by 104 papers in the map (data through 14 Oct 2026). Builds on 24 of its 33 references that are in the map. First posted on arXiv 3 Mar 2021.
Builds on
24 of 33 references.
References · 24
- 2017Attention Is All You Needcited by 60,361
- 2015Faster R-CNN: Towards Real-Time Object Detection with Region Proposal Networkscited by 9,639
- 2019Generative Modeling by Estimating Gradients of the Data Distributioncited by 4,101
- 2016Visual Genome: Connecting Language and Vision Using Crowdsourced Dense Image Annotationscited by 2,656
- 2016Aggregated Residual Transformations for Deep Neural Networkscited by 2,546
- 2015Gated Graph Sequence Neural Networkscited by 796
- 2017Scene Graph Generation by Iterative Message Passingcited by 462
- 2017Neural Motifs: Scene Graph Parsing with Global Contextcited by 414
- 2019Your Classifier is Secretly an Energy Based Model and You Should Treat it Like Onecited by 405
- 2020Unbiased Scene Graph Generation from Biased Trainingcited by 356
- 2016Visual Relationship Detection with Language Priorscited by 334
- 2018Image Generation from Scene Graphscited by 324
12 more
- 2018Graph R-CNN for Scene Graph Generationcited by 298
- 2018Bilinear Attention Networkscited by 280
- 2018Learning to Compose Dynamic Tree Structures for Visual Contextscited by 259
- 2018Auto-Encoding Scene Graphs for Image Captioningcited by 255
- 2020GPS-Net: Graph Property Sensing Network for Scene Graph Generationcited by 166
- 2019Graphical Contrastive Losses for Scene Graph Parsingcited by 100
- 2020Bridging Knowledge Graphs to Generate Scene Graphscited by 86
- 2019Unpaired Image Captioning via Scene Graph Alignmentscited by 82
- 2018Factorizable Net: An Efficient Subgraph-based Framework for Scene Graph Generationcited by 78
- 2019Learning Canonical Representations for Scene Graph to Image Generationcited by 60
- 2020Learning Visual Commonsense for Robust Scene Graph Generationcited by 47
- 2020Graph Density-Aware Losses for Novel Compositions in Scene Graph Generationcited by 31
Built on by
Citing papers per year: 2021: 8; 2022: 33; 2023: 34; 2024: 16; 2025: 10; 2026: 3.
104 papers.
Sort:
By area:
- 2021A Comprehensive Survey of Scene Graphs: Generation and Applicationcited by 125
- 2022Panoptic Scene Graph Generationcited by 114
- 2021SGTR: End-to-end Scene Graph Generation with Transformercited by 103
- 2022RelTR: Relation Transformer for Scene Graph Generationcited by 96
- 2022Fine-Grained Scene Graph Generation with Data Transfercited by 57
- 2022GFPose: Learning 3D Human Pose Prior with Gradient Fieldscited by 57
- 2022A Survey on Graph Neural Networks and Graph Transformers in Computer Vision: A Task-Oriented Perspectivecited by 56
- 2022Towards Open-vocabulary Scene Graph Generation with Prompt-based Finetuningcited by 54
- 2023GenPose: Generative Category-level Object Pose Estimation via Diffusion Modelscited by 48
- 2024EGTR: Extracting Graph from Transformer for Scene Graph Generationcited by 47
- 2023Panoptic Video Scene Graph Generationcited by 46
- 2021Structured Sparse R-CNN for Direct Scene Graph Generationcited by 44
38 more
- 2023VL-SAT: Visual-Linguistic Semantics Assisted Training for 3D Semantic Scene Graph Prediction in Point Cloudcited by 43
- 2023SceneGenie: Scene Graph Guided Diffusion Models for Image Synthesiscited by 41
- 2023Prototype-based Embedding Network for Scene Graph Generationcited by 37
- 2023LLM4SGG: Large Language Models for Weakly Supervised Scene Graph Generationcited by 30
- 2023Expanding Scene Graph Boundaries: Fully Open-vocabulary Scene Graph Generation via Visual-Concept Alignment and Retentioncited by 29
- 2023SwinGNN: Rethinking Permutation Invariance in Diffusion Models for Graph Generationcited by 28
- 2022Fine-Grained Predicates Learning for Scene Graph Generationcited by 27
- 2021Segmentation-grounded Scene Graph Generationcited by 25
- 2022Iterative Scene Graph Generationcited by 24
- 2023Devil's on the Edges: Selective Quad Attention for Scene Graph Generationcited by 23
- 2023GraspGF: Learning Score-based Grasping Primitive for Human-assisting Dexterous Graspingcited by 23
- 2022Iterative Scene Graph Generation with Generative Transformerscited by 22
- 20244D Panoptic Scene Graph Generationcited by 22
- 2021Not All Relations are Equal: Mining Informative Labels for Scene Graph Generationcited by 21
- 2023Unbiased Scene Graph Generation via Two-stage Causal Modelingcited by 21
- 2023SGRec3D: Self-Supervised 3D Scene Graph Learning via Object-Level Scene Reconstructioncited by 21
- 2022Meta Spatio-Temporal Debiasing for Video Scene Graph Generationcited by 20
- 2023LAW-Diffusion: Complex Scene Generation by Diffusion with Layoutscited by 20
- 2023Vision Relation Transformer for Unbiased Scene Graph Generationcited by 18
- 2025MMGraphRAG: Bridging Vision and Language with Interpretable Multimodal Knowledge Graphscited by 16
- 2021From SLAM to Situational Awareness: Challenges and Surveycited by 15
- 2022Hierarchical Memory Learning for Fine-Grained Scene Graph Generationcited by 15
- 2025Generative Models in Decision Making: A Surveycited by 15
- 2022Adaptive Fine-Grained Predicates Learning for Scene Graph Generationcited by 14
- 2023Manga109Dialog: A Large-scale Dialogue Dataset for Comics Speaker Detectioncited by 14
- 2024HiKER-SGG: Hierarchical Knowledge Enhanced Robust Scene Graph Generationcited by 14
- 2025A Causal Adjustment Module for Debiasing Scene Graph Generationcited by 14
- 2023Score-PA: Score-based 3D Part Assemblycited by 12
- 2025HOLa: Zero-Shot HOI Detection with Low-Rank Decomposed VLM Feature Adaptationcited by 12
- 2024Groupwise Query Specialization and Quality-Aware Multi-Assignment for Transformer-based Visual Relationship Detectioncited by 11
- 2025DIFFVSGG: Diffusion-Driven Online Video Scene Graph Generationcited by 11
- 2022Dual-branch Hybrid Learning Network for Unbiased Scene Graph Generationcited by 10
- 2022Knowledge-augmented Deep Learning and Its Applications: A Surveycited by 10
- 2024Hydra-SGG: Hybrid Relation Assignment for One-stage Scene Graph Generationcited by 10
- 2021RelTransformer: A Transformer-Based Long-Tail Visual Relationship Recognitioncited by 9
- 2022Unsupervised Vision-Language Parsing: Seamlessly Bridging Visual Scene Graphs with Language Structures via Dependency Relationshipscited by 9
- 2023Environment-Invariant Curriculum Relation Learning for Fine-Grained Scene Graph Generationcited by 9
- 2022Probabilistic Debiasing of Scene Graphscited by 8
50 most cited of 104 All 104 (JSON)
- 2026Modeling Scientific Experiment Scenes: Dataset and Model3 Aug 2026
- 2026Modular Diffusion Models for Structured Visual Recognition21 Jun 2026
- 2026SHOE: Semantic HOI Open-Vocabulary Evaluation Metric2 Apr 2026
- 2025SceneProp: Combining Neural Network and Markov Random Field for Scene-Graph Grounding30 Nov 2025
- 2025Learning to Generate Human-Human-Object Interactions from Textual Descriptions25 Nov 2025
- 2025MMGraphRAG: Bridging Vision and Language with Interpretable Multimodal Knowledge Graphs28 Jul 2025
- 2025HOLa: Zero-Shot HOI Detection with Low-Rank Decomposed VLM Feature Adaptation21 Jul 2025
- 2025Hallucinate, Ground, Repeat: A Framework for Generalized Visual Relationship Detection6 Jun 2025
- 2025A Reverse Causal Framework to Mitigate Spurious Correlations for Debiasing Scene Graph Generation29 May 2025
- 2025A Causal Adjustment Module for Debiasing Scene Graph Generation22 Mar 2025
- 2025DIFFVSGG: Diffusion-Driven Online Video Scene Graph Generation18 Mar 2025
- 2025Generative Models in Decision Making: A Survey24 Feb 2025
38 more
- 2025UniQ: Unified Decoder with Task-specific Queries for Efficient Scene Graph Generation10 Jan 2025
- 2024Graph Structure Refinement with Energy-based Contrastive Learning20 Dec 2024
- 2024Hydra-SGG: Hybrid Relation Assignment for One-stage Scene Graph Generation16 Sep 2024
- 2024Automatic Scene Generation: State-of-the-Art Techniques, Models, Datasets, Challenges, and Future Prospects14 Sep 2024
- 2024Fine-Grained Scene Graph Generation via Sample-Level Bias Prediction27 Jul 2024
- 2024Semantic Diversity-aware Prototype-based Learning for Unbiased Scene Graph Generation22 Jul 2024
- 2024Variational Potential Flow: A Novel Probabilistic Framework for Energy-Based Generative Modelling21 Jul 2024
- 2024Leveraging Predicate and Triplet Learning for Scene Graph Generation4 Jun 2024
- 20244D Panoptic Scene Graph Generation16 May 2024
- 2024AUG: A New Dataset and An Efficient Model for Aerial Image Urban Scene Graph Generation11 Apr 2024
- 2024EGTR: Extracting Graph from Transformer for Scene Graph Generation2 Apr 2024
- 2024Groupwise Query Specialization and Quality-Aware Multi-Assignment for Transformer-based Visual Relationship Detection26 Mar 2024
- 2024Scene-Graph ViT: End-to-End Open-Vocabulary Visual Relationship Detection21 Mar 2024
- 2024HiKER-SGG: Hierarchical Knowledge Enhanced Robust Scene Graph Generation18 Mar 2024
- 2024Any2Graph: Deep End-To-End Supervised Graph Prediction With An Optimal Transport Loss19 Feb 2024
- 2024Vehicle Behavior Prediction by Episodic-Memory Implanted NDT13 Feb 2024
- 2024SGTR+: End-to-end Scene Graph Generation with Transformer23 Jan 2024
- 2023HAtt-Flow: Hierarchical Attention-Flow Mechanism for Group Activity Scene Graph Generation in Videos28 Nov 2023
- 2023Panoptic Video Scene Graph Generation28 Nov 2023
- 2023RelVAE: Generative Pretraining for few-shot Visual Relationship Detection27 Nov 2023
- 2023Enhancing Scene Graph Generation with Hierarchical Relationships and Commonsense Knowledge21 Nov 2023
- 2023Expanding Scene Graph Boundaries: Fully Open-vocabulary Scene Graph Generation via Visual-Concept Alignment and Retention18 Nov 2023
- 2023Self-Supervised Learning for Visual Relationship Detection through Masked Bounding Box Reconstruction8 Nov 2023
- 2023LLM4SGG: Large Language Models for Weakly Supervised Scene Graph Generation16 Oct 2023
- 2023SGRec3D: Self-Supervised 3D Scene Graph Learning via Object-Level Scene Reconstruction27 Sep 2023
- 2023GraspGF: Learning Score-based Grasping Primitive for Human-assisting Dexterous Grasping12 Sep 2023
- 2023Score-PA: Score-based 3D Part Assembly8 Sep 2023
- 2023Zero-Shot Scene Graph Generation via Triplet Calibration and Reduction7 Sep 2023
- 2023RepSGG: Novel Representations of Entities and Relationships for Scene Graph Generation6 Sep 2023
- 2023Head-Tail Cooperative Learning Network for Unbiased Scene Graph Generation23 Aug 2023
- 2023Vision Relation Transformer for Unbiased Scene Graph Generation18 Aug 2023
- 2023LAW-Diffusion: Complex Scene Generation by Diffusion with Layouts13 Aug 2023
- 2023Informative Scene Graph Generation via Debiasing10 Aug 2023
- 2023Generalized Unbiased Scene Graph Generation9 Aug 2023
- 2023Environment-Invariant Curriculum Relation Learning for Fine-Grained Scene Graph Generation7 Aug 2023
- 2023Unbiased Scene Graph Generation via Two-stage Causal Modeling11 Jul 2023
- 2023CREPE: Learnable Prompting With CLIP Improves Visual Relationship Prediction10 Jul 2023
- 2023SwinGNN: Rethinking Permutation Invariance in Diffusion Models for Graph Generation4 Jul 2023
50 newest of 104 All 104 (JSON)
- 2023GenPose: Generative Category-level Object Pose Estimation via Diffusion Modelscited by 48
- 2023SceneGenie: Scene Graph Guided Diffusion Models for Image Synthesiscited by 41
- 2023SwinGNN: Rethinking Permutation Invariance in Diffusion Models for Graph Generationcited by 28
- 2023LAW-Diffusion: Complex Scene Generation by Diffusion with Layoutscited by 20
- 2026Modular Diffusion Models for Structured Visual Recognitioncited by 0
- 2026Modular Diffusion Models for Structured Visual Recognition21 Jun 2026
- 2023LAW-Diffusion: Complex Scene Generation by Diffusion with Layouts13 Aug 2023
- 2023SwinGNN: Rethinking Permutation Invariance in Diffusion Models for Graph Generation4 Jul 2023
- 2023GenPose: Generative Category-level Object Pose Estimation via Diffusion Models18 Jun 2023
- 2023SceneGenie: Scene Graph Guided Diffusion Models for Image Synthesis28 Apr 2023
- 2021A Comprehensive Survey of Scene Graphs: Generation and Applicationcited by 125
- 2022Panoptic Scene Graph Generationcited by 114
- 2021SGTR: End-to-end Scene Graph Generation with Transformercited by 103
- 2022RelTR: Relation Transformer for Scene Graph Generationcited by 96
- 2022Fine-Grained Scene Graph Generation with Data Transfercited by 57
- 2022GFPose: Learning 3D Human Pose Prior with Gradient Fieldscited by 57
- 2022A Survey on Graph Neural Networks and Graph Transformers in Computer Vision: A Task-Oriented Perspectivecited by 56
- 2022Towards Open-vocabulary Scene Graph Generation with Prompt-based Finetuningcited by 54
- 2024EGTR: Extracting Graph from Transformer for Scene Graph Generationcited by 47
- 2023Panoptic Video Scene Graph Generationcited by 46
- 2021Structured Sparse R-CNN for Direct Scene Graph Generationcited by 44
- 2023VL-SAT: Visual-Linguistic Semantics Assisted Training for 3D Semantic Scene Graph Prediction in Point Cloudcited by 43
12 most cited of 96 All 96 (JSON)
- 2026Modeling Scientific Experiment Scenes: Dataset and Model3 Aug 2026
- 2026SHOE: Semantic HOI Open-Vocabulary Evaluation Metric2 Apr 2026
- 2025Learning to Generate Human-Human-Object Interactions from Textual Descriptions25 Nov 2025
- 2025MMGraphRAG: Bridging Vision and Language with Interpretable Multimodal Knowledge Graphs28 Jul 2025
- 2025HOLa: Zero-Shot HOI Detection with Low-Rank Decomposed VLM Feature Adaptation21 Jul 2025
- 2025Hallucinate, Ground, Repeat: A Framework for Generalized Visual Relationship Detection6 Jun 2025
- 2025A Reverse Causal Framework to Mitigate Spurious Correlations for Debiasing Scene Graph Generation29 May 2025
- 2025A Causal Adjustment Module for Debiasing Scene Graph Generation22 Mar 2025
- 2025DIFFVSGG: Diffusion-Driven Online Video Scene Graph Generation18 Mar 2025
- 2025Generative Models in Decision Making: A Survey24 Feb 2025
- 2025UniQ: Unified Decoder with Task-specific Queries for Efficient Scene Graph Generation10 Jan 2025
- 2024Graph Structure Refinement with Energy-based Contrastive Learning20 Dec 2024
12 newest of 96 All 96 (JSON)