arXiv:2503.20282cs.CVcs.AI2025-03CVPR被引 8

通过精简冗余令牌,让参数高效微调更快更省内存。

Faster Parameter-Efficient Tuning with Token Redundancy Reduction

  • 用适配器学习令牌相似性,动态合并冗余令牌
  • 推理速度超越基线模型,内存效率更高
  • 适合需要快速部署的轻量化场景

参数高效微调(PET)通过仅学习少量参数,将预训练大模型迁移至下游任务,显著降低存储和传输成本。然而,多数PET方法继承了大模型的推理延迟,并因额外模块(如适配器)引入计算开销,限制其在计算密集型应用中的实用性。本文提出更快的参数高效微调(FPET),引入一个即插即用的令牌冗余减少模块。该模块利用适配器精炼自注意力层的令牌表示,通过可微分的令牌合并策略,基于直通估计器实现最优令牌剪枝。实验表明,FPET在保持与顶尖PET方法相当性能的同时,实现了比预训练基线更快的推理速度和更高的内存效率。

原文摘要 · Abstract (English)

Parameter-efficient tuning (PET) aims to transfer pre-trained foundation models to downstream tasks by learning a small number of parameters. Compared to traditional fine-tuning, which updates the entire model, PET significantly reduces storage and transfer costs for each task regardless of exponentially increasing pre-trained model capacity. However, most PET methods inherit the inference latency of their large backbone models and often introduce additional computational overhead due to additional modules (e.g. adapters), limiting their practicality for compute-intensive applications. In this paper, we propose Faster Parameter-Efficient Tuning (FPET), a novel approach that enhances inference speed and training efficiency while maintaining high storage efficiency. Specifically, we introduce a plug-and-play token redundancy reduction module delicately designed for PET. This module refines tokens from the self-attention layer using an adapter to learn the accurate similarity between tokens and cuts off the tokens through a fully-differentiable token merging strategy, which uses a straight-through estimator for optimal token reduction. Experimental results prove that our FPET achieves faster inference and higher memory efficiency than the pre-trained backbone while keeping competitive performance on par with state-of-the-art PET methods.

参数高效推理加速令牌压缩

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