arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 1704.03373cs.CVcs.AI
收藏

用于集对集识别的质量感知网络

Quality Aware Network for Set to Set Recognition

SenseTime Group Limited(商汤集团有限公司);University of Sydney(悉尼大学)

Yu Liu, Junjie Yan, Wanli Ouyang

摘要:

本文研究的是集对集识别问题,即学习两个图像集之间的度量。每个集合中的图像属于同一身份。由于集合中的图像可以是互补的,因此它们在实际应用中有望导致更高的精度。然而,每个样本的质量无法保证,质量差的样本会损害度量。在本文中,质量感知网络(QAN)提出了面对这个问题,每个样本的质量可以自动学习,虽然这些信息是没有明确提供的训练阶段。该网络有两个分支,其中第一个分支提取每个样本的外观特征嵌入,另一个分支预测每个样本的质量分数。然后,集合中所有样本的特征和质量分数被聚合以生成最终的特征嵌入。我们证明了这两个分支可以以端到端的方式进行训练,只需要设置级别的身份注释。对该机制的梯度扩散分析表明,网络学习的质量有利于集对集的识别,简化了网络需要拟合的分布。人脸验证和人的重新识别实验表明,所提出的QAN的优势。源代码和网络结构可以在https://github.com/sciencefans/Quality-Aware-Network下载。

英文摘要:

This paper targets on the problem of set to set recognition, which learns the metric between two image sets. Images in each set belong to the same identity. Since images in a set can be complementary, they hopefully lead to higher accuracy in practical applications. However, the quality of each sample cannot be guaranteed, and samples with poor quality will hurt the metric. In this paper, the quality aware network (QAN) is proposed to confront this problem, where the quality of each sample can be automatically learned although such information is not explicitly provided in the training stage. The network has two branches, where the first branch extracts appearance feature embedding for each sample and the other branch predicts quality score for each sample. Features and quality scores of all samples in a set are then aggregated to generate the final feature embedding. We show that the two branches can be trained in an end-to-end manner given only the set-level identity annotation. Analysis on gradient spread of this mechanism indicates that the quality learned by the network is beneficial to set-to-set recognition and simplifies the distribution that the network needs to fit. Experiments on both face verification and person re-identification show advantages of the proposed QAN. The source code and network structure can be downloaded at https://github.com/sciencefans/Quality-Aware-Network.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