English

Model Merging on Loss Landscape: A Geometry Perspective

Machine Learning 2026-05-27 v1 Artificial Intelligence Machine Learning

Abstract

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.

Keywords

Cite

@article{arxiv.2605.26693,
  title  = {Model Merging on Loss Landscape: A Geometry Perspective},
  author = {Juanwu Lu and Anand Bhaskar and Brian Axelrod and Ekaterina Tolstaya and Tristan Emrich},
  journal= {arXiv preprint arXiv:2605.26693},
  year   = {2026}
}

Comments

CVPR 2026 Findings Track. 18 pages, 4 figures, 6 tables