arXiv Machine Learning

OnlineQAT: On-Policy Distillation for Ultra-Low-Bit Large Language Models

OnlineQAT introduces a two-stage framework for ultra-low-bit large language models, beginning with block-wise quantization-aware training to create a usable low-bit initialization and then applying on-policy distillation (OPD) using student-generated responses. The method employs a frozen full-precision teacher to provide a sampled reverse-KL training signal at each visited prefix, addressing the mismatch between offline recovery data and the deployed model’s self-generated prefixes. Experiments on Qwen3-1.7B show that OnlineQAT achieves the best average performance among compared quantized methods, outperforming ReasoningQAT by 2.90 and 0.44 points at W3A16 and W2A16 respectively, highlighting the benefit of student-visited states for recovery.

arXiv AI
Jun 4

Recover-LoRA for Aggressive Quantization: Reclaiming Accuracy in 2-Bit Language Models via Low-Rank Adaptation with Knowledge Distillation on Synthetic Data

arXiv:2606. 04238v1 Announce Type: cross Abstract: Aggressive weight quantization to 2-bit precision offers substantial throughput and memory gains for large language model (LLM) inference, but typically incurs severe accuracy degradation.

By Devleena Das, Rajeev Patwari, Elliott Delaye, Ashish Sirasao
arXiv Machine Learning
Oct 2

QATFactory: A Versatile, Deployment-Aligned Framework for Quantization-aware Training and Distillation of LLMs

arXiv:2609.39223v2 Announce Type: new Abstract: Large language model (LLM) inference is increasingly moving toward lower precision to realize the throughput of hardware accelerators, but aggressive p...

By Weili Xu, Jisen Li, Yuqing Jian, Chenxi Li, Zhizhou Sha, Yifan Yu, Qingyang Wu, Chenfeng Xu, Zhongzhu Zhou, Tianyi Zhang, Ben Athiwaratkun
arXiv AI
Aug 6

Instruction-Conditioned Exploration for Reinforcement Learning with Self-Distillation to an Unconditioned Policy

arXiv:2608. 02087v2 Announce Type: replace Abstract: Post-training Large Language Models (LLMs) with Reinforcement Learning (RL) has become an important tool for improving model capabilities, but the LLM action-space structure introduces challenges distinct from classical RL, with implications for inducing exploration.

By Jim Dilkes, Vahid Yazdanpanah, Sebastian Stein