DiffReID: Discriminative Diffusion Model for Object Re-Identification

Diffusion models [6] belong to a class of generative models. They have emerged as mainstream generative methods due to their exceptional ability in modeling complex data distributions. Recent studies have explored how diffusion models can be adapted for discriminative tasks [14, 15]. In fact, current works can be broadly grouped into two categories. One category of methods directly leverages pre-trained diffusion models (e.g., Stable Diffusion [16] and Imagen [17]) as feature extractors. For example, Li et al. [18] show that diffusion models can be leveraged to perform zero-shot classification without any additional training. Soumik et al. [12] and Dahye et al. [13] further extract discriminative intermediate features from diffusion models for robust segmentation and classification. While pre-trained diffusion models show great potential for discriminative tasks, their complex denoisers and large timestep requirements lead to high computational cost and long inference time. Another category of methods focuses on designing task-specific denoising models that learn dataset-specific distributions to generate both diverse and discriminative features. For instance, Du et al. [19] employ diffusion models for class-wise prototype feature generation, significantly improving few-shot classification performance. Li et al. [14] introduce diffusion models for continuous distribution transformation across object, image, and text latent spaces. These methods utilize denoisers to model task-specific data distributions, effectively balancing feature generation and computational efficiency. However, most of them naively utilize diffusion models for diverse image generation or general feature extraction. In contrast, our method constructs a discriminative feature distribution learning process for diffusion models. It can effectively utilize fine-grained conditions to generate more discriminative features.

Early object ReID methods predominantly leverage Convolutional Neural Networks (CNNs) to extract fine-grained feature representations [20, 21]. However, the inherent locality of CNNs limits their ability to capture long-range dependencies. To this end, recent studies have integrated attention mechanisms [22] for robust object ReID. For example, Zhang et al. [23] construct a global pixel attention map to obtain more robust object representations. He et al. [1] introduce a Transformer-based backbone to capture global dependencies, achieving impressive performance in object ReID. Afterwards, many Transformer-based ReID models [24, 25, 26, 27, 28] are proposed. However, image-only methods easily overfit to salient regions while neglecting essential semantic information. Therefore, many researchers [2, 3, 29] have incorporated large vision-language models into object ReID. Besides, recent studies have also explored the model generalization. For example, Li et al. [30] investigate domain generalization object ReID by leveraging cross-camera unpaired samples. Zhang et al. [31] propose heterogeneous expert collaborative learning for weakly supervised visible-infrared person ReID. Zhang et al. [32] utilize a dual-granularity cross-modal identity association for text-based object ReID. Li et al. [33] improve clustering-based unsupervised object ReID via feature calibration. However, object ReID remains constrained by the limited availability of large-scale training datasets. To mitigate this issue, recent works [34, 10] leverage diffusion models to enrich training data. For example, Si et al. [34] introduce a diffusion model-based inpainting method to generate clothing-change samples. Similarly, Niu et al. [10] leverage diffusion models to construct large-scale person ReID datasets. However, simple image generation inevitably causes the loss of identity information. To this end, some researchers leverage diffusion models for both feature extraction and image generation. For example, Wang et al. [8] integrate feature extraction and denoising into a single image encoder. Additionally, Jia et al. [35] utilize diffusion models to learn the distribution of bounding boxes and ReID features, conditioned on the provided visual features. Unlike existing works that solely utilize generated features for objects, we explore the mutual learning between generated features and real visual features for more robust and discriminative representations.

The overall framework adopts a two-stage training procedure. In the first stage, we aim to obtain identity-aware text features by prompt tuning. As shown in Fig. 2(a), with the CLIP model, we freeze the pre-trained text encoder and image encoder, only optimizing prompt tokens. Inspired by [2], these prompt tokens are passed through the text encoder with the template “A photo of a [x] [x] [x] [x] person/vehicle”. To obtain identity-aware text features, we employ the following bidirectional contrastive losses:

where $P(y)$ is the image set of the same identity $y$. $B$ is the batch size. $f_{d}$ and $f_{v}$ are text and visual features, respectively.

