Simultaneous image classification and annotation

• Published in: 2009 IEEE Conference on Computer Vision and Pattern Recognition pp. 1903 - 1910
• Main Authors: Chong Wang, Blei, David, Fei-Fei Li
• Format: Conference Proceeding
• Language: English
• Published: IEEE 01.06.2009
• Subjects: Computer science; Computer vision; Image classification; Indexing; Inference algorithms; Layout; Marine animals; Predictive models; Road transportation; Traffic control
• ISBN: 1424439922; 9781424439928
• ISSN: 1063-6919; 1063-6919
• DOI: 10.1109/CVPR.2009.5206800

Image classification and annotation are important problems in computer vision, but rarely considered together. Intuitively, annotations provide evidence for the class label, and the class label provides evidence for annotations. For example, an image of class highway is more likely annotated with words "road," "car," and "traffic" than words "fish," "boat," and "scuba." In this paper, we develop a new probabilistic model for jointly modeling the image, its class label, and its annotations. Our model treats the class label as a global description of the image, and treats annotation terms as local descriptions of parts of the image. Its underlying probabilistic assumptions naturally integrate these two sources of information. We derive an approximate inference and estimation algorithms based on variational methods, as well as efficient approximations for classifying and annotating new images. We examine the performance of our model on two real-world image data sets, illustrating that a single model provides competitive annotation performance, and superior classification performance.