SYNTOLOGY HomeExplorerAtlasCodeMethodologyAboutDevelopersFeedPricing
Paper · 2402.05966 · 2024

Vanishing Feature: Diagnosing Model Merging and Beyond

arXiv · PDF · Open in the Atlas

Code that ran

We lifted 3 functions out of this paper's own repositories and ran 2 of them in a sandbox. "Ran" means the function executed on a synthesized input and returned a value. It is not a reproduction of the paper's results.

RepositoryRoleRan
xingyuqu/vf canonical 2 of 2
xingyuqu/rethink-re-basin extension 0 of 1
FunctionStatusWhere it lives
validate Ran xingyuqu/vf/PFM/cifar_cca_experiments.py
pointer only (licence: NONE) · get_code("5a9f46b9b120ef6e")
validate_ensemble Ran xingyuqu/vf/PFM/cifar_cca_experiments.py
pointer only (licence: NONE) · get_code("b015dad7216d04cf")
fim_diag Not yet run xingyuqu/rethink-re-basin/source/utils/fim.py
pointer only (licence: NONE) · get_code("258a4b4f429690a1")

Repositories linked to this paper

Some links come from the archived Papers with Code dataset (CC BY-SA 4.0): attribution and licence.

Abstract

Model merging offers an efficient way to combine pre-trained neural networks but often suffers from inconsistent performance, especially when merging models with different initializations. We identify the ``vanishing feature'' phenomenon, where input-induced features diminish during propagation through the merged model, degrading performance. Through theoretical and empirical analysis, we reveal that this phenomenon underpins challenges like variance collapse and explains techniques like permutation-based merging, post-merging normalization, etc. We show that existing normalization strategies can be enhanced by precisely targeting the vanishing feature issue. Leveraging these insights, we propose the ``Preserve-First Merging'' (PFM) strategy, which focuses on preserving early-layer features, enabling the merged models, for the first time, to outperform the original models in advanced settings without post-training. Furthermore, we demonstrate that the vanishing feature phenomenon extends to other contexts, such as model pruning. Applying post-pruning normalization to mitigate the issue significantly improves one-shot pruning performance at high sparsity, offering a simple and effective post-pruning solution. The code is available at https://github.com/XingyuQu/VF.

For agents

The same record, over MCP at https://syntology.ai/mcp:

get_harvested_code_for_paper("2402.05966")
get_code_for_paper("2402.05966")
have("2402.05966")

Connect an agent — have() is free.