Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Model Merging on Loss Landscape: A Geometry Perspective

About

Model merging offers a promising avenue for knowledge integration and parallel development without retraining. Yet, existing methods either ignore the geometry of the loss landscape or rely on intractable full-space Hessian approximations. We propose EpiMer, a framework that casts model merging as solving the Fr\'echet mean on a Riemannian manifold and restricts the computation to a low-rank subspace spanned by the task vectors. With the expected Hessian as the metric, we reveal a connection between local curvature and epistemic uncertainty of the parameters. Our theoretical analysis decomposes the merging error bound into the subspace Fr\'echet variance and the residual energy, and provides a closed-form characterization of when curvature-aware merging provably outperforms flat-geometry methods. In addition, our framework unifies both curvature-aware methods and recent spectral methods as special cases of the subspace Fr\'echet mean with different geometric metrics. Merging fine-tuned CLIP-ViT models on eight image classification tasks, Epistemic Merging strictly outperforms the baselines on all three CLIP-ViT backbones at matched rank, improving the across-task average accuracy and worst-task accuracy on every backbone.

Juanwu Lu, Anand Bhaskar, Brian Axelrod, Ekaterina Tolstaya, Tristan Emrich• 2026

Related benchmarks

TaskDatasetResultRank
Image ClassificationSUN397, Cars, RESISC45, EuroSAT, SVHN, GTSRB, MNIST, DTD (test)
Avg Acc90.6
114
Image Classificationeight-task benchmark CLIP-ViT (minimum over eight per-task cells)
Worst-task Top-1 Accuracy76.3
15
Showing 2 of 2 rows

Other info

Follow for update