atlas

Research map / Architectures and efficiency / RNNs and normalization

$C^2M^3$: Cycle-Consistent Multi-Model Merging

NeurIPS 2024 · Architectures and efficiency › RNNs and normalization · approach: Symmetry and equivariance in loss landscapes · 0 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Proposes C²M³, a data-free multi-model weight-matching method that uses Frank-Wolfe over all layers jointly and object-to-universe permutation factorization to guarantee cycle consistency, then averages in the universe space with REPAIR renormalization.

Problem

Merging n≥3 neural networks in weight space requires neuron permutations that are cycle-consistent; existing pairwise, layer-wise matching methods accumulate error along permutation cycles and depend on layer order and mapping direction.

Approach

Factorize each pairwise permutation as P^{pq} = P^p (P^q)^T via a shared universe space, so cycle consistency holds by construction; optimize all layers jointly with a Frank-Wolfe solver on a higher-order objective, then merge by averaging in the universe space.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Apply REPAIR after C²M³ merging (ablation)C²M³ · CIFAR10 test (ResNet 2×)accuracy: 0.40 -> 0.69Helped
Apply REPAIR after C²M³ merging (ablation)C²M³ · CIFAR100 test (ResNet 4×)accuracy: 0.30 -> 0.46Helped
Apply REPAIR after C²M³ merging (ablation)C²M³ · CIFAR10 test (VGG16)accuracy: 0.11 -> 0.62Helped
Apply REPAIR after C²M³ merging (ablation)C²M³ · EMNIST test (MLP)accuracy: 0.87 -> 0.87No clear effect
Apply REPAIR after C²M³ merging (ablation)C²M³ · CIFAR100 test (ResNet 16×)accuracy: 0.39 -> 0.49Helped
Use Frank-Wolfe global matching instead of Git Re-BasinPairwise ResNet 8× matching · CIFAR100test loss barrier: 1.54 ± 0.13 -> 1.42 ± 0.10Helped
Use Frank-Wolfe global matching instead of Git Re-BasinPairwise ResNet20 2× with BatchNorm · CIFAR10test loss barrier: 4.62 ± 0.65 -> 3.79 ± 0.57Helped
Initialize permutations to the barycenter rather than identity (ablation)Frank-Wolfe pairwise matching · ResNet20 2×, models (a, b)test loss barrier: $0.52$ -> $\mathbf{0.47}$Helped
Use Sinkhorn initialization rather than identity (ablation)Frank-Wolfe pairwise matching · ResNet20 2×, models (a, b)test loss barrier: $0.52$ -> $0.60 \pm 0.04$Hurt

Surprising findings

Limitations

Experimental setup

Datasets: MNIST EMNIST CIFAR10 CIFAR100

Benchmarks: MNIST EMNIST CIFAR10 CIFAR100

Metrics: accuracy loss loss barrier cosine similarity wall-clock time

Baselines: Git Re-Basin MergeMany Naive averaging REPAIR

Builds on: Git Re-Basin REPAIR MergeMany Frank-Wolfe algorithm

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy