Search papers, labs, and topics across Lattice.
The paper introduces SPD Learn, a Python library for geometric deep learning with symmetric positive definite (SPD) matrices, designed for neural decoding applications. It addresses the fragmentation and ad hoc constraint handling in existing SPD-based neural network implementations by providing a unified and modular framework. The library enforces Stiefel/SPD constraints through trivialization-based parameterizations, enabling standard backpropagation and optimization while ensuring manifold-constrained parameters.
SPD Learn streamlines geometric deep learning with SPD matrices by providing a unified Python package that ensures manifold constraints via trivialization, enabling standard backpropagation for neural decoding.
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.
