q-bio.NCFeb 26, 2026

SPD Learn: A Geometric Deep Learning Python Library for Neural Decoding Through Trivialization

Authors: Bruno Aristimunha, Ce Ju, Antoine Collas, Florent Bouchard, Ammar Mian, Bertrand Thirion, Sylvain Chevallier, Reinmar Kobler

Organizations: A&O, LISN, Université Paris-Saclay, CNRS, Inria TAU/NERV, France · Yneuro, France · University of San Diego, EUA · Inria, CEA, Université Paris-Saclay, Palaiseau, France · 

Abstract

Implementations of symmetric positive definite (SPD) matrix-based neural networks for neural decoding remain fragmented across research codebases and Python packages. Existing implementations often employ ad hoc handling of manifold constraints and non-unified training setups, which hinders reproducibility and integration into modern deep-learning workflows. To address this gap, we introduce SPD Learn, a unified and modular Python package for geometric deep learning with SPD matrices. SPD Learn provides core SPD operators and neural-network layers, including numerically stable spectral operators, and enforces Stiefel/SPD constraints via trivialization-based parameterizations. This design enables standard backpropagation and optimization in unconstrained Euclidean spaces while producing manifold-constrained parameters by construction. The package also offers reference implementations of representative SPDNet-based models and interfaces with widely used brain computer interface/neuroimaging toolkits and modern machine-learning libraries (e.g., MOABB, Braindecode, Nilearn, and SKADA), facilitating reproducible benchmarking and practical deployment.

Figures & tables

Explore similar work

CardsList
  1. FedSPDnet: Geometry-Aware Federated Deep Learning with SPDnet

    Apr 24, 2026Thibault Pautrel, Florent Bouchard, Ammar Mian +1Federated Learning AggregationFederated Learning

  2. RepSPD: Enhancing SPD Manifold Representation in EEGs via Dynamic Graphs

    Feb 26, 2026Haohui Jia, Zheng Chen, Lingwei Zhu +4EEG DecodingGeometric Representation Learning