TempR1: Improving Temporal Understanding of MLLMs via Temporal-Aware Multi-Task Reinforcement Learning
Tao Wu, Li Yang, Gen Zhan, Yabin Zhang, Yiting Liao, Junlin Li, Deliang Fu, Li Zhang, Limin Wang
TL;DR
TempR1 tackles the challenge of robust temporal understanding in multimodal large language models by training a single model across five interconnected temporal tasks with a unified, task-aware reinforcement learning objective. It introduces GRPO-based multi-task RL with a structured reward scheme, including a machine-parsable format reward and three-category temporal localization rewards to align predictions with ground truth. Leveraging a 60K multi-task corpus, TempR1 demonstrates state-of-the-art performance across TG, VHD, TAL, DTG, and GVQA benchmarks and shows clear synergistic gains from joint task optimization. This work provides a scalable, principled framework for temporal reasoning in MLLMs with broad generalization across diverse temporal scenarios.
Abstract
Enhancing the temporal understanding of Multimodal Large Language Models (MLLMs) is essential for advancing long-form video analysis, enabling tasks such as temporal localization, action detection, and time-sensitive question answering. While reinforcement learning (RL) has recently been explored for improving temporal reasoning, existing approaches are often confined to limited task types and data, restricting their generalization across diverse temporal understanding scenarios. To address this challenge, we present TempR1, a temporal-aware multi-task reinforcement learning framework that systematically strengthens MLLMs' temporal comprehension. We curate a multi-task corpus that exposes the model to diverse temporal structures and semantics, and build upon the Group Relative Policy Optimization (GRPO) algorithm to achieve stable and effective cross-task optimization. Specifically, we categorize temporal tasks into three correspondence types between predicted intervals and ground-truth instances, and design tailored localization rewards for each, enabling TempR1 to capture fine-grained temporal dependencies and adapt to different temporal patterns. Extensive experiments demonstrate that TempR1 attains state-of-the-art performance across multiple benchmarks. Moreover, its joint optimization over complementary tasks yields a strong synergistic effect, enhancing both generalization and single-task performance, establishing a scalable and principled paradigm for temporal reasoning in MLLMs.
