DR Pixel Trained

This is the trained DR Pixel model from DR Pixel: Page-Native Deep Research with Vision-Language Models: Qwen3.6-27B fine-tuned to answer questions about scientific papers by searching over and reading their rendered pages. It was trained with counterfactual credit reweighting on the 2,496 GPT-5.4 trajectories in the train/ directory of HughieHu/drpixel-bench, and evaluated on the benchmark in the same repository.

Files

model-00001-of-00012.safetensors …   the fine-tuned weights (bf16), the model evaluated in the paper
adapter/                              the same fine-tune as a LoRA adapter on Qwen/Qwen3.6-27B

The root holds the base model with the LoRA adapter merged in, so it serves like any Qwen3.6-27B checkpoint. adapter/ holds the adapter alone (rank 32, alpha 64, all linear layers of the language model), for loading on top of the base model with PEFT or for further training.

Evaluating on DRPixelBench

Serve the model with any OpenAI-compatible server (vLLM, SGLang, …) with a 131,072-token context, and run the harness from github.com/HughieHu/DRPixel (eval/) against it. The paper's runs used the sampling settings below; the harness passes them through when they are set.

MODEL_NAME=HughieHu/DR-Pixel-Trained
MODEL_BASE_URL=http://localhost:8000/v1
MODEL_API_KEY=EMPTY
MODEL_TEMPERATURE=0.6 MODEL_TOP_P=0.95 MODEL_TOP_K=20 MODEL_PRESENCE_PENALTY=1.0
SEARCH_URL=https://drpixel--drpixel-search-gate.us-east.modal.run    # hosted page search

python -m drbench.eval --config Page

Training

LoRA on all linear layers of the language model, one epoch at learning rate 1e-4 with an effective batch size of 16 and a maximum length of 131,072 tokens, with per-step loss weights from counterfactual credit and a constant weight of 1.5 on the final answer. train/README.md describes the method and the full recipe.

License

The weights are released under Apache 2.0, the license of the base model. The training data are trajectories generated by GPT-5.4 and remain subject to the terms of that model.

Citation

@misc{hu2026drpixel,
  title  = {DR Pixel: Page-Native Deep Research with Vision-Language Models},
  author = {Hu, Tiansheng and Wang, Yichuan and Talwar, James V. and Tomlin, Nicholas and Zhao, Chen and Min, Sewon},
  year   = {2026}
}
Downloads last month
23
Safetensors
Model size
27B params
Tensor type
BF16
·
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Model tree for HughieHu/DR-Pixel-Trained

Base model

Qwen/Qwen3.6-27B
Finetuned
(421)
this model

Dataset used to train HughieHu/DR-Pixel-Trained