Checkpoint
Table | |
|---|
| Method | ReMaRk (the method proposed in the paper) 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-9B-Base |
| Initialised from | ReMaRk-Qwen3.5-9B-RedditV2-SFT |
| Seed | 42 |
| Selected step | 2000 (RL steps of the checkpoint with the best validation R@20). The trainer_state.json in this repository records only the last resumed segment of training, so its global_step is smaller than the selected step above. |
| Paper reference | Table 8 (seed 42 pair, 9B) |
| Note | Seed-42 partner of ReMaRk-Qwen3.5-9B-RedditV2-RankGRPO-s42 in Table 8; the main-table seed-3407 run is ReMaRk-Qwen3.5-9B-RedditV2-ReMaRk-s3407. |
| Weights | bf16 safetensors, 1 file, 18.82 GB (9.41 B stored parameters, including the unused vision encoder). |
Usage
# pip install "transformers>=5.3.0" accelerate torchimport torchfrom transformers import AutoModelForCausalLM, AutoTokenizer repo = "XINLI1997/ReMaRk-Qwen3.5-9B-RedditV2-ReMaRk-s42"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
Reinforcement learning (ReMaRk) from the SFT initialisation on the training requests of Reddit-v2 (movies); the released checkpoint is the one with the best validation R@20.
Full configurations and scripts are in the code repository.
Evaluation
This checkpoint is reported in the paper, Table 8 (seed 42 pair, 9B); 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-9B-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}}