论文

αTransfer:跨模型规模迁移合并系数

$α$Transfer: Coefficient Transfer for Efficient Model Merging

模型优化模型合并

摘要

模型合并提供了一种很有前景的解决方案,可以通过参数算法将多个经过微调的检查点合并为单个模型。然而,寻找最佳合并系数需要进行广泛的搜索,由于记忆的高要求和搜索空间中的组合增长,随着模型大小和数量的扩展,搜索的成本变得异常昂贵。我们表明,在同一模型系列中,模型在不同模型大小的合并系数上表现出高度一致的性能分布。这种分布相似性实现了我们称为 \textit{$α$Transfer} 的实用范例:在小型代理模型上搜索最佳系数,然后将它们直接转移到更大的目标模型。我们跨多种合并方法、模型系列和任务验证 $α$Transfer。实验结果表明,视觉 Transformer 的速度提高了 6$\times$,记忆降低了 70%,大语言模型的速度提高了 20$\times$,记忆降低了 85%,同时保持了相当的性能。我们的研究结果表明 $α$Transfer 是一种有效且可推广的扩展模型合并方法。

αTransfer:跨模型规模迁移合并系数:论文原图
图 1:$\alpha$Transfer 概述。轴代表任务向量($\tau_{1},\tau_{2}$);上标 $p$、$t$ 表示代理模型和目标模型。等高线显示与系数合并时的性能景观 ($\alpha_{1},\alpha_{2}$)。在廉价代理模型上找到的最佳系数直接转移到大目标,跳过昂贵的搜索。