Awesome

BLIVA: A Simple Multimodal LLM for Better Handling of Text-rich Visual Questions

Wenbo Hu*, Yifan Xu*, Yi Li, Weiyue Li, Zeyuan Chen, and Zhuowen Tu. *Equal Contribution

UC San Diego, Coinbase Global, Inc.

<a href='https://ojs.aaai.org/index.php/AAAI/article/view/27999'><img src='https://img.shields.io/badge/Video/Poster- -yellow'></a> <a href='https://gordonhu608.github.io/bliva/'><img src='https://img.shields.io/badge/Project-Page-Green'></a> <a href='https://arxiv.org/abs/2308.09936'><img src='https://img.shields.io/badge/Paper-Arxiv-red'></a> <a href='https://huggingface.co/spaces/mlpc-lab/BLIVA'><img src='https://img.shields.io/badge/%F0%9F%A4%97%20Hugging%20Face-Demo-blue'></a> <a href='https://huggingface.co/mlpc-lab/BLIVA_Vicuna'><img src='https://img.shields.io/badge/%F0%9F%A4%97%20Hugging%20Face-BLIVA_Vicuna_Model-blue'></a> <a href='https://huggingface.co/mlpc-lab/BLIVA_FlanT5'><img src='https://img.shields.io/badge/%F0%9F%A4%97%20Hugging%20Face-BLIVA_FlanT5_Model-blue'></a> <a href='https://huggingface.co/datasets/mlpc-lab/YTTB-VQA'><img src='https://img.shields.io/badge/%F0%9F%A4%97%20Hugging%20Face-Youtube Thumbnails Datasets-blue'></a> <a href='docs/BLIVA Demo.pdf'><img src='https://img.shields.io/badge/Demo%20Slides-PDF-red'></a>

Release

• [12/08] 🔥 BLIVA is accepted by AAAI 2024.
• [9/13] 🔥 We released the training code of BLIVA.
• [9/06] 🔥 We released a Demo Slides for researchers and related workers to learn about BLIVA's abilities and use cases efficiently.
• [8/28] 🔥 Our model achieved No.3 in Perception tasks and No.2 in Cognition tasks on the MME benchmark, improving 6 positions than our baseline in Perception tasks and 5 positions in Cognition tasks. BLIVA achieved No.1 in Color, Poster, and Commonsense Reasoning subtasks.
• [8/21] 🔥 We released BLIVA: A Simple Multimodal LLM for Better Handling of Text-Rich Visual Questions. Checkout the paper.
• [8/21] We released our demo at here which is publicly available for everyone to play with. If the huggingface space instance is stopped, feel free to relaunch it. The wait time is around 30s.
• [8/21] We released our Model weight for BLIVA Vicuna Version at here and FLAN T5 Version at here which is available for commercial use.
• [8/21] Our Youtube Visual Question Answering Dataset (YTTB-VQA) is available at here.

Performance on Text-Rich VQA Benchmarks

Performance on General (not particularly text-rich) VQA Benchmarks

Installation

1. Creating conda environment

conda create -n bliva python=3.9
conda activate bliva

2. build from source

git clone https://github.com/mlpc-ucsd/BLIVA
cd BLIVA
pip install -e .

Prepare Weight

1. BLIVA Vicuna 7B

   Our Vicuna version model is released at here. Download our model weight and specify the path in the model config here at line 8.

   The LLM we used is the v0.1 version from Vicuna-7B. To prepare Vicuna's weight, please refer to our instruction here. Then, set the path to the vicuna weight in the model config file here at Line 21.

2. BLIVA FlanT5 XXL (Available for Commercial Use)

   The FlanT5 version model is released at here. Download our model weight and specify the path in the model config here at line 8.

   The LLM weight for Flant5 will automatically begin to download from huggingface when running our inference code.

Inference

To answer one question from the image, run the following evaluation code. For example,

python evaluate.py --answer_qs \
        --model_name bliva_vicuna \
        --img_path images/example.jpg \
        --question "what is this image about?"

We also support answer multiple choice question, which is the same as we used for evaluation tasks in paper. To provide a list of chioce, it should be a string split by comma. For example,

python evaluate.py --answer_mc \
        --model_name bliva_vicuna \
        --img_path images/mi6.png \
        --question "Which genre does this image belong to?" \
        --candidates "play, tv show, movie"

Demo

Our Demo is publicly available at here. To run our demo locally on your machine. Run:

python demo.py

Train

After downloading the training datasets and specify their path in dataset configs, we are ready for training. We utilized 8x A6000 Ada in our experiments. Please adjust hyperparamters according to your GPU resources. It may take transformers around 2 minutes to load the model, give some time for the model to start training. Here we give an example of traning BLIVA Vicuna version, the Flant5 version follows the same format.

1. Pretraining of BLIVA's visual assistant branch

torchrun --nnodes=1 --nproc_per_node=8 \
    train.py \
    --cfg-path train_configs/pretrain_bliva_vicuna.yaml

2. Instruction Finetuning BLIVA

torchrun --nnodes=1 --nproc_per_node=8 \
    train.py \
    --cfg-path train_configs/finetune_bliva_vicuna.yaml

Or, we also support training Vicuna7b together with BLIVA using LoRA during the second step, by default we don't use this version.

torchrun --nnodes=1 --nproc_per_node=8 \
    train.py \
    --cfg-path train_configs/finetune_bliva_and_vicuna.yaml

Citation

If you find BLIVA useful for your research and applications, please cite using this BibTeX:

@misc{hu2023bliva,
      title={BLIVA: A Simple Multimodal LLM for Better Handling of Text-Rich Visual Questions}, 
      author={Wenbo Hu and Yifan Xu and Yi Li and Weiyue Li and Zeyuan Chen and Zhuowen Tu},
      publisher={arXiv:2308.09936},
      year={2023},
}

Acknowledgement

• BLIP2 The model architecture of BLIVA follows BLIP-2. Don't forget to check this great open-source work if you don't know it before.
• Lavis The codebase we built upon.
• Vicuna Vicuna-13B demonstrates fantastic language ability and it's open source.
• MultimodalOCR A great Mulimodality OCR evaluation pipeline that we followed.
• MME A comprehensive evaluation benchmark for multimodal large language models that we tested our model on.

License

This repository's code is under BSD 3-Clause License. Many codes are based on Lavis with BSD 3-Clause License here.

For our model parameters of BLIVA Vicuna Version, it's should be used under LLaMA's model license. For the model weight of BLIVA FlanT5, it's under Apache 2.0 License. For our YTTB-VQA data, it's under CC BY NC 4.0