arXiv:2503.08154cs.CV2025-03

S2A框架让模型微调更省显存,适合部署在内存受限设备上。

S2A: A Unified Framework for Parameter and Memory Efficient Transfer Learning

  • 通过结构化模块设计减少可调参数和激活内存
  • 4比特量化非参数结构激活,平均降低4倍显存占用
  • 兼顾低参数量与高精度,适合边缘设备实用

参数高效微调(PETL)旨在降低预训练模型规模以适配多个下游任务。然而,随着模型持续增大,现有PETL方法的激活内存占用并未显著降低,限制了其在内存受限设备上的实际应用。为此,我们提出统一框架S2A,通过结构化激活设计(如偏置、提示、旁路模块)大幅减少可调参数与激活内存;同时对非参数结构(如非线性函数)的激活进行基于导数的4比特量化,在保持精度的同时显著降低内存开销。实验表明,S2A在不同骨干网络与数据集上均优于现有PETL方法,平均降低4倍GPU显存占用,且参数量更少、性能相当,非常适合硬件资源受限场景下的实用迁移学习。

原文摘要 · Abstract (English)

Parameter-efficient transfer learning (PETL) aims to reduce the scales of pretrained models for multiple downstream tasks. However, as the models keep scaling up, the memory footprint of existing PETL methods is not significantly reduced compared to the reduction of learnable parameters. This limitation hinders the practical deployment of PETL methods on memory-constrained devices. To this end, we proposed a new PETL framework, called Structure to Activation (S2A), to reduce the memory footprint of activation during fine-tuning. Specifically, our framework consists of: 1) Activation modules design(i.e., bias, prompt and side modules) in the parametric model structure, which results in a significant reduction of adjustable parameters and activation memory; 2) 4-bit quantization of activations based on their derivatives for non-parametric structures (e.g., nonlinear functions), which maintains accuracy while significantly reducing memory usage. Our S2A method consequently offers a lightweight solution in terms of both parameters and memory footprint. We evaluated S2A with different backbones and performed extensive experiments on various datasets to evaluate the effectiveness. The results show that our methods not only outperform existing PETL techniques, achieving a fourfold reduction in GPU memory footprint on average, but also shows competitive performance in accuracy with fewer tunable parameters. These demonstrate that our method is highly suitable for practical transfer learning on hardware-constrained devices.

参数高效显存优化边缘部署

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