作者投稿和查稿 主编审稿 专家审稿 编委审稿 远程编辑

计算机工程 ›› 2012, Vol. 38 ›› Issue (14): 181-183. doi: 10.3969/j.issn.1000-3428.2012.14.054

• 图形图像处理 • 上一篇    下一篇

基于隐含狄利克雷分配模型的图像分类算法

杨 赛,赵春霞   

  1. (南京理工大学计算机科学与技术学院,南京 210094)
  • 收稿日期:2011-09-08 出版日期:2012-07-20 发布日期:2012-07-20
  • 作者简介:杨 赛(1981-),女,博士研究生,主研方向:计算机视觉,机器学习;赵春霞,教授
  • 基金资助:

    国家自然科学基金资助重大项目(90820306)

Image Classification Algorithm Based on Latent Dirichlet Allocation Model

YANG Sai, ZHAO Chun-xia   

  1. (School of Computer Science and Technology, Nanjing University of Science and Technology, Nanjing 210094, China)
  • Received:2011-09-08 Online:2012-07-20 Published:2012-07-20

摘要: 概率隐含语义分析模型不适用于大规模图像数据集,为此,提出一种基于隐含狄利克雷分配模型(LDA)的图像分类算法。以BOF特征作为图像内容的初始描述,利用Gibbs抽样算法近似估算LDA模型参数,得到图像的隐含主题分布特征,并采用k近邻算法对图像进行分类。实验结果表明,与基于概率隐含语义分析模型的分类算法相比,该算法的分类性能较优。

关键词: BOF模型, 中层语义特征, 隐含狄利克雷分配模型, 隐含主题分布特征, k近邻算法, 图像分类

Abstract: To solve the problem that probabilistic Latent Semantic Analysis(pLSA) model is not suitable for large-scale image dataset, a new image classification algorithm based on Latent Dirichlet Allocation(LDA) model is proposed. It uses Bag-of-Features(BOF) model as images initial description, applies Gibbs sampling to estimate the parameters of LDA model, and gets images distribution in the latent topic space. Images are finally classified by k Nearest Neighbor(kNN) algorithm. Experimental results indicate that, compared with algorithm based on pLSA model, the image classification algorithm based on LDA has more powerful classification performances.

Key words: Bag-of-Features(BOF) model, middle-level semantic feature, Latent Dirichlet Allocation(LDA) model, latent topic distribution feature, k Nearest Neighbor(kNN) algorithm, image classification

中图分类号: