T$^\star$: Progressive Block Scaling for Masked Diffusion Language Models Through Trajectory Aware Reinforcement Learning

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Xia, Hanchen, Chen, Baoyou, Ge, Yutang, Zhao, Guojiang, Zhu, Siyu
Formato: Preprint
Publicado: 2026
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866911547783118848
author Xia, Hanchen
Chen, Baoyou
Ge, Yutang
Zhao, Guojiang
Zhu, Siyu
author_facet Xia, Hanchen
Chen, Baoyou
Ge, Yutang
Zhao, Guojiang
Zhu, Siyu
contents We present T$^\star$, a simple TraceRL-based training curriculum for progressive block-size scaling in masked diffusion language models (MDMs). Starting from an AR-initialized small-block MDM, T$^\star$ transitions smoothly to larger blocks, enabling higher-parallelism decoding with minimal performance degradation on math reasoning benchmarks. Moreover, further analysis suggests that T$^\star$ may actually converge to an alternative decoding schedule that achieves comparable performance.
format Preprint
id arxiv_https___arxiv_org_abs_2601_11214
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle T$^\star$: Progressive Block Scaling for Masked Diffusion Language Models Through Trajectory Aware Reinforcement Learning
Xia, Hanchen
Chen, Baoyou
Ge, Yutang
Zhao, Guojiang
Zhu, Siyu
Computation and Language
We present T$^\star$, a simple TraceRL-based training curriculum for progressive block-size scaling in masked diffusion language models (MDMs). Starting from an AR-initialized small-block MDM, T$^\star$ transitions smoothly to larger blocks, enabling higher-parallelism decoding with minimal performance degradation on math reasoning benchmarks. Moreover, further analysis suggests that T$^\star$ may actually converge to an alternative decoding schedule that achieves comparable performance.
title T$^\star$: Progressive Block Scaling for Masked Diffusion Language Models Through Trajectory Aware Reinforcement Learning
topic Computation and Language
url https://arxiv.org/abs/2601.11214