In the second stage, we feed person images into the learnable image encoder to extract visual tokens $\mathcal{F}\in\mathbb{R}^{N\times D}$. They are sent to the VNG and LWD for noise initialization and feature generation, respectively. It involves two key processes: a forward noising process and a backward denoising process. Unlike most diffusion models [35], we integrate textual semantics into the denoising process. Specifically, as shown in the red line of Fig. 2, the forward noising process gradually adds visual-guided noises $\varepsilon$ to the text feature $f_{d}$ over $T$ timesteps. It should be noted that VNG generates visual-guided noises based on visual priors. At each timestep, visual-guided noises are injected with a timestep-dependent variance $\alpha_{t}$:

where $\bar{\alpha}_{t}=\prod_{i=1}^{t}\alpha_{i}$. This process ultimately transforms $f_{d}$ into the final state $\bar{f}_{T}$, which closely approximates its visual-aware Gaussian distribution $\mathcal{N}(\mu,\sigma^{2})$. Noting that, $\mu$ and $\sigma^{2}$ are obtained by the VNG. As shown in the blue line of Fig. 2, we then treat the visual feature as a condition to the backward denoising process. It aims at gradually denoising $p(\bar{f}_{T})$ to $q(\bar{f}_{0})$ as follows:

where $p_{\theta}(\bar{f}_{t-1}\mid\bar{f}_{t},\hat{c}_{t})$ is the backward denoising distribution. $f_{d}\sim q(\bar{f}_{0})$, $\theta$ denotes learnable parameters, and $\hat{c}_{t}$ is a timestep-specific condition. To obtain guided features, our framework first generates an initial noise state $\bar{f}_{T}$, and then progressively samples $\bar{f}_{t-1}$ from $p_{\theta}(\bar{f}_{t-1}\mid\bar{f}_{t},\hat{c}_{t})$ at each denoising timestep $h\in\{1,\ldots,T\}$. After all denoising timesteps, the final denoised state $\bar{f}_{0}$ is taken as the guided feature $g$. Here, $t$ and $h$ are the numbers of noising and denoising steps, respectively. They satisfy $T=t+h$. Note that, the above diffusion process is performed in the latent feature space rather than the image space. It avoids reconstructing high-dimensional pixel-level details and provides a more compact semantic guidance for identity-aware distribution learning. Moreover, we perform the diffusion process on identity-aware text features rather than visual features. In fact, visual features may contain appearance-biased cues, such as clothing colors, bags, or backgrounds, while identity-aware text features provide a compact semantic distribution aligned with identity labels. Therefore, with visual features as conditions and identity-aware text features as diffusion targets, our framework can generate identity-aware guided features that preserve visual discrimination while benefiting from semantic guidance.

As demonstrated in [14], the noise initialization significantly affects the generation process of diffusion models. In our framework, the initialized noise is used to corrupt identity-aware text features and determines the starting state of the denoising process. Therefore, the quality of initial noises directly influences whether the guided features can preserve identity-related information. However, most existing methods [6, 19] sample noises directly from a standard Gaussian distribution. Such randomly sampled noises are sample-agnostic and contain no visual prior related to the input image. They may introduce perturbations inconsistent with the object identity and degrade the discrimination of generated features. To address this issue, we introduce the Vision-guided Noise Generator (VNG) to estimate the mean and variance of noise distributions, as shown in Fig. 2(b). Instead of using a fixed Gaussian prior, our VNG estimates the mean and variance of the noise distribution $p_{\phi}(\bar{f}_{T}|\hat{f})$ based on the visual feature $\hat{f}$. The noise is then sampled from this learned distribution, making it more adaptive for feature generation. In particular, we use two fully connected layers to transform the visual feature $\hat{f}$ into the mean $\mu$ and variance $\sigma^{2}$:

