Set Augmented Triplet Loss for Video Person Re-Identification

Publication Date: 1/5/2021

Event: WACV 2021, Virtual

Reference: pp. 1- 10, 2021

Authors: Pengfei Fang, Australian National University/DATA61-CSIRO; Pan Ji, NEC Laboratories America, Inc.; Lars Petersson, DATA61-CSIRO; Mehrtash Harandi, Monash University

Abstract: Modern video person re-identification (re-ID) machines are often trained using a metric learning approach, supervised by a triplet loss. The triplet loss used in video re-ID is usually based on so-called clip features, each aggregated from a few frame features. In this paper, we propose to model the video clip as a set and instead study the distance between sets in the corresponding triplet loss. In contrast to the distance between clip representations, the distance between clip sets considers the pair-wise similarity of each element (i.e., frame representation) between two sets. This allows the network to directly optimize the feature representation at a frame level. Apart from the commonly-used set distance metrics (e.g., ordinary distance and Hausdorff distance), we further propose a hybrid distance metric, tailored for the set-aware triplet loss. Also, we propose a hard positive set construction strategy using the learned class prototypes in a batch. Our proposed method achieves state-of-the-art results across several standard benchmarks, demonstrating the advantages of the proposed method.

Publication Link: https://ieeexplore.ieee.org/document/9423126