Skip to content

iLearn-Lab/TMM25-RA-BLIP

Repository files navigation

RA-BLIP: Multimodal Adaptive Retrieval-Augmented Bootstrapping Language-Image Pre-training

Muhe Ding1, Yang Ma2, Pengda Qin3, Jianlong Wu1✉, Yuhong Li3, Liqiang Nie1

1School of Computer Science and Technology, Harbin Institute of Technology, Shenzhen, China
2School of Computer Science, University of Sydney, Sydney, Australia
3Security Department, Alibaba Group, Hangzhou, China

Corresponding author

💡 Overview

Multimodal large language models store rich visual-language knowledge in their parameters, but updating that knowledge can be expensive and hard to interpret. RA-BLIP introduces a retrieval-augmented framework for multimodal question answering that learns to use retrieved visual and textual evidence while reducing irrelevant noise.

RA-BLIP contains three main components:

  • Query-Instructed Visual Extraction: uses the question to guide visual feature extraction through learnable queries.
  • Multimodal Adaptive Fusion: projects visual and textual evidence into a unified semantic space for question-to-multimodal retrieval.
  • Adaptive Selection Knowledge Generation (ASKG): trains the generator to select useful retrieved knowledge and suppress noisy evidence.

RA-BLIP framework

This code release focuses on the WebQA training and evaluation pipeline.

📂 Data and Checkpoints

Prepare the WebQA annotations and image features before training or evaluation. Please download and prepare the WebQA data following the official WebQA repository:

After preparing the WebQA files, organize them as follows:

RA-BLIP/
├── webqa_dataset/
│   ├── WebQA_train_val.json
│   ├── webqa_test_retrieval_89.json
│   ├── imgs.tsv
│   └── imgs.lineidx
└── checkpoints/
    └── <ra_blip_checkpoint>.pth

Download the released RA-BLIP weights from Hugging Face:

⚙️ Installation

conda create -n rablip python=3.9
conda activate rablip

pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
pip install -r requirements.txt

💻 Training on WebQA

Update the WebQA data paths and output directory in train_webqa.sh if needed, then run:

bash train_webqa.sh

🔍 Evaluation on WebQA

Update the WebQA data paths and checkpoint name in eval_webqa.sh if needed, then run:

bash eval_webqa.sh

📊 Results

The paper reports RA-BLIP results on WebQA and compares with retrieval-augmented multimodal QA baselines.

WebQA results

🤗 Citation

If you find this repository useful, please cite:

@article{ding2025rablip,
  title={RA-BLIP: Multimodal Adaptive Retrieval-Augmented Bootstrapping Language-Image Pre-training},
  author={Ding, Muhe and Ma, Yang and Qin, Pengda and Wu, Jianlong and Li, Yuhong and Nie, Liqiang},
  journal={IEEE Transactions on Multimedia},
  year={2025}
}

🙏 Acknowledgement

This work builds on the following codebases. Thanks to their great work.

  • WebQA, for the WebQA benchmark and data format.
  • InstructBLIP, for the InstructBLIP/LAVIS implementation.

References

  • WebQA: Multihop and Multimodal QA. CVPR 2022.
  • InstructBLIP: Towards General-purpose Vision-Language Models with Instruction Tuning. NeurIPS 2023.

About

TMM 2025-RA-BLIP: Multimodal Adaptive Retrieval-Augmented Bootstrapping Language-Image Pre-training

Resources

Stars

6 stars

Watchers

0 watching

Forks

Releases

No releases published

Packages

 
 
 

Contributors

Languages