Unsupervised Person Re-identification with Stochastic Training Strategy

TL;DR

This paper introduces a stochastic training strategy for unsupervised person re-identification that improves clustering reliability and reduces camera bias, leading to more robust feature learning.

Contribution

It proposes a stochastic memory update mechanism and a unified distance matrix to enhance unsupervised re-ID performance and address clustering and camera variance issues.

Findings

Stochastic memory update reduces clustering errors.

Unified distance matrix mitigates camera bias.

Improved re-ID accuracy demonstrated on benchmark datasets.

Abstract

Unsupervised person re-identification (re-ID) has attracted increasing research interests because of its scalability and possibility for real-world applications. State-of-the-art unsupervised re-ID methods usually follow a clustering-based strategy, which generates pseudo labels by clustering and maintains a memory to store instance features and represent the centroid of the clusters for contrastive learning. This approach suffers two problems. First, the centroid generated by unsupervised learning may not be a perfect prototype. Forcing images to get closer to the centroid emphasizes the result of clustering, which could accumulate clustering errors during iterations. Second, previous methods utilize features obtained at different training iterations to represent one centroid, which is not consistent with the current training sample, since the features are not directly comparable. To this end, we propose an unsupervised re-ID approach with a stochastic learning strategy. Specifically, we adopt a stochastic updated memory, where a random instance from a cluster is used to update the cluster-level memory for contrastive learning. In this way, the relationship between randomly selected pair of images are learned to avoid the training bias caused by unreliable pseudo labels. The stochastic memory is also always up-to-date for classifying to keep the consistency. Besides, to relieve the issue of camera variance, a unified distance matrix is proposed during clustering, where the distance bias from different camera domain is reduced and the variances of identities is emphasized.