Task-Driven Exploration: Decoupling and Inter-Task Feedback for Joint Moment Retrieval and Highlight Detection
Jin Yang, Ping Wei, Huan Li, Ziyang Ren
TL;DR
This work addresses the joint problem of moment retrieval (MR) and highlight detection (HD) by introducing TaskWeave, a task-driven top-down framework that decouples task-specific and common representations and adds an inter-task feedback loop to exchange guidance via masks. The architecture follows a DETR-like pipeline where a task-decoupled unit produces MR and HD features, and two decoders enable cross-task refinement, with a principled, learnable joint loss that balances MR and HD via uncertainties $\delta_{mr}$ and $\delta_{hd}$. Key innovations include the task-decoupled unit (shared plus task-specific experts), moment-to-mask and highlightness-to-mask feedback, and a dynamic, task-dependent loss, all validated on QVHighlights, Charades-STA, and TVSum to achieve state-of-the-art or competitive results. These contributions demonstrate that explicit modeling of task specificity and inter-task influence, coupled with adaptive optimization, can significantly improve performance on complex, multi-task video understanding tasks, with practical impact for robust video search and content analysis tooling.
Abstract
Video moment retrieval and highlight detection are two highly valuable tasks in video understanding, but until recently they have been jointly studied. Although existing studies have made impressive advancement recently, they predominantly follow the data-driven bottom-up paradigm. Such paradigm overlooks task-specific and inter-task effects, resulting in poor model performance. In this paper, we propose a novel task-driven top-down framework TaskWeave for joint moment retrieval and highlight detection. The framework introduces a task-decoupled unit to capture task-specific and common representations. To investigate the interplay between the two tasks, we propose an inter-task feedback mechanism, which transforms the results of one task as guiding masks to assist the other task. Different from existing methods, we present a task-dependent joint loss function to optimize the model. Comprehensive experiments and in-depth ablation studies on QVHighlights, TVSum, and Charades-STA datasets corroborate the effectiveness and flexibility of the proposed framework. Codes are available at https://github.com/EdenGabriel/TaskWeave.
