Riemannian--Lorentz Fusion of Vision Transformers and State-Space Models
Not on the map. Why?
Abstract (arXiv)
Scaling deep learning faces critical bottlenecks: data exhaustion, exponential training costs, and resource concentration. Model merging combines pre-trained checkpoints without gradient descent, offering orders-of-magnitude savings versus retraining. Combining independently trained vision models is difficult when their architectures and parameter shapes differ. Existing weight-space merging methods generally assume aligned, shape-compatible checkpoints, whereas a Vision Transformer (ViT) and a state-space model (SSM) implement token mixing with different operators. We study a hybrid Heterogeneous merging setting that retains both architectures while aligning parameter groups by semantic role. Our proposed Riemannian--Lorentz Parameter Fusion (RLPF) method projects aligned groups to common coordinates, lifts selected coordinates to the Lorentz hyperboloid model of hyperbolic space, computes a regularized geodesic barycenter, and decodes the result into the two branches. A learned gate then combines branch logits for each input. Component groups use fixed curvature values, with normalization parameters treated as Euclidean. In the results available in this manuscript, the fine-tuned system obtains 82.37% on CIFAR-10, 75.04% on Oxford-IIIT Pet, and 78.58% top-1 accuracy on ImageNet-1K; the corresponding best-parent accuracies are 76.54%, 71.42%, and 76.42%. On ImageNet-1K, the reported pre-fine-tuning initialization reaches 77.80%. These results support further study of geometry-aware heterogeneous fusion, but not a training-free single-checkpoint merge: RLPF is a two-branch hybrid whose gate and reported final models are trained.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 23 of its 33 references that are in the map. First posted on arXiv 16 Sep 2026.
Builds on
23 of 33 references.
References · 23
- 2017Attention Is All You Needcited by 60,361
- 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 30,186
- 2023Mamba: Linear-Time Sequence Modeling with Selective State Spacescited by 4,718
- 2020Training data-efficient image transformers & distillation through attentioncited by 4,586
- 2021Efficiently Modeling Long Sequences with Structured State Spacescited by 2,465
- 2022Model soups: averaging weights of multiple fine-tuned models improves accuracy without increasing inference timecited by 1,396
- 2022Editing Models with Task Arithmeticcited by 1,120
- 2024VMamba: Visual State Space Modelcited by 969
- 2023TIES-Merging: Resolving Interference When Merging Modelscited by 818
- 2023Language Models are Super Mario: Absorbing Abilities from Homologous Models as a Free Lunchcited by 694
- 2017Poincar'e Embeddings for Learning Hierarchical Representationscited by 684
- 2022Git Re-Basin: Merging Models modulo Permutation Symmetriescited by 471
11 more
- 2019Hyperbolic Graph Convolutional Neural Networkscited by 402
- 2018Representation Tradeoffs for Hyperbolic Embeddingscited by 206
- 2018Riemannian Adaptive Optimization Methodscited by 194
- 2024PlainMamba: Improving Non-Hierarchical Mamba in Visual Recognitioncited by 141
- 2024SiMBA: Simplified Mamba-Based Architecture for Vision and Multivariate Time seriescited by 82
- 2023SpectFormer: Frequency and Attention is what you need in a Vision Transformercited by 48
- 2024Revisiting Weight Averaging for Model Mergingcited by 25
- 2023Scattering Vision Transformer: Spectral Mixing Matterscited by 17
- 2025MergeME: Model Merging Techniques for Homogeneous and Heterogeneous MoEscited by 13
- 2026Transport and Merge: Cross-Architecture Merging for Large Language Modelscited by 7
- 2026HAMSA: Scanning-Free Vision State Space Models via SpectralPulseNetcited by 3
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2026hyperbolix: Hyperbolic Deep Learning in JAX
- 2026Task-Induced Riemannian Metrics for Vision Transformer Feature Spaces
- 2026OrthoFuse: Training-free Riemannian Fusion of Orthogonal Style-Concept Adapters for Diffusion Models
- 2025FastVGGT: Training-Free Acceleration of Visual Geometry Transformer
- 2025Riemannian Consistency Model
- 2026RGSQ: Riemannian Geometry-Sensitive Quantization for Large Vision-Language Models
- 2024NRDF: Neural Riemannian Distance Fields for Learning Articulated Pose Priors
- 2023Flow Matching on General Geometries