Dual-Manifold Geometry Guided Representation Learning: Adaptive Coupling between Kernel and Data Spaces
Not on the map. Why?
Abstract (arXiv)
Deep representation learning has primarily focused on how features evolve across network layers, while largely overlooking the structured geometry embedded in network parameters. We introduce a dual-manifold perspective in which each convolutional layer contains two coupled geometric spaces: a Kernel Manifold induced by convolutional filters and a Data Manifold characterized by intermediate feature representations. Because these manifolds share the same channel space, parameter geometry can provide complementary structural information to guide feature evolution. Based on this insight, we propose Kernel-Guided Feature Transform (KGFT), a lightweight module that derives a geometric guidance matrix from the kernel Gram matrix and uses it to transform the covariance structure of feature representations. Unlike conventional attention mechanisms that reweight feature responses, KGFT explicitly reshapes feature relationships by transferring geometric information from the kernel manifold to the data manifold. To accommodate network hierarchy, we further introduce Exploit and Explore modes with a depth-aware scheduling strategy and a learnable guidance strength that adaptively controls the contribution of geometric transformation. This design promotes geometric alignment in shallow layers while encouraging feature diversity in deeper layers, without imposing excessive constraints on representation learning. Theoretical analysis establishes the validity of the proposed transformation and characterizes its effect on feature covariance. Extensive experiments across CNN- and Transformer-based architectures, including ResNet, ViT, and LLaMA-7B, demonstrate consistent improvements on image classification and arithmetic reasoning tasks, validating the generality and effectiveness of kernel-guided dual-manifold representation learning. Code will be publicly available.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 37 of its 50 references that are in the map. First posted on arXiv 13 Aug 2026.
Builds on
37 of 50 references.
References · 37
- 2017Attention Is All You Needcited by 60,361
- 2015Deep Residual Learning for Image Recognitioncited by 52,096
- 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 30,186
- 2014Very Deep Convolutional Networks for Large-Scale Image Recognitioncited by 15,364
- 2021LoRA: Low-Rank Adaptation of Large Language Modelscited by 14,380
- 2023LLaMA: Open and Efficient Foundation Language Modelscited by 14,132
- 2022Chain-of-Thought Prompting Elicits Reasoning in Large Language Modelscited by 12,459
- 2021Swin Transformer: Hierarchical Vision Transformer using Shifted Windowscited by 12,139
- 2020A Simple Framework for Contrastive Learning of Visual Representationscited by 12,049
- 2021Training Verifiers to Solve Math Word Problemscited by 10,109
- 2021Masked Autoencoders Are Scalable Vision Learnerscited by 7,403
- 2016Grad-CAM: Visual Explanations from Deep Networks via Gradient-based Localizationcited by 6,733
25 more
- 2019EfficientNet: Rethinking Model Scaling for Convolutional Neural Networkscited by 6,442
- 2017Squeeze-and-Excitation Networkscited by 4,789
- 2023Mamba: Linear-Time Sequence Modeling with Selective State Spacescited by 4,718
- 2013Efficient Estimation of Word Representations in Vector Spacecited by 4,375
- 2014Going Deeper with Convolutionscited by 4,241
- 2017MobileNets: Efficient Convolutional Neural Networks for Mobile Vision Applicationscited by 3,845
- 2018Neural Tangent Kernel: Convergence and Generalization in Neural Networkscited by 2,621
- 2018CBAM: Convolutional Block Attention Modulecited by 2,477
- 2019Similarity of Neural Network Representations Revisitedcited by 1,755
- 2017Improved Regularization of Convolutional Neural Networks with Cutoutcited by 1,628
- 2024VMamba: Visual State Space Modelcited by 969
- 2021Are NLP Models really able to Solve Simple Math Word Problems?cited by 878
- 2017Program Induction by Rationale Generation : Learning to Solve and Explain Algebraic Word Problemscited by 650
- 2019ECA-Net: Efficient Channel Attention for Deep Convolutional Neural Networkscited by 575
- 2019Selective Kernel Networkscited by 403
- 2024ReFT: Representation Finetuning for Language Modelscited by 197
- 2020Deep learning versus kernel learning: an empirical study of loss landscape geometry and the time evolution of the Neural Tangent Kernelcited by 187
- 2017Towards Faster Training of Global Covariance Pooling Networks by Iterative Matrix Square Root Normalizationcited by 65
- 2017Is Second-order Information Helpful for Large-scale Visual Recognition?cited by 44
- 2016Low-rank Bilinear Pooling for Fine-Grained Classificationcited by 42
- 2025Scaling Laws and Spectra of Shallow Neural Networks in the Feature Learning Regimecited by 20
- 2021A singular Riemannian geometry approach to Deep Neural Networks I. Theoretical foundationscited by 11
- 2023Non-Parametric Representation Learning with Kernelscited by 11
- 2024Feature learning as alignment: a structural property of gradient descent in non-linear neural networkscited by 10
- 2026The Weight Gram Matrix Captures Sequential Feature Linearization in Deep Networkscited by 5
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2026Geometry-aware similarity metrics for neural representations on Riemannian and statistical manifolds
- 2026The Geometric Wall: Manifold Structure Predicts Layerwise Sparse Autoencoder Scaling Laws
- 2023Manifold-Preserving Transformers are Effective for Short-Long Range Encoding
- 2026From Layers to Networks: Comparing Neural Representations via Diffusion Geometry
- 2026Understanding Geometric Representations in Self-Supervised Vision Transformers via Subspace Intervention
- 2022Guided Deep Metric Learning
- 2026RiT: Vanilla Diffusion Transformers Suffice in Representation Space
- 2025Preserve and Sculpt: Manifold-Aligned Fine-tuning of Vision-Language Models for Few-Shot Learning