Long-RL: Scaling RL to Long Videos

Yukang Chen*, Wei Huang*, Baifeng Shi, Qinghao Hu, Hanrong Ye, Ligeng Zhu, Zhijian Liu, Pavlo Molchanov, Jan Kautz, Xiaojuan Qi, Sifei Liu, Hongxu Yin, Yao Lu, Song Han
NVIDIA, MIT, HKU, UC Berkeley
(* indicates equal contribution)

News

Awards

No items found.

Competition Awards

No items found.

Abstract

We introduce a full-stack framework that scales up reasoning in vision-language models (VLMs) to long videos, leveraging reinforcement learning. We address the unique challenges of long video reasoning by integrating three critical components: (1) a large-scale dataset, LongVideo-Reason, comprising 104K long video QA pairs with high-quality reasoning annotations across diverse domains such as sports, games, and vlogs; (2) a two-stage training pipeline that extends VLMs with chain-of-thought supervised fine-tuning (CoT-SFT) and reinforcement learning (RL); and (3) a training infrastructure for long video RL, named Multi-modal Reinforcement Sequence Parallelism (MR-SP), which incorporates sequence parallelism and a vLLM-based engine tailored for long video, using cached video embeddings for efficient rollout and prefilling. In our experiments, LongVILA-R1-7B achieves strong performance on video benchmarks, reaching 65.1% and 71.1% accuracy on VideoMME without and with subtitles, respectively, and consistently outperforming LongVILA-7B across multiple benchmarks. Moreover, LongVILA-R1-7B supports processing up to 8,192 video frames per video, and configurable FPS settings. Notably, our MR-SP system achieves up to 2.1x speedup on long video RL training. In addition, we release our training system for public availability that supports RL training on various modalities (video, text, and audio), various models (VILA and Qwen series), and even image and video generation models. On a single A100 node (8 GPUs), it supports RL training on hour-long videos (e.g., 3,600 frames).

Highlights

LongVILA-R1-7B supports both multiple-choice questions and open-ended questions. It can switch between thinking and non-thinking modes. It demonstrates strong performance in long video reasoning, achieving 71.1% on VideoMME (w/ sub.) and surpassing Gemini-1.5-Pro across diverse reasoning tasks. It supports processing up to 8,192 video frames per video, with configurable FPS settings.

Long-RL is a codebase that accelerates long video RL training by up to 2.1× through its MR-SP system. It supports RL training on image, video, and omni inputs across VILA, Qwen/Qwen-VL, and diffusion models.

LongVideo-Reason is a large-scale dataset, comprising 104K long video QA pairs with high-quality reasoning annotations across diverse domains such as sports, games, and vlogs.

Examples of LongVILA-R1. The illustration demonstrates sample tasks and their reasoning. From left to right, the examples include predicting the results of a football match, decision-making reasoning in Texas Holdem Poker, and trajectory for spatial dynamics of objects. Notably, the spatial tracking video involves a relatively complex dynamic moving, for which the model fails to achieve accurate reasoning until the number of input video frames increases to 128.

Data Distribution of LongVideo-Reason and total data in the LongVILA-R1 training framework. LongVideo-Reason comprises a total of 18K videos and 104K QAs with reasoning annotations. Additionally, we include 102K QAs from existing works.

Training efficiency comparison with MR-SP (SP degree=4) on Qwen2.5-VL-7B and LongVILA-R1-7B and a single node 8x A100 GPUs. It achieves 2.1x speed-up and avoids GPU OOM issue on long frames.

The workflow of multi-modal reinforcement sequential parallel (MR-SP). To accommodate multi-modal input in reinforcement learning, we develop a custom sharding strategy to ensure balanced workload distribution and compatibility with SP communication. Efficient video embedding reuse and vLLM rollout acceleration strategies are implemented, while meeting the demands of policy model pre-filling for dense video frames.

Performance on several benchmarks. LongVILA-R1-7B outperforms LongVILA-7B across all these benchmarks, with varying margins.

Video

Citation

@inproceedings{chen2025longvila-r1,

        title={Scaling RL to Long Videos},

        author={Yukang Chen and Wei Huang and Baifeng Shi and Qinghao Hu and Hanrong Ye and Ligeng Zhu and Zhijian Liu and Pavlo Molchanov and Jan Kautz and Xiaojuan Qi and Sifei Liu and Hongxu Yin and Yao Lu and Song Han},

        booktitle={Advances in Neural Information Processing Systems (NeurIPS)},

        year={2025}

}

Media

No media articles found.

Acknowledgment

Team Members