压缩与微调联合进行,提升模型适配效率与精度。
Compress Then Adapt? No, Do It Together via Task-aware Union of Subspaces

- 联合优化压缩与微调,通过任务感知子空间统一实现
- 在80%参数保留下,视觉任务平均准确率达89.2%
- 适合需要高效部署和快速适配的场景
将大模型适配到多样化任务已成为常态,但主流的参数高效微调(PEFT)与低秩压缩通常分步进行。这种分离方式先压缩再微调,可能导致压缩子空间与下游目标错位,浪费全局参数预算。为此,我们提出JACTUS(任务感知子空间联合压缩与适配),一个统一框架。从少量校准数据中估计输入与预激活梯度协方差,将其正交并入预训练权重子空间,在此联合空间内进行投影低秩近似,按每参数边际增益全局分配秩,并仅训练一个紧凑核心矩阵。该方法显式缓解压缩方向与适配需求之间的错位问题,生成可部署的低秩模型,无需保留完整冻结权重,支持快速稳健微调。在视觉任务上,ViT-Base在8个数据集上平均准确率达89.2%(保留80%参数),优于100%参数量的强基线(如DoRA:87.9%)。在语言任务上,Llama2-7B在常识问答任务平均得分为80.9%(同参数保留率),超过100%参数量基线(如DoRA:79.7%)及同类压缩-微调流程。
原文摘要 · Abstract (English)
Adapting large pretrained models to diverse tasks is now routine, yet the two dominant strategies of parameter-efficient fine-tuning (PEFT) and low-rank compression are typically composed in sequence. This decoupled practice first compresses and then fine-tunes adapters, potentially misaligning the compressed subspace with downstream objectives and squandering a global parameter budget. To overcome this limitation, we introduce JACTUS (Joint Adaptation and Compression with a Task-aware Union of Subspaces), a single framework that unifies compression and adaptation. From a small calibration set, JACTUS estimates input and pre-activation gradient covariances, forms their orthogonal union with the pretrained weight subspace, performs a projected low-rank approximation inside this union, allocates rank globally by marginal gain per parameter, and trains only a compact core matrix. This explicitly mitigates the potential misalignment between the compressed subspace and downstream objectives by coupling the directions preserved for compression with those required for adaptation, yielding a deployable low-rank model that avoids retaining full frozen weights while enabling fast and robust tuning. On vision, JACTUS attains an average 89.2% accuracy on ViT-Base across eight datasets at 80% retained parameters, surpassing strong 100% PEFT baselines (e.g., DoRA 87.9%). On language, JACTUS achieves an 80.9% average on Llama2-7B commonsense QA at the same 80% retained-parameter budget, outperforming 100% PEFT (e.g., DoRA 79.7%) and exceeding prior compress-then-finetune pipelines under the same ratained-parameter budget. We will release code.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。