Escaping the Capacity Ceiling: Routing on the Stiefel Manifold for Bilinear SPD Layers
Not on the map. Why?
Abstract (arXiv)
Deep networks on the symmetric positive-definite (SPD) manifold promise expressive representations by encoding data geometry as an inductive bias, but stacking BiMap layers with the standard ReEig nonlinearity often adds no capacity: on real, preconditioned EEG data, ReEig rarely activates, so the stack behaves as a single layer at any depth. In the worst case, when domains share no discriminative directions, we prove a single filter has a capacity ceiling, so it cannot fully align every domain at once. To overcome that, we propose SCAP (Stiefel Cross-Attention Pool), a layer implementing a family of Stiefel filters by combining a pool of K experts into a sample-specific bilinear map via cross-attention. We show that it matches a per-domain filter bank to first order with fewer experts than domains when domain-optimal filters span few directions near a shared tangent-space basepoint; in the worst case, its alignment empirically stays nearly flat as domains grow, escaping the fixed-filter ceiling. Naively trained, however, this routing can collapse to a fixed filter; we diagnose why and adapt three mechanisms to mitigate it. SCAP significantly improves balanced accuracy over fixed-filter SPDNet on all five cross-domain EEG motor-imagery datasets, and matches or exceeds three domain-adaptive baselines on four out of five.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 13 of its 35 references that are in the map. First posted on arXiv 29 May 2026.
Builds on
13 of 35 references.
References · 13
- 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 30,186
- 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,305
- 2017Outrageously Large Neural Networks: The Sparsely-Gated Mixture-of-Experts Layercited by 3,509
- 2021Switch Transformers: Scaling to Trillion Parameter Models with Simple and Efficient Sparsitycited by 3,073
- 2021Learning to Prompt for Continual Learningcited by 944
- 2016A Riemannian Network for SPD Matrix Learningcited by 162
- 2019Riemannian batch normalization for SPD neural networkscited by 72
- 2022SPD domain-specific batch normalization to crack interpretable unsupervised domain adaptation in EEGcited by 53
- 2024The largest EEG-based BCI reproducibility study for open science: the MOABB benchmarkcited by 29
- 2022Deep Riemannian Networks for End-to-End EEG Decodingcited by 11
- 2024Geometric Neural Network based on Phase Space for BCI-EEG decodingcited by 5
- 2023Riemannian Self-Attention Mechanism for SPD Networkscited by 4
Built on by
Not cited yet.