UMT: Unified Multi-modal Transformers for Joint Video Moment Retrieval and Highlight Detection

Code

• github.com/tencentarc/umt
  OfficialIn paperpytorch★ 237
• github.com/MS-P3/code7/tree/main/umt5
  mindspore★ 0
• github.com/MindCode-4/code-5/tree/main/umt5
  mindspore★ 0

Abstract

Finding relevant moments and highlights in videos according to natural language queries is a natural and highly valuable common need in the current video content explosion era. Nevertheless, jointly conducting moment retrieval and highlight detection is an emerging research topic, even though its component problems and some related tasks have already been studied for a while. In this paper, we present the first unified framework, named Unified Multi-modal Transformers (UMT), capable of realizing such joint optimization while can also be easily degenerated for solving individual problems. As far as we are aware, this is the first scheme to integrate multi-modal (visual-audio) learning for either joint optimization or the individual moment retrieval task, and tackles moment retrieval as a keypoint detection problem using a novel query generator and query decoder. Extensive comparisons with existing methods and ablation studies on QVHighlights, Charades-STA, YouTube Highlights, and TVSum datasets demonstrate the effectiveness, superiority, and flexibility of the proposed method under various settings. Source code and pre-trained models are available at https://github.com/TencentARC/UMT.

Tasks

Benchmark Results

Reproductions