Knowledge-Based Topic Model for Unsupervised Object Discovery and Localization

• Published in: IEEE transactions on image processing Vol. 27; no. 1; pp. 50 - 63
• Main Authors: Niu, Zhenxing, Hua, Gang, Wang, Le, Gao, Xinbo
• Format: Journal Article
• Language: English
• Published: United States IEEE 01.01.2018
• Subjects: Buildings; Coherence; Computational modeling; Image segmentation; latent Dirichlet allocation; Object discovery; object localization; Resource management; Semantics; topic model; Visualization; Image segmentation; Visualization; Computational modeling; Buildings; Semantics; Coherence; Resource management
• ISSN: 1057-7149; 1941-0042
• DOI: 10.1109/TIP.2017.2718667

Unsupervised object discovery and localization is to discover some dominant object classes and localize all of object instances from a given image collection without any supervision. Previous work has attempted to tackle this problem with vanilla topic models, such as latent Dirichlet allocation (LDA). However, in those methods no prior knowledge for the given image collection is exploited to facilitate object discovery. On the other hand, the topic models used in those methods suffer from the topic coherence issue-some inferred topics do not have clear meaning, which limits the final performance of object discovery. In this paper, prior knowledge in terms of the so-called must-links are exploited from Web images on the Internet. Furthermore, a novel knowledge-based topic model, called LDA with mixture of Dirichlet trees, is proposed to incorporate the must-links into topic modeling for object discovery. In particular, to better deal with the polysemy phenomenon of visual words, the must-link is re-defined as that one must-link only constrains one or some topic(s) instead of all topics, which leads to significantly improved topic coherence. Moreover, the must-links are built and grouped with respect to specific object classes, thus the must-links in our approach are semantic-specific, which allows to more efficiently exploit discriminative prior knowledge from Web images. Extensive experiments validated the efficiency of our proposed approach on several data sets. It is shown that our method significantly improves topic coherence and outperforms the unsupervised methods for object discovery and localization. In addition, compared with discriminative methods, the naturally existing object classes in the given image collection can be subtly discovered, which makes our approach well suited for realistic applications of unsupervised object discovery.