Classification by Attention: Scene Graph Classification with Prior Knowledge


  • Sahand Sharifzadeh Ludwig Maximilian University of Munich
  • Sina Moayed Baharlou Ludwig Maximilian University of Munich
  • Volker Tresp Ludwig Maximilian University of Munich Siemens AG



Neuro-Symbolic AI (NSAI), Visual Reasoning & Symbolic Representations, Common-Sense Reasoning, Scene Analysis & Understanding


A major challenge in scene graph classification is that the appearance of objects and relations can be significantly different from one image to another. Previous works have addressed this by relational reasoning over all objects in an image or incorporating prior knowledge into classification. Unlike previous works, we do not consider separate models for perception and prior knowledge. Instead, we take a multi-task learning approach by introducing schema representations and implementing the classification as an attention layer between image-based representations and the schemata. This allows for the prior knowledge to emerge and propagate within the perception model. By enforcing the model also to represent the prior, we achieve a strong inductive bias. We show that our model can accurately generate commonsense knowledge and that the iterative injection of this knowledge to scene representations, as a top-down mechanism, leads to significantly higher classification performance. Additionally, our model can be fine-tuned on external knowledge given as triples. When combined with self-supervised learning and with 1% of annotated images only, this gives more than 3% improvement in object classification, 26% in scene graph classification, and 36% in predicate prediction accuracy.




How to Cite

Sharifzadeh, S., Moayed Baharlou, S., & Tresp, V. (2021). Classification by Attention: Scene Graph Classification with Prior Knowledge. Proceedings of the AAAI Conference on Artificial Intelligence, 35(6), 5025-5033.



AAAI Technical Track Focus Area on Neuro-Symbolic AI