How to use from
vLLM
Install from pip and serve model
# Install vLLM from pip:
pip install vllm
# Start the vLLM server:
vllm serve "Jingcz/RobustRDP"
# Call the server using curl (OpenAI-compatible API):
curl -X POST "http://localhost:8000/v1/chat/completions" \
	-H "Content-Type: application/json" \
	--data '{
		"model": "Jingcz/RobustRDP",
		"messages": [
			{
				"role": "user",
				"content": [
					{
						"type": "text",
						"text": "Describe this image in one sentence."
					},
					{
						"type": "image_url",
						"image_url": {
							"url": "https://cdn.britannica.com/61/93061-050-99147DCE/Statue-of-Liberty-Island-New-York-Bay.jpg"
						}
					}
				]
			}
		]
	}'
Use Docker
docker model run hf.co/Jingcz/RobustRDP
Quick Links

RobustRDP

Fine-tuned model for the paper: RobustRDP: Advancing Reaction Diagram Parsing via Synthetic-to-Real Data Scaling and Robustness-Oriented Training.

Description

This model is a fine-tuned checkpoint based on Qwen2.5-VL-3B-Instruct, trained with a three-stage pipeline:

  1. Pretrain stage: Synthetic data pretraining on large-scale synthetic reaction diagrams.
  2. SFT stage: Supervised fine-tuning with three specialized tasks:
    • Vanilla Reaction Parsing (VRP): Standard reaction diagram parsing
    • Region-Guided Reaction Parsing (RGRP): Region-aware parsing with spatial guidance
    • Prefix-Perturbed Reaction Parsing (PPRP): Robustness-oriented parsing with prefix perturbations
  3. DPO stage: Direct Preference Optimization to further align model outputs with ground-truth annotations.

Training Details

Config Pretrain SFT DPO
Base Model Qwen2.5-VL-3B-Instruct Qwen2.5-VL-3B-Instruct Qwen2.5-VL-3B-Instruct
Learning Rate 1.0×10⁻⁶ 1.0×10⁻⁵ 3.0×10⁻⁷
Batch Size 16 4 64
Epochs 1 1 1
Scheduler Cosine (warmup 0.03) Cosine (warmup 0.03) Cosine (warmup 0.03)
Optimizer AdamW AdamW AdamW
Trainable Params LLM only Full (vision + LLM) LLM only

Usage

from transformers import AutoModelForCausalLM, AutoTokenizer

model_name = "Jingcz/RobustRDP"
tokenizer = AutoTokenizer.from_pretrained(model_name)
model = AutoModelForCausalLM.from_pretrained(model_name)

Related Resources

Downloads last month
26
Safetensors
Model size
4B params
Tensor type
BF16
·
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Model tree for Jingcz/RobustRDP

Finetuned
(813)
this model