ImageNet classification with deep convolutional neural networks
Alex Krizhevsky, Ilya Sutskever et al.
6.6k
Citations
768
Influential Citations
International Journal of Computer Vision
Venue
2017
Year
Despite progress in perceptual tasks such as image classification, computers still perform poorly on cognitive tasks such as image description and question answering. Cognition is core to tasks that involve not just recognizing, but reasoning about our visual world. However, models used to tackle the rich content in images for cognitive tasks are still being trained using the same datasets designed for perceptual tasks. To achieve success at cognitive tasks, models need to understand the interactions and relationships between objects in an image. When asked “What vehicle is the person riding?”, computers will need to identify the objects in an image as well as the relationships riding(man, carriage) and pulling(horse, carriage) to answer correctly that “the person is riding a horse-drawn carriage.” In this paper, we present the Visual Genome dataset to enable the modeling of such relationships. We collect dense annotations of objects, attributes, and relationships within each image to learn these models. Specifically, our dataset contains over 108K images where each image has an average of $$35$$ objects, $$26$$ attributes, and $$21$$ pairwise relationships between objects. We canonicalize the objects, attributes, relationships, and noun phrases in region descriptions and questions answer pairs to WordNet synsets. Together, these annotations represent the densest and largest dataset of image descriptions, objects, attributes, relationships, and question answer pairs.
This paper addresses a critical gap in computer vision: while perceptual tasks like image classification have seen significant progress, cognitive tasks such as image description and question answering remain challenging. The authors argue that existing datasets designed for perceptual tasks (e.g., ImageNet) are insufficient for training models that need to reason about interactions and relationships between objects. By introducing Visual Genome, they provide a resource that explicitly captures the rich relational structure of visual scenes, enabling the development of models that can answer complex queries like "What vehicle is the person riding?" by understanding relationships such as riding(man, carriage) and pulling(horse, carriage).
The dataset's scale and density are unprecedented: over 108K images, each with an average of 35 objects, 26 attributes, and 21 pairwise relationships. This level of annotation supports training models for scene graph generation, visual question answering, and image captioning. The canonicalization to WordNet synsets ensures consistency across annotations, making the dataset reusable for a wide range of tasks.
The paper does not present model performance results; instead, it provides detailed statistics of the dataset: 108,077 images, 5.4 million region descriptions, 1.7 million QA pairs, 3.8 million object instances, 2.8 million attributes, and 2.3 million relationships. The average number of objects per image is 35, attributes 26, and relationships 21. These numbers demonstrate the dataset's density and coverage.
Visual Genome has become a cornerstone dataset for visual relationship detection and scene graph generation. It has been widely adopted in the research community (over 6,600 citations) and has spurred the development of models that reason about visual relationships. The dataset's canonicalization approach has influenced subsequent annotation efforts, and its focus on cognitive tasks has shifted the field toward more holistic scene understanding. By enabling models to learn not just what objects are present but how they interact, Visual Genome has advanced the frontier of visual reasoning and grounded language understanding.
Alex Krizhevsky, Ilya Sutskever et al.
Ashish Vaswani, Noam Shazeer et al.
Douglas M. Bates, Martin Mächler et al.
Diederik P. Kingma, Jimmy Ba