Skip to content

Latest commit

 

History

History
161 lines (96 loc) · 6.85 KB

README.md

File metadata and controls

161 lines (96 loc) · 6.85 KB

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

Website paper


📢 Latest Updates

  • 📦 27-Dec-2023: Code, models released! 🚀

Overview

PG-Video-LLaVA is the first video-based Large Multimodal Model (LMM) with pixel-level grounding capabilities. 🔥🔥🔥

PG-Video-LLaVA Architectural Overview


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


PG-Video-LLaVA : Architecture

PG-Video-LLaVA Architectural Overview


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.

Video-Grounding Qualitative Results


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

Qualitative Results: Audio modality


Video-ChatGPT vs PG-Video-LLaVA

PG-Video-LLaVA is based on a stronger image-LMM baseline which gives it better conversational ability compared to its predecessor.

Video-ChatGPT vs PG-Video-LLaVA


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-Grounding Quantitative Results


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.

Video-based Generative Performance Benchmarking


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.

Zero-shot QA Quantitative Results

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