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. 🔥🔥🔥

🏆 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

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.

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.).

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.

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.

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.

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.

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}
  }