论文提出 PRISM:预测并修复模型合并中的性能坍塌
Predicting and Repairing Merge Collapse in Large Language Models
合并几个微调模型结果崩了还没预警?这篇论文用任务向量方差提前预测坍塌,还给出 PRISM 算子免调参修复,代码开源了。
arXiv 论文 2610.03199 研究多个微调模型通过任务向量平均合并后性能远低于基座模型的坍塌现象。作者用专家模型任务向量的方差作为干扰度量,证明平均操作移除的功率等于该方差,并据此构建合并前评分。在来自 4 个模型家族的 22 种合并配置上,只有破坏性合并超过该评分阈值;在 14 次合并结果预测中猜中 12 次。修复方面提出 PRISM 算子,对平均后的任务向量按每层干扰水平做软阈值处理,5 个破坏性合并全部恢复到基座模型评估噪声范围内,而普通平均至少低 14.4 个分点。
Predicting and Repairing Merge Collapse in Large Language Models
Large language models fine-tuned from a shared base can be merged by averaging their task vectors, but some merges collapse far below the base model, and common merge operators give no warning before evaluation. We show that one statistic of the specialists' task vectors both predicts this collapse and calibrates its repair. The power that averaging removes equals the variance of the task vectors across specialists, our measure of interference. Under a working noise model, the disturbance that a merge injects grows with the merge coefficient and with interference, yielding a pre-merge score. In our experiments on twenty-two merge configurations from four model families, only destructive merges exceed a threshold on this score. We find that statistics of sign conflict between specialists, a common target of existing merge operators, are anti-predictive. We then predicted the outcomes of fourteen merges before evaluating them, and twelve predictions were correct, including the destructive outcome of a specialist pair pushed past the threshold by continued pretraining. To address this collapse, we introduce PRISM, an operator that averages the task vectors first and then soft-thresholds each layer at a level set by the layer's interference. Without data or tuning, PRISM keeps all five destructive merges above the threshold within evaluation noise of the base model, where plain averaging falls at least 14.4 points below it or collapses entirely. We apply PRISM only above the threshold and keep the plain average for merges below it, which include all fifteen harmless ones. Code is available at https://github.com/js-lee-AI/PRISM.