arXiv:2602.10431cs.LG2026-02

让大模型在量化后仍能动态跳过层数,保持高精度

QTALE: Quantization-Robust Token-Adaptive Layer Execution for LLMs

  • 训练时保证多种执行路径被探索,避免冗余损失
  • 推理时可灵活调整跳层比例,恢复必要冗余
  • 兼容量化与跳层,性能接近纯量化模型

大语言模型(LLMs)需要大量计算和内存资源,部署效率面临挑战。现有两种方法:通过选择性跳过层数减少浮点运算量(FLOPs),以及通过降低权重精度减少内存占用。但直接结合会因模型冗余减少导致精度下降。本文提出QTALE框架,实现跳层执行与量化无缝融合且保持精度。传统跳层方法削弱了训练路径多样性及推理时的活跃参数数量。QTALE引入两项关键机制:(1) 训练策略确保微调时充分探索多样化执行路径;(2) 后训练机制允许推理时灵活调节执行比例,按需恢复冗余。实验表明,在CommonsenseQA基准上,QTALE与纯量化模型的准确率差距低于0.5%。该方法同时实现降低FLOPs与内存占用,为高效部署提供有效方案。

原文摘要 · Abstract (English)

Large language models (LLMs) demand substantial computational and memory resources, posing challenges for efficient deployment. Two complementary approaches have emerged to address these issues: token-adaptive layer execution, which reduces floating-point operations (FLOPs) by selectively bypassing layers, and quantization, which lowers memory footprint by reducing weight precision. However, naively integrating these techniques leads to additional accuracy degradation due to reduced redundancy in token-adaptive models. We propose QTALE (Quantization-Robust Token-Adaptive Layer Execution for LLMs), a novel framework that enables seamless integration of token-adaptive execution with quantization while preserving accuracy. Conventional token-adaptive methods reduce redundancy in two ways: (1) by limiting the diversity of training paths explored during fine-tuning, and (2) by lowering the number of parameters actively involved in inference. To overcome these limitations, QTALE introduces two key components: (1) a training strategy that ensures diverse execution paths are actively explored during fine-tuning, and (2) a post-training mechanism that allows flexible adjustment of the execution ratio at inference to reintroduce redundancy when needed. Experimental results show that QTALE enables seamless integration of token-adaptive layer execution with quantization, while keeping the accuracy gap to quantization-only models below 0.5% on CommonsenseQA benchmarks. By combining tokenadaptive execution for FLOPs reduction and quantization for memory savings, QTALE provides an effective solution for efficient LLM deployment.

大模型压缩量化跳层执行

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。