论文标题

学习通过信心和连通性估计来群集面孔

Learning to Cluster Faces via Confidence and Connectivity Estimation

论文作者

Yang, Lei, Chen, Dapeng, Zhan, Xiaohang, Zhao, Rui, Loy, Chen Change, Lin, Dahua

论文摘要

面部聚类是利用未标记的面部数据的重要工具,并且具有广泛的应用程序,包括面部注释和检索。最近的作品表明,监督聚类可能会导致明显的性能增长。但是,它们通常涉及启发式步骤,并且需要大量重叠的子图,从而严重限制了它们的准确性和效率。在本文中,我们提出了一个完全可以学习的聚类框架,而无需大量重叠的子图。相反,我们将聚类问题转换为两个子问题。具体而言,两个名为GCN-V和GCN-E的图形卷积网络旨在估计顶点的置信度和边缘的连通性。凭借顶点的置信度和边缘连接,我们可以自然地在亲和力图上组织更多相关的顶点,并将它们分组为簇。在两个大规模基准上进行的实验表明,我们的方法显着提高了聚类的精度,从而表现了在顶部训练的识别模型的性能,但它比现有监督方法更有效。

Face clustering is an essential tool for exploiting the unlabeled face data, and has a wide range of applications including face annotation and retrieval. Recent works show that supervised clustering can result in noticeable performance gain. However, they usually involve heuristic steps and require numerous overlapped subgraphs, severely restricting their accuracy and efficiency. In this paper, we propose a fully learnable clustering framework without requiring a large number of overlapped subgraphs. Instead, we transform the clustering problem into two sub-problems. Specifically, two graph convolutional networks, named GCN-V and GCN-E, are designed to estimate the confidence of vertices and the connectivity of edges, respectively. With the vertex confidence and edge connectivity, we can naturally organize more relevant vertices on the affinity graph and group them into clusters. Experiments on two large-scale benchmarks show that our method significantly improves clustering accuracy and thus performance of the recognition models trained on top, yet it is an order of magnitude more efficient than existing supervised methods.

扫码加入交流群

加入微信交流群

微信交流群二维码

扫码加入学术交流群,获取更多资源