d3LLM_Dream / README.md
d3LLM-model's picture
Update README.md
7d877e8 verified
|
Raw
History Blame Contribute Delete
2.13 kB
metadata
datasets:
  - d3LLM/trajectory_data_dream_32
pipeline_tag: text-generation
library_name: transformers
license: apache-2.0
base_model: Dream-org/Dream-v0-Instruct-7B
tags:
  - diffusion
  - text-generation
  - fast-inference
  - d3llm

d3LLM: Ultra-Fast Diffusion LLM using Pseudo-Trajectory Distillation πŸš€

This repository contains the d3LLM-Dream model, an ultra-fast diffusion language model introduced in the paper d3LLM: Ultra-Fast Diffusion LLM using Pseudo-Trajectory Distillation.

Model Description

d3LLM-Dream is an ultra-fast diffusion language model that achieves high generation speed while maintaining competitive performance. It strikes a balance between accuracy and parallelism by using pseudo-trajectory distillation during training and entropy-based multi-block decoding during inference.

Key Features

  • πŸš€ High throughput: 4.5Γ— faster than autoregressive models (Qwen-2.5-7B) on H100 GPU, 2.5Γ— faster on A100 GPU. Achieves 235.34 tokens/s on H100 on GSM8K-CoT.
  • πŸ“Š High AUP: Optimized for Accuracy Under Parallelism across benchmarks.
  • πŸ”§ Specialized: Optimized for coding and math reasoning tasks.

Usage

For more chat examples and evaluation scripts, visit the official repository.

Citation

@inproceedings{ICML'26:d3llm,
    title = {d3LLM: Ultra-Fast Diffusion LLM using Pseudo-Trajectory Distillation},
    author = {Yu-Yang Qian and Junda Su and Lanxiang Hu and Peiyuan Zhang and Zhijie Deng and Peng Zhao and Hao Zhang},
    booktitle = {Proceedings of the 43rd International Conference on Machine Learning (ICML)},
    pages = {to appear},
    year = {2026}
}