arXiv:2502.10743cs.CL2025-02被引 5

用1比特量化压缩专业模型,高效合并同时保持性能

1bit-Merging: Dynamic Quantized Merging for Large Language Models

  • 按任务特点分层压缩:聊天用注意力层,数学代码用MLP层
  • 相比传统方法,存储量大幅降低,性能不降反升
  • 适合需要多任务模型合并且怕占空间的场景

大型语言模型的发展催生了各领域专用模型,亟需高效合并技术。传统方法将参数静态融合,损害特定任务表现;而任务路由虽保精度却增加存储负担。本文提出1bit-Merging框架,结合任务路由与1比特量化任务向量,在性能与存储间取得平衡。观察发现:聊天模型知识集中在注意力层,数学/代码模型则在MLP层,据此实施针对性压缩。在LLaMA2与Mistral系列模型上,覆盖对话、数学推理、代码生成任务的实验表明,1bit-Merging性能媲美或超越现有方法,显著降低存储开销。该框架为保留专用模型优势的同时实现高效合并提供了实用方案。

原文摘要 · Abstract (English)

Recent advances in large language models have led to specialized models excelling in specific domains, creating a need for efficient model merging techniques. While traditional merging approaches combine parameters into a single static model, they often compromise task-specific performance. However, task-specific routing methods maintain accuracy but introduce substantial storage overhead. We present \texttt{1bit}-Merging, a novel framework that integrates task-specific routing with 1-bit quantized task vectors to balance performance and storage efficiency. Our approach leverages the observation that different task-specific models store knowledge in distinct layers-chat models primarily in attention layers and math/code models in MLP layers, enabling targeted compression strategies. Through extensive experiments with LLaMA2 and Mistral model families across chat, mathematical reasoning, and code generation tasks, we demonstrate that 1bit-Merging achieves comparable or superior performance to existing methods while significantly reducing storage requirements. Our framework offers a practical solution for combining specialized models while maintaining their individual strengths and addressing the storage challenges of current approaches.

模型合并1比特量化大模型压缩

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