where $W_{\mu}$, $W_{\sigma}$, $b_{\mu}$ and $b_{\sigma}$ are learnable parameters. Then, we sample initial visual-guided noises from the distribution $p_{\phi}(\bar{f}_{T}|\hat{f})$ using the reparameterization trick [37]:

where $\epsilon\sim\mathcal{N}(0,I)$ and $\otimes$ denotes the element-wise multiplication. To align with the distribution assumption in diffusion models [6], we impose a Kullback-Leibler (KL) divergence regularization to constrain $p_{\phi}(\bar{f}_{T}|\hat{f})$ toward a standard Gaussian distribution. With the proposed VNG, our framework can take visual features as priors and gradually corrupt identity-aware text features.

Many works [12, 13, 38] have demonstrated that diffusion models can learn comprehensive data distributions, improving the robustness of generated features. However, applying complex diffusion models to discriminative tasks presents two major challenges: (a) the inconsistency between training and inference processes; (b) the high computational cost during inference. To address these issues, we propose the Light Weight Denoiser (LWD) to denoise identity-aware text features, while maintaining computational efficiency. As shown in Fig. 2(c), LWD utilizes several Multilayer Perceptron (MLP) to transform identity-aware text features instead of complex networks. To enhance the modeling capacity, we stack $M$ denoisers for LWD. Note that, LWD is lightweight because our diffusion process is performed in a compact latent feature space for identity-aware distribution learning [19, 14]. Unlike image-based diffusion models that require complex denoising networks and long denoising trajectories, LWD models the transformation between semantic feature representations. Therefore, an MLP-based lightweight denoiser with a small number of timesteps is sufficient to generate discriminative features while maintaining computational efficiency. To obtain the condition of LWD, we first aggregate the intermediate features from the image encoder and then pass them through a linear layer to obtain the condition representation $\check{\mathcal{F}}$,

where $f_{l}$ represents the output from the $l$-th Transformer layer and $W_{1}$ is the learnable parameter of a linear layer.

During the denoising process, the features change progressively across timesteps, and different stages require different types of visual guidance. At early timesteps, the corrupted features contain stronger noise, so the denoiser mainly needs coarse and global contextual information to recover the overall identity-related structure. At later timesteps, the features are gradually refined, and the denoiser requires more fine-grained and identity-discriminative cues to further improve feature discrimination. Therefore, using the same condition for all timesteps may limit the ability of LWD to adapt to the evolving feature state. To address this issue, we introduce a set of learnable Temporal Condition Queries (TCQ) to obtain timestep-specific conditions for LWD. Specifically, at each timestep $t$, we initialize a set of learnable queries $\mathcal{Q}=\{q^{1}_{t},q^{2}_{t},\dots,q^{N}_{t}\}$, where $N$ denotes the number of queries per timestep. Inspired by [39], we introduce the noise $\bar{f}_{t}$ as a prior into the corresponding condition queries to stabilize training and accelerate convergence:

where $t\in\{1,2\cdots,T\}$, $n\in\{1,2\cdots,N\}$ and $W_{2}$ is the learnable parameter. Finally, these temporal queries $\check{q}^{n}_{t}$ interact with the condition representation $\check{\mathcal{F}}$ through a cross-attention to obtain the condition features at each timestep:

where $W_{q},W_{k}$ and $W_{v}$ are learnable parameters. $D$ is the feature dimension for scaling. As shown in Fig. 2(c), we obtain the timestep-specific condition $\hat{c}_{t}$ by averaging the condition features as follows:

Then, the noised feature $\bar{f}_{t}$, the time embedding $v_{t}$, and the timestep-specific condition $\hat{c}_{t}$ are jointly processed to denoise the corrupted identity-aware text features:

where $W_{3}$, $W_{4}$, $W_{5}$ and $W_{6}$ are learnable parameters. With LWD, we can take visual features as conditions and denoise the corrupted text features step-by-step for identity-aware distribution learning.

