How to use from the
Use from the
Transformers library
# Use a pipeline as a high-level helper
from transformers import pipeline

pipe = pipeline("text-generation", model="amao0o0/InfoDensity-DeepSeek-R1-Distill-Qwen-7B")
messages = [
    {"role": "user", "content": "Who are you?"},
]
pipe(messages)
# pip install -U transformers accelerate
# Load model directly
from transformers import AutoTokenizer, AutoModelForCausalLM

tokenizer = AutoTokenizer.from_pretrained("amao0o0/InfoDensity-DeepSeek-R1-Distill-Qwen-7B")
model = AutoModelForCausalLM.from_pretrained("amao0o0/InfoDensity-DeepSeek-R1-Distill-Qwen-7B", device_map="auto")
messages = [
    {"role": "user", "content": "Who are you?"},
]
inputs = tokenizer.apply_chat_template(
	messages,
	add_generation_prompt=True,
	tokenize=True,
	return_dict=True,
	return_tensors="pt",
).to(model.device)

outputs = model.generate(**inputs, max_new_tokens=256)
print(tokenizer.decode(outputs[0][inputs["input_ids"].shape[-1]:]))
Quick Links

InfoDensity-DeepSeek-R1-Distill-Qwen-7B

deepseek-ai/DeepSeek-R1-Distill-Qwen-7B trained with InfoDensity, a reinforcement-learning reward that favours information-dense reasoning traces. The reward combines an entropy-trajectory quality term with a group-relative length scaling term, and is applied only to traces that reach a correct answer, so the model learns to reach the right answer with markedly less deliberation. Method details are in the paper.

Results

Accuracy (pass@1) and mean generated length averaged over AMC23, AIME24, MATH500 and GPQA-Diamond, as reported in the paper. AES is the Accuracy–Efficiency Score (α=1, β=3, γ=5).

Model Accuracy Length AES
DeepSeek-R1-Distill-Qwen-7B 58.1 8.5k —
InfoDensity-DeepSeek-R1-Distill-Qwen-7B 72.2 4.5k +1.20

Evaluation protocol

Greedy decoding, pass@1, generation capped at 16384 tokens, with the prompt

<question>

Please reason step by step, and put your final answer within \boxed{}

The evaluation scripts that produce these numbers are in the GitHub repository.

Usage

from transformers import AutoModelForCausalLM, AutoTokenizer

model_id = "amao0o0/InfoDensity-DeepSeek-R1-Distill-Qwen-7B"
tokenizer = AutoTokenizer.from_pretrained(model_id)
model = AutoModelForCausalLM.from_pretrained(model_id, torch_dtype="bfloat16", device_map="auto")

question = "What is the smallest positive integer n such that n^2 + n + 41 is not prime?"
messages = [{"role": "user",
             "content": f"{question}\n\nPlease reason step by step, and put your final answer within \\boxed{{}}"}]
prompt = tokenizer.apply_chat_template(messages, add_generation_prompt=True, tokenize=False)

out = model.generate(**tokenizer(prompt, return_tensors="pt").to(model.device), max_new_tokens=16384)
print(tokenizer.decode(out[0], skip_special_tokens=True))

With vLLM:

from vllm import LLM, SamplingParams

llm = LLM(model="amao0o0/InfoDensity-DeepSeek-R1-Distill-Qwen-7B")
print(llm.generate([prompt], SamplingParams(temperature=0, max_tokens=16384))[0].outputs[0].text)

License

Released under mit, inherited from the base model deepseek-ai/DeepSeek-R1-Distill-Qwen-7B.

Citation

@inproceedings{wei2026infodensity,
  title     = {InfoDensity: Rewarding Information-Dense Traces for Efficient Reasoning},
  author    = {Wei, Chengwei and Kim, Jung-jae and Zhang, Longyin and Chen, Shengkai and Chen, Nancy F.},
  booktitle = {Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing},
  year      = {2026}
}
Downloads last month
16
Safetensors
Model size
8B params
Tensor type
BF16
·
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Model tree for amao0o0/InfoDensity-DeepSeek-R1-Distill-Qwen-7B

Finetuned
(315)
this model