inclusionAI 发布 Ling-3.0-tiny 投机解码草稿模型 Ling3-Tiny-UBTree-B16

Ling3-Tiny-UBTree-B16

Ling3-Tiny-UBTree-B16 is a UBTree speculative-decoding draft model for inclusionAI/Ling-3.0-tiny. It combines a five-layer DFlash backbone with a learned bigram selector to propose multiple candidate continuations for target-model verification.

SGLang integration — stay tuned! Our UBTree adaptation for Ling3 has not yet been released to the open-source SGLang community. Public installation instructions and serving recipes will be added here when it becomes available.

This checkpoint is used alongside the matching target model; it cannot serve as a standalone text-generation model.

How UBTree works

UBTree pipeline

UBTree pairs a unigram proposer with a bigram selector:

A DFlash backbone proposes candidate tokens for future positions in parallel.
The selector scores transitions between adjacent candidates in parallel.
Tree construction retains promising branches within a verification budget.
The target verifies the tree and commits the accepted continuation.

The paper also introduces tree-native training with renormalized KL supervision on high-temperature target traces, encouraging plausible alternative branches. See the paper for the method and experiments.

Model specifications

Item	Value
Target model	inclusionAI/Ling-3.0-tiny
Draft architecture	`Ling3UBTreeModel`
Draft backbone	5 full-attention Transformer layers
Parameters stored in this checkpoint	183,127,326 (183.1M)
Weight dtype	BF16
Hidden size	1,536
Attention heads / KV heads	16 / 16
Target auxiliary feature layers (zero-based)	4, 8, 12, 16, 22
Draft block size	16 (anchor-inclusive runtime convention)
Candidate top-k	64 per draft position
Default verification tree budget	64 non-root nodes + 1 anchor
Training	DFlash warm start, then joint backbone + UBTree selector training

The parameter count covers tensors distributed in this repository. The draft shares token embeddings and the language-model head with the target; these shared weights are not included here. B16 denotes the draft block size, not the verification tree's node budget.

Serving with SGLang

UBTree support for Ling3 in SGLang is coming soon. We will share installation instructions and serving examples once the integration is publicly available. Stay tuned!

Benchmark results

Results for this checkpoint will be added here. TBD means not yet reported.

Dataset	Acceptance length	Speedup
GSM8K	TBD	TBD
MATH-500	TBD	TBD
AIME 2025	TBD	TBD
HumanEval	TBD	TBD
MBPP	TBD	TBD
LiveCodeBench	TBD	TBD

Acceptance length is the mean number of tokens committed per verification step, including the target token. Speedup is measured against target-only autoregressive decoding with the same hardware, inputs, and decoding settings. The runtime version, tree budget, concurrency, and timing protocol will be reported alongside the results.

Citation

If you use UBTree in your work, please cite:

@article{liang2026ubtree,
  title   = {UBTree: Parallel Tree Drafting via Unigram and Bigram Models for Speculative Decoding},
  author  = {Liang, Chumeng and Wang, Linxuan and Peng, Xinyu and Liu, Huabin and Chen, Yuxin and Liu, Ge and Lin, Guang and Song, Qifan and Li, Jianguo},
  journal = {arXiv preprint arXiv:2609.39972},
  year    = {2026},
  url     = {https://arxiv.org/abs/2609.39972}
}

License

The model weights are released under the MIT License. The pipeline figure is from Liang et al., Figure 2, reproduced without modification under CC BY 4.0.