[NeurIPS 2025] Domain-RAG: Retrieval-Guided Compositional Image Generation for Cross-Domain Few-Shot Object Detection

πŸ”₯ Paper (NeurIPS 2025) | 🌐 Project Page | πŸ“¦ Dataset Scripts | 🧠 Model Zoo | πŸš€ Quick Start | πŸŽ₯ Video | πŸ“Š Evaluation | πŸ“ž Contact


Domain-RAG is a novel retrieval-augmented generative framework designed for Cross-Domain Few-Shot Object Detection (CD-FSOD). We leverage large-scale vision-language models (GroundingDINO), a curated COCO-style retrieval corpus, and Flux-based background generation to synthesize diverse, domain-aware training data that enhances FSOD generalization under domain shift.


✨ Highlights

  • πŸ” Retrieval-Augmented Generation: retrieve semantically similar source images for novel-class prompts.
  • 🎨 Flux-Redux Integration: compose diverse backgrounds with target foregrounds for domain-aligned generation.
  • πŸ“¦ Support for Multiple Target Domains: ArTAXOr, Clipart1k, DIOR, DeepFish, UODD, NEU-DET, and more.
  • πŸ§ͺ Strong Benchmarks: surpasses GroundingDINO baseline in 1-shot and 5-shot CD-FSOD across 6 datasets.

πŸ”§ Installation

git clone https://github.com/LiYu0524/Domain-RAG.git
cd Domain-RAG
conda create -n domainrag python=3.10
conda activate domainrag
pip install -r requirements.txt

Pretrained Models

we will relase the fine-tuned grounding-dino model soon

Dataset Preparation

You can prepare CDFSOD with CDVITO

Quick start

You can refer to ./domainrag.sh

Video

Walkthrough video(Chinese version): Watch here

Contact

For questions and collaboration, please contact:

  • Yu Li : <liyu24@m.fudan.edu.cn>

Citation

If you find Domain-RAG useful in your research, please cite:

@inproceedings{li2025domainrag,
 author={Li, Yu and Qiu, Xingyu and Fu, Yuqian and Chen, Jie and Qian, Tianwen and Zheng, Xu and Paudel, Danda Pani and Fu, Yanwei and Huang, Xuanjing and Van Gool, Luc and others},
 booktitle = {Advances in Neural Information Processing Systems},
 title = {Domain-RAG: Retrieval-Guided Compositional Image Generation for Cross-Domain Few-Shot Object Detection},
 year = {2025}
}

If you find CD-Vito useful in your research, please cite:

@inproceedings{fu2024cross,
  title={Cross-domain few-shot object detection via enhanced open-set object detector},
  author={Fu, Yuqian and Wang, Yu and Pan, Yixuan and Huai, Lian and Qiu, Xingyu and Shangguan, Zeyu and Liu, Tong and Fu, Yanwei and Van Gool, Luc and Jiang, Xingqun},
  booktitle={European Conference on Computer Vision},
  pages={247--264},
  year={2024},
  organization={Springer}
}
Downloads last month

-

Downloads are not tracked for this model. How to track
Inference Providers NEW
This model isn't deployed by any Inference Provider. πŸ™‹ Ask for provider support

Model tree for YuLillll/Domain-RAG

Finetuned
(30)
this model