Skip to content

Commit 8d51dbc

Browse files
authored
update checkpoint link
1 parent c894b51 commit 8d51dbc

1 file changed

Lines changed: 20 additions & 2 deletions

File tree

README.md

Lines changed: 20 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,12 @@
22

33
*A model that learns to actively search for relevant temporal clips through end-to-end reinforcement learning.*
44

5+
[📄 [Paper](https://arxiv.org/abs/2511.05489)] [🤗 [Model](https://huggingface.co/Time-Search/TimeSearch-R)]
6+
7+
## 📰 News
8+
9+
🔥 **[2025/11/13]** Our [Model Checkpoint](https://huggingface.co/Time-Search/TimeSearch-R) is uploaded!
10+
511
## 👁️ Overview
612

713
TimeSearch-R reformulates temporal search as interleaved text–video thinking, seamlessly integrating searching video clips into the reasoning process through reinforcement learning (RL).
@@ -45,12 +51,12 @@ python3 -m clip_server
4551

4652
### 📦️ Dataset & Model
4753

48-
We provide the preprocessed JSON files for [Haystack-LVBench](https://huggingface.co/datasets/). The corresponding `.mp4` video files can be downloaded from [the original LongVideoBench dataset](https://huggingface.co/datasets/longvideobench/LongVideoBench/tree/main).
54+
We provide the preprocessed JSON files for [Haystack-LVBench](https://huggingface.co/datasets/MLL-Lab/LongVideoHaystack). The corresponding `.mp4` video files can be downloaded from the original [LongVideoBench](https://huggingface.co/datasets/longvideobench/LongVideoBench) dataset.
4955

5056
Download the pre-trained TimeSearch-R model.
5157

5258
```bash
53-
huggingface-cli download --resume-download ByteDance/TimeSearch-R --local-dir /path/to/your/local/filedir
59+
huggingface-cli download --resume-download Time-Search/TimeSearch-R --local-dir /path/to/your/local/filedir
5460
```
5561

5662
**(Recommended) Prepare the frame cache and feature cache.**
@@ -133,6 +139,18 @@ We provide the training script for TimeSearch-R with GRPO-CSV in `scripts/train.
133139
bash scripts/train.sh
134140
```
135141

142+
## 🔖 Citation
143+
144+
If you find TimeSearch-R useful for your research and applications, please cite using this BibTeX:
145+
```bibtex
146+
@article{timesearch-r,
147+
title={TimeSearch-R: Adaptive Temporal Search for Long-Form Video Understanding via Self-Verification Reinforcement Learning},
148+
author={Pan, Junwen and Zhang, Qizhe and Zhang, Rui and Lu, Ming and Wan, Xin and Zhang, Yuan and Liu, Chang and She, Qi},
149+
journal={arXiv preprint arXiv:2511.05489},
150+
year={2025}
151+
}
152+
```
153+
136154
## 🎟️ License
137155

138156
This project is released under the [Apache 2.0 license](LICENSE).

0 commit comments

Comments
 (0)