Dual-objective Language Models: Training Efficiency Without Overfitting

View PDF HTML (experimental)

Abstract:This paper combines autoregressive and masked-diffusion training objectives without any architectural modifications, resulting in flexible language models that outperform single-objective models. Autoregressive modeling has been a popular approach, partly because of its training efficiency; however, that comes at the cost of sensitivity to overfitting. On the other hand, masked-diffusion models are less efficient to train while being more resilient to overfitting. In this work, we demonstrate that dual-objective training achieves the best of both worlds. To derive the optimal balance between both objectives, we train and evaluate 50 language models under varying levels of data repetition. We show that it is optimal to combine both objectives under all evaluated settings and that the optimal balance is similar whether targeting autoregressive or masked-diffusion downstream performance.

Subjects:

Computation and Language (cs.CL); Artificial Intelligence (cs.AI)

Cite as: arXiv:2512.14549 [cs.CL]

(or arXiv:2512.14549v3 [cs.CL] for this version)

https://doi.org/10.48550/arXiv.2512.14549

arXiv-issued DOI via DataCite

Journal reference: The Fourteenth International Conference on Learning Representations (ICLR 2026)

Submission history

From: David Samuel [view email] [v1] Tue, 16 Dec 2025 16:25:33 UTC (1,520 KB) [v2] Wed, 28 Jan 2026 17:06:52 UTC (1,960 KB) [v3] Fri, 27 Mar 2026 10:33:12 UTC (1,959 KB)