提出频域滤波模型融合方法,高效整合多任务模型能力
FREE-Merging: Fourier Transform for Efficient Model Merging
- 从参数频域视角解决任务干扰问题,设计频域滤波机制
- 在多个视觉、语言和多模态任务上实现性能与部署成本的平衡
- 轻量级专家模块动态补偿信息损失,适合资源受限场景
随着深度学习快速发展,各类开源模型日益丰富。但单个微调模型难以满足用户多样化需求,模型融合因此成为将多个模型能力整合为统一模型的高效方法。然而现有方法在性能与部署成本间存在显著权衡,主要源于任务干扰。我们首次揭示任务干扰存在于模型参数的频域中,而现有方法仅关注空间域处理,对频域干扰无效。为此,我们提出FR-Merging,通过在主干网络上施加轻量频域滤波,有效抑制有害频域干扰,计算开销极小。由于无代价方法不可避免带来性能损失,我们进一步引入轻量级任务特异性专家模块,动态补偿融合过程中的信息丢失。该框架名为FREE-Merging(FR-Merging with experts),在训练成本、推理延迟、存储需求与性能之间取得良好平衡。我们在计算机视觉、自然语言处理及多模态等多个任务上验证了FR-Merging与FREE-Merging的有效性,并证明其可灵活适配特定需求。
原文摘要 · Abstract (English)
With the rapid growth of deep learning, there is an increasing availability of open-source models for various tasks. However, single fine-tuned models often fall short of meeting the diverse needs of users. Model merging has thus emerged as an efficient method to integrate the capabilities of existing models into a unified model. Nevertheless, existing model merging methods face challenging trade-offs between performance and deployment costs, primarily due to task interference. For the first time, we reveal that task interference is evident in the frequency domain of model parameters, yet current efforts only focus on spatial domain solutions, which are largely ineffective in addressing frequency domain interference. To mitigate the impact of frequency domain interference, we propose FR-Merging, an innovative method that effectively filters harmful frequency domain interference on the backbone with minimal computational overhead. Since performance loss is inevitable with cost-free methods, we propose a lightweight task-specific expert module that dynamically compensates for information loss during merging. This proposed framework, FREE-Merging (FR-Merging with experts), strikes a balanced trade-off between training cost, inference latency, storage requirements, and performance. We demonstrate the effectiveness of both FR-Merging and FREE-Merging on multiple tasks across CV, NLP, and Multi-Modal domains and show that they can be flexibly adapted to specific needs.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。