提出自适应混合位激活量化,降低大模型协同训练通信开销。
AMAQ: Adaptive Mixed-bit Activation Quantization for Collaborative Parameter Efficient Fine-tuning
- 根据通道与层重要性动态分配比特,实现自适应压缩。
- 相比固定精度,生成准确率提升2.5%,分类准确率提升1.3%。
- 适合资源受限设备上的高效协同微调,稳定且低通信成本。
大规模语言模型快速发展,带来协同式服务器-客户端分布式训练中通信效率与计算开销的挑战。为此,我们采用参数高效的分片学习,在低资源设备上实现效率与性能的平衡。为降低协同训练中的通信开销,提出自适应混合位激活量化(AMAQ),将激活和梯度从高精度(6至8位)逐步压缩至低精度(3至4位)。AMAQ通过比特正则化,依据特征与层的重要性动态分配比特预算。在相同比特预算下,相比固定精度方法,其在LLaMA3 8B与Qwen2.5 7B模型上分别实现约2.5%的生成准确率提升和约1.3%的分类准确率提升。同时显著增强训练稳定性,减少超低比特表示崩溃问题。实验表明,AMAQ可有效集成于多机协同训练场景,仅需轻微比特适应开销即可获得更优推理精度,是一种通信成本极低的实用解决方案。
原文摘要 · Abstract (English)
Large Language Models (LLMs) are scaling rapidly, creating significant challenges for collaborative server client distributed training, particularly in terms of communication efficiency and computational overheads. To address these challenges, we implement Parameter-efficient Split Learning, which effectively balances efficiency and performance for collaborative training on low-resource devices. To reduce communication overhead in collaborative training, we introduce Adaptive Mixed bit Activation Quantization (AMAQ), a strategy that progressively compresses activations and gradients from high precision (6 to 8 bits) to low precision (3 to 4 bits). AMAQ achieves this by effectively allocating bit budgets across channels based on feature wise and layer wise importance using bit regularization. Under the same bit budgets, AMAQ outperforms fixed-precision approaches, delivering about 2.5% higher generation accuracy and about 1.3% better classification accuracy for models like LLaMA3 8B and Qwen2.5 7B. In addition, it significantly enhances training stability and reducing ultra-low bit representation collapse during the training. Experiments demonstrate that AMAQ integrates effectively into practical multi-machine collaborative training setups, offering superior inference accuracy with only a modest communication overhead for bits adaptation during training. This trade off makes AMAQ a practical and effective solution for collaborative training with minimal communication cost.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。