Imagine Before Concentration: Diffusion-Guided Registers Enhance Partially Relevant Video Retrieval
Organizations: Harbin Institute of Technology, Shenzhen · Tsinghua Shenzhen International Graduate School, Tsinghua University · Peng Cheng Laboratory
Abstract
Partially Relevant Video Retrieval (PRVR) aims to retrieve untrimmed videos based on text queries that describe only partial events. Existing methods suffer from incomplete global contextual perception, struggling with query ambiguity and local noise induced by spurious responses. To address these issues, we propose DreamPRVR, which adopts a coarse-to-fine representation learning paradigm. The model first generates global contextual semantic registers as coarse-grained highlights spanning the entire video and then concentrates on fine-grained similarity optimization for precise cross-modal matching. Concretely, these registers are generated by initializing from the video-centric distribution produced by a probabilistic variational sampler and then iteratively refined via a text-supervised truncated diffusion model. During this process, textual semantic structure learning constructs a well-formed textual latent space, enhancing the reliability of global perception. The registers are then adaptively fused with video tokens through register-augmented Gaussian attention blocks, enabling context-aware feature learning. Extensive experiments show that DreamPRVR outperforms state-of-the-art methods. Code is released at https://github.com/lijun2005/CVPR26-DreamPRVR.
Figures & tables
| Model | ActivityNet Captions | Charades-STA | TVR | ||||||||||||||
| R@1 | R@5 | R@10 | R@100 | SumR | R@1 | R@5 | R@10 | R@100 | SumR | R@1 | R@5 | R@10 | R@100 | SumR | |||
| Text-to-Video Retrieval (T2VR) Models | |||||||||||||||||
| RIVRL [ 13 ] | 5.2 | 18.0 | 28.2 | 66.4 | 117.8 | 1.6 | 5.6 | 9.4 | 37.7 | 54.3 | 9.4 | 23.4 | 32.2 | 70.6 | 135.6 | ||
| DE++ [ 12 ] | 5.3 | 18.4 | 29.2 | 68.0 | 121.0 | 1.7 | 5.6 | 9.6 | 37.1 | 54.1 | 8.8 | 21.9 | 30.2 | 67.4 | 128.3 | ||
| CLIP4Clip [ 42 ] | 5.9 | 19.3 | 30.4 | 71.6 | 127.3 | 1.8 | 6.5 | 10.9 | 44.2 | 63.4 | 9.9 | 24.3 | 34.3 | 72.5 | 141.0 | ||
| Cap4Video [ 74 ] | 6.3 | 20.4 | 30.9 | 72.6 | 130.2 | 1.9 | 6.7 | 11.3 | 45.0 | 65.0 | 10.3 | 26.4 | 36.8 | 74.0 | 147.5 | ||
| Model | Train time/epoch (ms) | Model parameters (M) | Inference time (ms) | Retrieval time (ms) | SumR |
| GMMFormer | 26887 | 12.85 | 2876 | 3238 | 72.9 |
| HLFormer | 31463 | 28.43 | 3816 | 3655 | 78.7 |
| GMMFormerV2 | 38004 | 30.79 | 3843 | 3688 | 78.2 |
| Dream PRVR | 33609 | 36.14 | 4001 | 3686 | 80.0 |
| ID | Model | ActivityNet Captions | Charades-STA | TVR | ||||||||||||||
| R@1 | R@5 | R@10 | R@100 | SumR | R@1 | R@5 | R@10 | R@100 | SumR | R@1 | R@5 | R@10 | R@100 | SumR | ||||
| (0) | Dream PRVR (full) | 8.7 | 27.5 | 40.3 | 79.5 | 156.1 | 2.6 | 8.7 | 14.5 | 54.2 | 80.0 | 17.4 | 39.0 | 50.4 | 86.2 | 193.1 | ||
| Efficacy of Register Generation Strategy | ||||||||||||||||||
| (1) | registers | 8.5 | 26.6 | 39.5 | 78.8 | 153.4 | 2.4 | 7.9 | 13.6 | 52.9 | 76.8 | 15.7 | 36.6 | 48.7 | 86.0 | 187.0 | ||
| (2) | AP | 8.5 | 26.2 | 39.3 | 77.9 | 151.9 | 2.6 | 8.6 | 14.4 | 52.6 | 78.1 | 17.0 | 38.3 | 49.6 | 86.4 | 191.4 | ||
| (3) | DRE | 8.2 | 25.7 | 38.7 | 78.0 | 150.6 | 2.4 | 8.5 | 14.8 | 52.6 | 78.3 | 16.8 | 38.2 | 49.6 | 86.3 | 190.8 | ||
Appendix figures & tables1 asset
Supplementary material from the paper’s appendix.
Appendix
Cross-Modal Retrieval
Cross-modal retrieval matches queries and items across different data types, such as text, images, video, audio, and sensor signals.
See all 184 papers