Technically, most of diffusion model-based feature learning methods [8, 19] directly adopt generated features as the final representations. However, they overlook the complementary between features extracted by general visual encoders and those generated by diffusion models. In fact, features extracted by general visual encoders capture discriminative information but tend to overfit to salient appearance patterns [40]. In contrast, generated features of diffusion models are encouraged to follow the identity-aware feature distribution [38], making them more robust and semantic. To better integrate the advantages of both features, we introduce the Mutual Enhancement Constraint (MEC), which encourages visual features and generated features to learn from each other. Specifically, generated features provide identity-aware semantic cues to improve the generalization of visual features. Visual features provide image-specific evidence to constrain the generated features and suppress unreliable semantic deviations.

In each iteration, the diffusion model first generates guided features that follow the identity-aware distribution. Then, these guided features and visual features mutually refine each other. Specifically, for each input image $I_{b}$, we obtain a feature set as follows:

where $b\in\{1,2,\cdots,B\}$, $\hat{f}$ is the visual feature and $g^{s}$ is the guided feature. $S$ is the total number of guided features. Then, we can construct the feature set within each batch:

To achieve mutual learning, we introduce the MEC:

where $D_{a,p}$ and $D_{a,n}$ are the distances between the anchor–positive and anchor–negative pairs within $\mathcal{V}$, respectively. $m_{1}$ is the margin parameter. As observed, MEC combines the visual and guided features into a joint feature set and constructs cross-feature triplets. The triplet loss pulls each guided feature closer to the same-identity visual feature and pushes it away from different identity features. Thus, the visual feature acts as a stable identity reference that reduces stochastic shifts of the guided feature toward incorrect identities. Meanwhile, the guided feature introduces distributional variations that encourage the visual feature to retain identity-consistent cues. Through this bidirectional interaction, the advantages of visual and guided features are jointly amplified, enabling the framework to learn more discriminative representations.

Model Training. In the first stage, we follow previous works [2, 41] and apply the bidirectional contrastive losses for prompt tuning:

In the second stage, we employ multiple losses to supervise the framework. To optimize the visual feature $\hat{f}$, we adopt the following combined loss:

where $L_{ce}$ is the identity loss. $L_{tri}$ is the triplet loss. $\gamma_{1}$ and $\gamma_{2}$ are hyper-parameters to balance the loss terms.

As for the diffusion procedure, we supervise the noise prediction at each timestep by the following loss:

As for guided features, we further ensure that they have the same identity with the following loss:

To align with the distribution assumption, the Kullback-Leibler (KL) divergence loss is reformulated as:

Thus, the overall loss for the second stage is formulated as:

Model Inference. We extract the visual feature $\hat{f}$ and use LWD to generate guided features from visual-guided noises. The final feature $f_{r}$ is formed by averaging the guided features:

Since each guided feature is sampled from the learned identity-aware distribution, a single guided feature may only reflect one possible representation of this distribution. Thus, we average multiple guided features to obtain a more comprehensive and stable representation.

To comprehensively evaluate the effectiveness of our proposed framework, we conduct experiments on five large-scale object ReID datasets, i.e., Market1501 [62], DukeMTMC [63], MSMT17 [64], CUHK03 [65] and VeRi-776 [66]. The details of these datasets can be found in corresponding references. To assess the generalization ability, we employ two domain generalization settings [67]. The first setting involves training on one dataset and testing on another unseen dataset. The second setting utilizes the training sets of multiple datasets for training and evaluates on the test set of an unseen dataset. Following previous works [20, 1], we use mean Average Precision (mAP) and Cumulative Matching Characteristics (CMC) at Rank-1 as our evaluation metrics.

