Awesome
<img src="docs/images/logos/logo.png" height="40"> PG-Video-LLaVA: Pixel Grounding in Large Multimodal Video Models
Shehan Munasinghe* , Rusiru Thushara* , Muhammad Maaz , Hanoona Rasheed, Salman Khan, Mubarak Shah, Fahad Shahbaz Khan.
*Equal Contribution
Mohamed bin Zayed University of Artificial Intelligence, UAE
π’ Latest Updates
- π¦ 27-Dec-2023: Code, models released! π
<img src="docs/images/logos/logo.png" height="25"> Overview
PG-Video-LLaVA is the first video-based Large Multimodal Model (LMM) with pixel-level grounding capabilities. π₯π₯π₯
<p align="center"> <img src="docs/images/figures/teaser.png" width="70%" alt="PG-Video-LLaVA Architectural Overview"> </p>π Contributions
The key contributions of this work are:
-
We propose PG-Video-LLaVA, the first video-based LMM with pixel-level grounding capabilities, featuring a modular design for enhanced flexibility. Our framework uses an off-the-shelf tracker and a novel grounding module, enabling it to spatially ground objects in videos following user instructions.
-
We introduce a new benchmark specifically designed to measure prompt-based object grounding performance.
-
By incorporating audio context, PG-Video-LLaVA significantly enhances its understanding of video content, making it more comprehensive and aptly suited for scenarios where the audio signal is crucial for video understanding (e.g., dialogues and conversations, news videos, etc.).
-
We introduce improved quantitative benchmarks for video-based conversational models. Our benchmarks utilize open-source Vicuna LLM to ensure better reproducibility and transparency. We also propose benchmarks to evaluate the grounding capabilities of video-based conversational models.
<img src="docs/images/logos/logo.png" height="25"> PG-Video-LLaVA : Architecture
<p align="center"> <img src="docs/images/figures/1-architecture.png" alt="PG-Video-LLaVA Architectural Overview"> </p>Installation and CLI Demo
For installation and setting up the CLI demo, please refer to the instructions here.
Training
For training, please refer to the instructions here.
Qualitative Analysis π
Video Grounding π―
Our framework uses an off-the-shelf tracker and a novel grounding module, enabling it to localize objects in videos following user instructions.
<p align="center"> <img src="docs/images/figures/grounding-qual.png" alt="Video-Grounding Qualitative Results"> </p>Including Audio Modality π§
By incorporating audio context, PG-Video-LLaVA significantly enhances its understanding of video content, making it more comprehensive and aptly suited for scenarios where the audio signal is crucial for video understanding (e.g., dialogues and conversations, news videos, etc.).
<p align="center"> <img src="docs/images/figures/audio-qual.png" alt="Qualitative Results: Audio modality"> </p>Video-ChatGPT vs PG-Video-LLaVA<img src="docs/images/logos/logo.png" height="20">
PG-Video-LLaVA is based on a stronger image-LMM baseline which gives it better conversational ability compared to its predecessor.
<p align="center"> <img src="docs/images/figures/comparison-prev_versions.png" alt="Video-ChatGPT vs PG-Video-LLaVA"> </p>Quantitative Evaluation π
We evaluate PG-Video-LLaVA using video-based generative and question-answering benchmarks. We also introduce new benchmarks specifically designed to measure prompt-based object grounding performance in videos.
Video Grounding π―
To quantitatively assess PG-Video-LLaVAβs spatial grounding capability, we conducted quantitative evaluations of PG-Video-LLaVAβs spatial grounding capabilities using two benchmarks that are derived from the test set of the VidSTG and HC-STVG datasets.
For detailed instructions on performing quantitative evaluation on video grounding, please refer this.
<p align="center"> <img src="docs/images/figures/quant_grounding.png" width="60%" alt="Video-Grounding Quantitative Results"> </p>Video-based Generative Performance Benchmarking π€
We apply the benchmarking framework from Video-ChatGPT which measures performance on several axes critical for video-based conversational agents, including correctness of information, detail orientation, contextual understanding, temporal understanding, and consistency. In order to facilitate a reliable and reproducible evaluation, we have updated our assessment pipeline by replacing GPT-3.5 with Vicuna-13b-v1.5.
<p align="center"> <img src="docs/images/figures/quant_our_benchmark.png" alt="Video-based Generative Performance Benchmarking"> </p>Zero-Shot Question Answering π¬
Zero-shot question-answering (QA) capabilities were evaluated quantitatively using several established open-ended QA datasets: MSRVTT-QA, MSVD-QA, TGIF-QA, and ActivityNet-QA.
<p align="center"> <img src="docs/images/figures/quant_zero_shot.png" alt="Zero-shot QA Quantitative Results"> </p>For detailed instructions on video-based generative performance benchmarking and zero-shot question answering benchmark, please refer this.
Acknowledgements π
- LLaMA: a great attempt towards open and efficient LLMs!
- Vicuna: has the amazing language capabilities!
- LLaVA: our architecture is inspired from LLaVA.
- Video-ChatGPT: the predecessor to PG-Video-LLaVA
Citation π
If you're using PG-Video-LLaVA in your research or applications, please cite using this BibTeX:
@article{munasinghe2023PGVideoLLaVA,
title={PG-Video-LLaVA: Pixel Grounding Large Video-Language Models},
author={Shehan Munasinghe and Rusiru Thushara and Muhammad Maaz and Hanoona Abdul Rasheed and Salman Khan and Mubarak Shah and Fahad Khan},
journal={ArXiv 2311.13435},
year={2023}
}
<img src="docs/images/logos/IVAL_logo.png" width="200" height="100"> <img src="docs/images/logos/Oryx_logo.png" width="100" height="100"> <img src="docs/images/logos/MBZUAI_logo.png" width="360" height="85">