Dec, 2023

任务子空间中的模型匹配合并

TL;DR模型合并通过廉价地将个别任务特定模型合并成一个多任务模型。在本研究中,我们将过去的合并方法视为在合并之前利用不同概念的 “任务子空间” 进行模型匹配。我们将给定模型的任务子空间与其损失函数空间联系起来,并确立了模型合并方法可以看作是解线性方程组的形式化过程。我们探讨了使用共轭梯度方法寻找解决方案的可能性,并展示了共轭梯度方法可以超越解析解的性能,实现对其他难以求解线性系统的合并,并且可以灵活选择 “任务子空间” 的初始值和估计值。我们最终证明了我们的合并框架 “在任务子空间中匹配模型”(MaTS)在多任务和中间任务模型合并方面取得了最先进的结果。我们在此网址 https://URL 中发布了我们工作中使用的所有代码和检查点。