We implement our framework using the PyTorch toolbox. All experiments are conducted on a single NVIDIA RTX 4090 GPU with 24GB memory. We adopt the pre-trained CLIP [36] as the feature extraction backbone for both images and texts. Person images are resized to $256\times 128$ and vehicle images are resized to $256\times 256$. In the first stage, we employ the Adam optimizer [68] with an initial learning rate of $3.5\times 10^{-4}$ and cosine decay. Training is performed with a batch size of 64, without augmentation, only optimizing the learnable prompt tokens. In the second stage, a mini-batch of 64 images is sampled, which contains 16 identities and each with 4 images. Data augmentation consists of random cropping, horizontal flipping, and random erasing [69]. We train the model for 60 epochs using Adam, setting the initial learning rate to $5\times 10^{-6}$ for the image encoder and $5\times 10^{-4}$ for the discriminative diffusion model. The model is warmed up for 10 epochs, with the learning rate reduced by a factor of 0.1 at the 30th and 50th epochs. The hyper-parameters $\gamma_{1}$, $\gamma_{2}$, $m_{1}$, $T$ and $M$ are set to 0.25, 1.0, 0.3, 10 and 3, respectively.

In Tab. I and Tab. II, our method is compared with other state-of-the-art methods on five benchmarks under both single domain and domain generalization settings. Note that no post-processing techniques are employed in these experiments.

Single Domain Comparison. As shown in Tab. I, our method achieves state-of-the-art results on Market1501, MSMT17, DukeMTMC and VeRi-776. More remarkably, our method achieves $79.0\%$ in mAP on MSMT17, outperforming most of the compared methods, e.g., DenoiseRep [8] and CLIP-ReID [2] by $2.7\%$ and $3.2\%$, respectively. In fact, DenoiseRep [8] also introduces diffusion models for person ReID. However, its complex denoising process and long diffusion steps may hinder model convergence, ultimately limiting performance improvements. CLIP-ReID [2] introduces the large-scale vision-language models into object ReID. However, due to the lack of class-wise descriptions, CLIP-ReID tends to focus on semantic patterns rather than identity-aware feature distributions. Unlike these methods, our method introduces a diffusion model to learn identity-aware feature distributions and further generate diverse guided features, leading to more generalized and robust representations.

Domain Generalization Comparison. As shown in Tab. II, we evaluate the generalization of different methods in different domain generalization settings. The results show that our method achieves competitive performance compared with other domain generalization methods. Especially, our method achieves the best performance in the setting of Market1501+DukeMTMC+MSMT17$\rightarrow$CUHK03, reaching 50.7% mAP, which surpasses OGNorm by 10.4%. These results indicate that our method effectively learns identity-aware feature distributions and prevents the model from overfitting to semantic regions.

In this section, we perform ablation studies on MSMT17 and DukeMTMC datasets to investigate the effect of key components and hyper-parameters.

Effect of Key Components. Tab. III presents the ablation results with key components. The results show that using the LWD achieves performance gains on the MSMT17 and DukeMTMC, consistently. Furthermore, with VNG, the model achieves an additional improvement of $1.1\%$ mAP on MSMT17 and $0.6\%$ mAP on DukeMTMC. It confirms the effectiveness of adaptive noise initialization in enhancing feature discrimination. With MEC, the model further improves mAP by 0.6% on MSMT17 and 0.7% on DukeMTMC. These results demonstrate that mutually learning visual features and guided features allows them to complement each other. Finally, incorporating all components achieves the best results: $79.0\%$ mAP on MSMT17 and $85.6\%$ mAP on DukeMTMC. The consistent improvements across all configurations validate the effectiveness of our proposed components.

Effect of the Total Number of Diffusion Steps. To analyze the effect of the total number of diffusion steps $T$, we conduct additional experiments in Tab. IV. As observed, increasing $T$ to 2 significantly boosts performance, reaching $77.0\%$ mAP on MSMT17 and $84.4\%$ mAP on DukeMTMC. From $T=4$ to $T=8$, our model maintains competitive performance. There is a trend that the discrimination of guided features gradually improves as $T$ increases. With $T=10$, our model yields the best performance. This result indicates that even a small total number of diffusion steps can enhance feature representation.

Effect of Mutual Enhancement in MEC. In Tab. V, we show the effect of MEC with visual and guided features on MSMT17. The MEC consistently improves the performance with both visual and guided features. We further evaluate the intra-class and inter-class cosine similarities. For each input image, three guided features are generated using independently sampled noises, and their average pairwise cosine similarity is used to measure the generation consistency. With MEC, the intra-class similarity increases, while the inter-class similarity decreases. The generation consistency also increases. These results indicate that MEC produces more discriminative and stable guided features. These results demonstrate that MEC benefits both features and support their mutual enhancement.

Effect of Guided Features. In Tab. VI, we evaluate the effect of guided features. As observed, using one guided feature can improve the performance over the baseline. This suggests that introducing guided features enhances the robustness of representation learning. Increasing the number of guided features to two leads to further improvements, indicating that diverse guided features contribute to a more comprehensive feature. When using three or four guided features, the model consistently achieves better performance. To balance the performance and efficiency, we adopt three guided features as our default setting.

Effect of Different Numbers of TCQ. In Tab. VII, we evaluate the effect of using different numbers of TCQ. As the number of TCQ increases from 2 to 16, the performance consistently improves on both MSMT17 and DukeMTMC. This trend indicates that richer temporal conditions help the denoiser better model the evolving feature distribution across diffusion steps. In particular, using 16 queries achieves the best results. These results confirm the effectiveness of TCQ in enhancing the discrimination of guided features.

Effect of Intermediate Features for the Generation. We investigate the impact of incorporating intermediate-layer features from the backbone as conditions for the denoiser. In Tab. VIII, $f_{12}$ refers to using only the output from the 12-th layer as the condition, while $f_{12}+f_{11}$ indicates the addition of the outputs from the 11-th and 12-th layers as the condition. These results demonstrate that features from earlier layers consistently improve the performance. This suggests that intermediate-layer features provide additional information that enhances the generation. Meanwhile, the results from the last line of Tab. VIII also indicate that incorporating too many intermediate-layer features leads to a noticeable performance drop. A plausible explanation is that early-layer features mainly capture low-level patterns and lack semantic discrimination, which may reduce the representational ability.

Effect of Different Random Seeds. To evaluate the training robustness, we conduct additional experiments with four different random seeds. As shown in Tab. IX, our method consistently outperforms the baseline under all random seeds. Specifically, the mAP ranges from 78.4% to 79.0%, and the Rank-1 accuracy ranges from 90.7% to 91.4%. The performance fluctuation is very small, indicating that our method is not sensitive to the random seed initialization. These results demonstrate that the improvement of our method is stable and does not come from a specific random seed initialization.

Effect of Different Diffusion Features. As shown in Tab. X, performing diffusion on visual features improves the baseline from 75.4% to 77.8% mAP and from 89.3% to 90.5% Rank-1 on MSMT17. This indicates that diffusion-based feature distribution learning is beneficial for discriminative representation learning. Furthermore, performing diffusion on identity-aware text features achieves better performance, reaching 79.0% mAP and 91.2% Rank-1. These results demonstrate that identity-aware text features provide a more suitable semantic space for diffusion modeling. As a result, the model can learn more discriminative and generalizable identity-aware feature distributions rather than appearance-biased visual feature distributions.

Effect of the KL Regularization. As shown in Tab. XI, removing the KL regularization still achieves 77.6% mAP and 90.8% Rank-1, outperforming the baseline by 2.2% mAP and 1.5% Rank-1. This indicates that the visual-guided noise initialization in VNG is beneficial for discriminative feature generation. With the KL regularization, the performance is further improved to 79.0% mAP and 91.2% Rank-1. These results demonstrate that the KL regularization helps constrain the learned noise distribution and stabilizes the diffusion process, leading to more effective guided-feature generation.

Tab. XII compares the computational cost of different methods. Note that, our LWD adopts an MLP-based structure instead of the complex U-Net. Thus, it is more efficient than previous diffusion-based methods, such as SD-ReID [78]. Compared with other methods, our method has a comparable computational overhead but achieves superior ReID performance. These results demonstrate that our method provides a trade-off between computational cost and performance.

To better understand our proposed framework, we present comprehensive qualitative results in this subsection.

