基于深度特征聚类的海量人脸图像检索  

Massive face image retrieval based on depth feature clustering

作　　者：李振东[1,2] 钟勇 张博言[1,2] 曹冬平 LI Zhendong;ZHONG Yong;ZHANG Boyan;CAO Dongping(Chengdu Institute of Computer Applications,Chinese Academy of Sciences,Chengdu,610041,China;University of Chinese Academy of Science,Beijing 100049,China)

机构地区：[1]中国科学院成都计算机应用研究所,成都610041 [2]中国科学院大学,北京100049

出　　处：《哈尔滨工业大学学报》2018年第11期101-109,共9页Journal of Harbin Institute of Technology

基　　金：四川省科技厅重点研发项目(2017SZ0010);四川省科技支撑计划项目(2016JZ0035)

摘　　要：针对海量人脸图像数据库检索时长的问题,提出了一种基于深度特征聚类的海量人脸图像检索算法.该算法首先使用人脸图像训练集对深度卷积神经网络模型进行人脸图像分类训练,在此基础上采用三元组损失方法对已训练好的人脸图像分类网络模型进行微调,使得网络能够更加有效地提取人脸图像的高层语义特征,构建更具有表征性的人脸图像深度特征.其次采用K-means聚类算法对提取的人脸图像深度特征进行聚类,使得同一个人的人脸图像能够划分到同一簇中,然后在相应的簇中进行人脸图像的深度特征相似度匹配执行人脸图像检索任务.为了进一步提高系统的检索性能,提出人脸图像深度特征融合的查询扩展方法,对待检索的人脸图像深度特征进行融合再次执行检索任务得到最终的检索结果.通过在两个人脸检索数据集(Celebrities Face Set和Labeled Faces in the Wild dataset)上进行详尽实验验证,结果表明,该算法能极大地缩小海量人脸图像数据库的检索范围,在保证一定准确率的前提下有效地提高了人脸图像检索的速度.A massive face image retrieval method based on depth feature clustering is proposed to overcome the long time retrieving in a huge scale face image database.Firstly,the convolutional neural network model is trained for face classification by face image training set.Based on it,the triplet loss method is used to fine-tune the model so that the network can be more efficient to extract high-level semantic features and construct a more representative depth features of face image.Secondly,the K-means clustering algorithm is used to cluster the extracted depth features,so that face images of the same person can be divided into the same cluster,and then similarity matching of face images is performed in the corresponding clusters to perform the retrieval task.In order to further improve the performance of system retrieval,the face image feature fusion query expansion method is proposed to fuse the depth features of the face image to be retrieved.Through exhaustive experimental verification on two face retrieval datasets(Celebrities Face Set and Labeled Faces in the Wild dataset),the results show that the proposed method can significantly reduce the retrieval range of massive face image database,thus effectively increasing the face image retrieval speed while ensuring similar retrieval accuracy.

关 键 词：图像检索 卷积神经网络 特征提取 K-MEANS聚类 

分 类 号：TP391.41[自动化与计算机技术—计算机应用技术]