HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

RoME: Role-aware Mixture-of-Expert Transformer for Text-to-Video Retrieval

Burak Satar Hongyuan Zhu Hanwang Zhang Joo Hwee Lim

RoME: Role-aware Mixture-of-Expert Transformer for Text-to-Video Retrieval

Abstract

Seas of videos are uploaded daily with the popularity of social channels; thus, retrieving the most related video contents with user textual queries plays a more crucial role. Most methods consider only one joint embedding space between global visual and textual features without considering the local structures of each modality. Some other approaches consider multiple embedding spaces consisting of global and local features separately, ignoring rich inter-modality correlations. We propose a novel mixture-of-expert transformer RoME that disentangles the text and the video into three levels; the roles of spatial contexts, temporal contexts, and object contexts. We utilize a transformer-based attention mechanism to fully exploit visual and text embeddings at both global and local levels with mixture-of-experts for considering inter-modalities and structures' correlations. The results indicate that our method outperforms the state-of-the-art methods on the YouCook2 and MSR-VTT datasets, given the same visual backbone without pre-training. Finally, we conducted extensive ablation studies to elucidate our design choices.

Code Repositories

buraksatar/RoME_video_retrieval
Official
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
video-retrieval-on-msr-vttRoME
text-to-video Median Rank: 17
text-to-video R@1: 10.7
text-to-video R@10: 41.2
text-to-video R@5: 29.6
video-retrieval-on-youcook2RoME
text-to-video Median Rank: 53
text-to-video R@1: 6.3
text-to-video R@10: 25.2
text-to-video R@5: 16.9

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
RoME: Role-aware Mixture-of-Expert Transformer for Text-to-Video Retrieval | Papers | HyperAI