Top-5 Retrieval Results. Fig. 3 shows retrieval results with CLIP-ReID [2] and our method. In general, our method captures more fine-grained information while reducing the influence of local salient features. In Fig. 3(a) and (b), CLIP-ReID is distracted by visually prominent attributes, such as similar black clothing and white backpack, leading to incorrect matches. In contrast, our method focuses on more comprehensive information. It avoids over-reliance on a few specific attributes and successfully identifies the correct match among visually similar samples. A similar pattern is observed in Fig. 3(c) and (d). In Fig. 3(d), even though the person in the query image carries a white backpack, our method does not get misled by this prominent detail. In Fig. 3(c), despite significant distortions in the query image, our method still accurately retrieves the correct match.

Visualization of Feature Distribution. We further analyze the feature distributions on the MSMT17 datasets. Specifically, as shown in Fig. 5, we randomly sample twenty identities from the MSMT17 test set and visualize their feature distributions using t-SNE [79]. Each point represents a sample and different colors represent different identities. Compared with the baseline, our DiffReID can better reduce intra-class distances and increase inter-class distances. The main reason is that our DiffReID directly captures the identity-aware distribution rather than the salient semantic regions.

Furthermore, we visualize the cosine similarity distributions on the MSMT17 test set to analyze the discrimination of different feature representations. As shown in Fig. 6(a), the baseline exhibits a small distribution gap between positive and negative pairs. It indicates that the baseline struggles to distinguish samples with similar visual patterns. When our DiffReID is introduced with random Gaussian noises, as shown in Fig. 6(b), the distribution gap becomes much larger. It demonstrates that diffusion-based guided feature generation improves feature discrimination. Moreover, when visual-guided noises are adopted in Fig. 6(c), the negative-pair distribution further shifts to a lower similarity range while the positive-pair distribution remains in a high-similarity range. As a result, the distribution gap increases from 0.5 to 0.598. This indicates that VNG provides a more suitable noise initialization for generating discriminative features. It improves the inter-identity separation while preserving the intra-identity consistency. Overall, these results show that our DiffReID captures identity-aware distributions and obtains more discriminative representations.

Visualization of Guided Features Across the Diffusion Process. To investigate how the model gradually captures the identity-aware feature distribution, we visualize guided features at successive denoising timesteps using t-SNE, as shown in Fig. 4. Specifically, we randomly sample ten identities from the MSMT17 test set. Each point represents a sample and different colors denote different identities. At the early denoising timesteps, the guided features exhibit large variance and strong overlap across identities, indicating that they primarily encode coarse and noisy structural information. As the diffusion denoising process progresses, the features begin to form clearer local structures, and samples belonging to the same identity become increasingly aggregated. In the later denoising timesteps, the guided features evolve into well-separated and compact clusters for each identity. It demonstrates that the denoiser effectively captures discriminative semantic cues and refines the representation toward the underlying identity-aware distribution. These observations confirm that our model progressively improves the discrimination of guided features through step-wise denoising. Noted that, the feature shift between $h=6$ and $h=7$ arises from the frequency-dependent reconstruction behavior in the diffusion model [80]. Specifically, early denoising steps mainly recover low-frequency components that encode global structures. Later denoising steps progressively restore high-frequency details, introducing fine-grained identity cues. Thus, it reflects the shift from low-frequency structure to high-frequency identity, rather than instability in the denoising process.

Visual Effect of TCQ. To better understand the effect of TCQ, we visualize attention maps of learned queries at different timesteps. As shown in Fig. 7, our method attends to different information at different timesteps. In early denoising steps, it focuses more on the global structure of persons. In later denoising steps, it gradually shifts attention to discriminative regions highlighted by the red boxes. Moreover, as shown in Fig.7, our method can effectively suppress irrelevant distractions. Notably, in the last row of Fig. 7, even when an irrelevant person appears, our method still focuses on the correct target. In summary, these visualization results indicate that our method progressively generates discriminative features while avoiding the influence of irrelevant contents.