Checkpoint
Table | |
|---|
| Method | SFT initialisation on Reddit-v2 (movies) |
| Data | The processed Reddit-v2 release of Rank-GRPO (Zhu et al.), yaochenzhu/reddit-v2; SFT targets come from that release |
| Base model | Qwen/Qwen3.5-2B-Base |
| Initialised from | Qwen/Qwen3.5-2B-Base |
| Selected step | 1500 (SFT steps) |
| Paper reference | Table 2 (SFT row, 2B) |
| Weights | bf16 safetensors, 1 file, 5.44 GB (2.72 B stored parameters, including the unused vision encoder). The file also stores lm_head.weight, a bit-identical copy of the tied input embeddings. |
Usage
# pip install "transformers>=5.3.0" accelerate torchimport torchfrom transformers import AutoModelForCausalLM, AutoTokenizer repo = "XINLI1997/ReMaRk-Qwen3.5-2B-RedditV2-SFT"tokenizer = AutoTokenizer.from_pretrained(repo)model = AutoModelForCausalLM.from_pretrained(repo, dtype=torch.bfloat16, device_map="auto") messages = [{"role": "user", "content": "I loved Inception, Arrival and Interstellar. What other movies would you recommend? Give me 20 titles."}]inputs = tokenizer.apply_chat_template(messages, add_generation_prompt=True, enable_thinking=False, return_tensors="pt", return_dict=True).to(model.device)output = model.generate(**inputs, max_new_tokens=512, do_sample=False)print(tokenizer.decode(output[0, inputs["input_ids"].shape[1]:], skip_special_tokens=True))
The model answers with a recommendation list, one title (year) (or name (year) for music) per line. The exact instruction header
wrapped around each Reddit request in training and evaluation, the decoding settings and the evaluation scripts are in the
code repository; the request above is only an illustration.
Notes:
- Qwen3.5 checkpoints need
transformers>=5.3.0. They keep the Qwen3.5 multimodal layout (Qwen3_5ForConditionalGeneration); AutoModelForCausalLM loads the language model only (vLLM loads the folder directly; preprocessor_config.json and video_preprocessor_config.json are included for that). When training with TRL, pass the text tokenizer as processing_class.
- The snippet uses non-thinking mode (
enable_thinking=False).
trainer_state.json (training curves of the run) is included; no optimizer state.
Training
Supervised fine-tuning of the base model on the SFT targets of Reddit-v2 (movies).
Full configurations and scripts are in the code repository.
Evaluation
This checkpoint is reported in the paper, Table 2 (SFT row, 2B); see the paper for its scores.
Limitations
- Trained for conversational recommendation in one domain (movies); it is not a general assistant.
- Recommendations reflect what the Reddit community recommended and the catalog used for grounding; popular items are favoured and the lists can carry the community's biases.
- English only.
License
Apache-2.0, inherited from the base model Qwen/Qwen3.5-2B-Base (see LICENSE). Third-party data keep their own terms (Reddit; UCSD Book Graph: academic use; MusicBrainz: CC0); see the dataset card of yaochenzhu/reddit-v2.
Citation
@misc{li2026remark, title = {Many Right Answers, One Training Example: Removing Responder Noise from GRPO for LLM Recommenders}, author = {Li, Xin and Jiang, Hao and Guo, Jinghao and Lin, Weisi and Yuen, Chau}, year = {2026}, note = {Preprint; arXiv link coming soon}}