PrGCN: Probability Prediction with Graph Convolutional Network for Person Re-Identification

Hongmin Liu,Zhenzhen Xiao,Bin Fan,Hui Zeng,Yifan Zhang,Guoquan Jiang
DOI: https://doi.org/10.1016/j.neucom.2020.10.019
IF: 6
2020-01-01
Neurocomputing
Abstract:Robust similarity measurement is an important issue for person re-identification (ReID). Most existing ReID models estimate the similarity between query and gallery images by computing their Euclidean distances while ignoring the rich context information contained in the image space. In this paper, we pro pose a graph convolutional network (GCN) based method to improve the similarity measurement in ReID, which regards the ReID task as a prediction problem of the link probability between node pairs. Our method is named as PrGCN (Probability GCN), in which each person is regarded as an instance node. Firstly, an Instance Centered Sub-graphs (ICS) is constructed for each instance node to depict its rich local context information. Secondly, the constructed ICS is input to a GCN to infer and predict the link probability of node pairs, followed by a similarity ranking between the query and gallery images according to the predicted probabilities. Extensive experiments show that the proposed method improves the mAP and Top-1 accuracy of ReID significantly, yielding better or comparable results to the state-of-the-art methods on various benchmarks (Market1501, DukeMTMC-ReID and CUHK03). In addition, we validate that the proposed PrGCN can be easily embedded into other deep learning architectures to replace Euclidean distance metric and achieve significant performance improvements. (c) 2020 Elsevier B.V. All rights reserved.
What problem does this paper attempt to address?