activitynet-stage1 / README.md
happy8825's picture
Add pipeline tag and links to paper, code, and project page (#1)
017a5f5
|
Raw
History Blame Contribute Delete
1.44 kB
metadata
base_model: Qwen/Qwen3-VL-4B-Instruct
library_name: transformers
license: apache-2.0
pipeline_tag: video-text-to-text
tags:
  - video-retrieval
  - temporal-grounding
  - videosearch-r1

VideoSearch-R1 ActivityNet Stage 1

This is the Stage 1 VideoSearch-R1 checkpoint trained for ActivityNet, presented in the paper VideoSearch-R1: Iterative Video Retrieval and Reasoning via Soft Query Refinement.

Stage 1 is trained from the Qwen3-VL base model on constructed video retrieval and temporal grounding supervision. It is intended as the initialization for Stage 2 training or as a released checkpoint for ablations.

Usage

Use with the VideoSearch-R1 codebase:

bash scripts/data_construct/download_preextracted_data.bash activitynet
EVAL_GPUS=0 bash scripts/inference/inference.bash activitynet --checkpoint VideoSearchR1/activitynet-stage1

Citation

@inproceedings{lee2026videosearchr1,
  title     = {VideoSearch-R1: Iterative Video Retrieval and Reasoning via Soft Query Refinement},
  author    = {Lee, Seohyun and Choi, Seoung and Ko, Dohwan and Kim, Jongha and Kim, Hyunwoo J.},
  booktitle = {European Conference on Computer Vision (ECCV)},
  year      = {2026}
}