TensorX
返回文献探索

Paper · arXiv 2404.00308

ST-LLM: Large Language Models Are Effective Temporal Learners

Ruyang Liu, Chen Li, Haoran Tang, Yixiao Ge, Ying Shan, Ge Li

7 upvotesMarch 30, 2024arXiv 预印本
AI 摘要

ST-LLM leverages large language models to effectively encode and understand videos for dialogue systems by integrating spatial-temporal modeling with dynamic masking and global-local input strategies, achieving state-of-the-art performance on VideoChatGPT-Bench and MVBench.

Spatial-Temporal sequence modelingdynamic masking strategyglobal-local input moduleST-LLM

Abstract

Large Language Models (LLMs) have showcased impressive capabilities in text comprehension and generation, prompting research efforts towards video LLMs to facilitate human-AI interaction at the video level. However, how to effectively encode and understand videos in video-based dialogue systems remains to be solved. In this paper, we investigate a straightforward yet unexplored question: Can we feed all spatial-temporal tokens into the LLM, thus delegating the task of video sequence modeling to the LLMs? Surprisingly, this simple approach yields significant improvements in video understanding. Based upon this, we propose ST-LLM, an effective video-LLM baseline with Spatial-Temporal sequence modeling inside LLM. Furthermore, to address the overhead and stability issues introduced by uncompressed video tokens within LLMs, we develop a dynamic masking strategy with tailor-made training objectives. For particularly long videos, we have also designed a global-local input module to balance efficiency and effectiveness. Consequently, we harness LLM for proficient spatial-temporal modeling, while upholding efficiency and stability. Extensive experimental results attest to the effectiveness of our method. Through a more concise model and training pipeline, ST-LLM establishes a new state-of-the-art result on VideoChatGPT-Bench and MVBench. Codes have been available at https://github.com/TencentARC/ST-LLM.

北京市昌平区探索星信息技术及软件开发工作室

京ICP备2026059466号
ST-LLM: Large Language Models Are Effective Temporal Learners | TensorX