HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

Improving Video-Text Retrieval by Multi-Stream Corpus Alignment and Dual Softmax Loss

Xing Cheng Hezheng Lin Xiangyu Wu Fan Yang Dong Shen

Improving Video-Text Retrieval by Multi-Stream Corpus Alignment and Dual Softmax Loss

Abstract

Employing large-scale pre-trained model CLIP to conduct video-text retrieval task (VTR) has become a new trend, which exceeds previous VTR methods. Though, due to the heterogeneity of structures and contents between video and text, previous CLIP-based models are prone to overfitting in the training phase, resulting in relatively poor retrieval performance. In this paper, we propose a multi-stream Corpus Alignment network with single gate Mixture-of-Experts (CAMoE) and a novel Dual Softmax Loss (DSL) to solve the two heterogeneity. The CAMoE employs Mixture-of-Experts (MoE) to extract multi-perspective video representations, including action, entity, scene, etc., then align them with the corresponding part of the text. In this stage, we conduct massive explorations towards the feature extraction module and feature alignment module. DSL is proposed to avoid the one-way optimum-match which occurs in previous contrastive methods. Introducing the intrinsic prior of each pair in a batch, DSL serves as a reviser to correct the similarity matrix and achieves the dual optimal match. DSL is easy to implement with only one-line code but improves significantly. The results show that the proposed CAMoE and DSL are of strong efficiency, and each of them is capable of achieving State-of-The-Art (SOTA) individually on various benchmarks such as MSR-VTT, MSVD, and LSMDC. Further, with both of them, the performance is advanced to a big extend, surpassing the previous SOTA methods for around 4.6\% R@1 in MSR-VTT.

Code Repositories

starmemda/camow
Official
pytorch
Mentioned in GitHub
starmemda/CAMoE
Official
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
video-retrieval-on-activitynetCAMoE
text-to-video Mean Rank: 6.3
text-to-video Median Rank: 1
text-to-video R@1: 51.0
text-to-video R@10: 87.6
text-to-video R@5: 77.7
video-retrieval-on-didemoCAMoE
text-to-video Mean Rank: 16.3
text-to-video Median Rank: 2.0
text-to-video R@1: 43.8
text-to-video R@10: 79.9
text-to-video R@5: 71.4
video-to-text Mean Rank: 10.2
video-to-text Median Rank: 2
video-to-text R@1: 45.5
video-to-text R@10: 80.5
video-retrieval-on-lsmdcCAMoE
text-to-video Mean Rank: 54.4
text-to-video R@1: 25.9
text-to-video R@10: 53.7
text-to-video R@5: 46.1
video-retrieval-on-msr-vttCAMoE
text-to-video Mean Rank: 42.6
text-to-video Median Rank: 3
text-to-video R@1: 32.9
text-to-video R@10: 68.4
text-to-video R@5: 58.3
video-to-text Mean Rank: 3.8
video-to-text Median Rank: 1
video-to-text R@1: 59.8
video-to-text R@10: 92.8
video-to-text R@5: 86.2
video-retrieval-on-msr-vtt-1kaCAMoE
text-to-video Mean Rank: 12.4
text-to-video Median Rank: 2
text-to-video R@1: 48.8
text-to-video R@10: 85.3
text-to-video R@5: 75.6
video-to-text Mean Rank: 9.9
video-to-text Median Rank: 2
video-to-text R@1: 50.3
video-to-text R@10: 83.8
video-to-text R@5: 74.6
video-retrieval-on-msvdCAMoE
text-to-video Mean Rank: 8.9
text-to-video Median Rank: 1
text-to-video R@1: 51.8
text-to-video R@10: 87.6
text-to-video R@5: 87.6
video-to-text Mean Rank: 3.1
video-to-text Median Rank: 1
video-to-text R@1: 69.3
video-to-text R@10: 94.6
video-to-text R@5: 90.6

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp