Stiefel-SPD Manifold Graph Convolution for End-to-End EEG Learning
Abstract
Electroencephalographic (EEG) decoding relies heavily on second-order (covariance) structure that lives on the manifold of symmetric positive-definite (SPD) matrices. Conventional deep networks in Euclidean space ignore this geometry, distorting geodesic relations between covariances; classical Riemannian pipelines respect SPD metrics but typically use fixed projections and a single global tangent embedding, which limits task adaptivity and incurs cubic costs in the channel dimension. We propose a fully geometry-consistent architecture that preserves manifold structure end-to-end while remaining trainable at scale. A compact depthwise-separable convolutional neural network (CNN) produces features whose regularized covariances lie on the SPD manifold. A learnable orthonormal projection, optimized on the Stiefel manifold via Riemannian stochastic gradient descent (SGD) with QR-factorization (QR) retraction, reduces dimensionality without breaking positive-definiteness and preserves an eigenvalue floor. We then perform tangent space graph-SPD aggregation on a scalp $k$ -nearest-neighbor graph-neighbor covariances are transported to the reference tangent space, attention-averaged, and mapped back via the exponential-followed by a log-Euclidean mapping and linear softmax classification. This Stiefel $\!\to $ Graph-SPD $\!\to \log $ chain explains why full geometric consistency matters: it avoids Euclidean shortcuts, keeps all intermediates SPD, and makes log/exp costs cubic in the reduced rank $d$ . In cross-subject evaluation on three public datasets, the model attains ${83}.{2}\%\!/\!{81}.{5}\%\!/\!{79}.{7}\%$ accuracy with improved macro- ${F}_{{1}}$ , strong separability (macro-AUROC $\approx {0}.{90}$ ), and well-calibrated probabilities (ECE $\le {0}.{04}$ ), outperforming strong Euclidean CNNs and Riemannian baselines while remaining computationally pragmatic.
Fields of Study & Subjects
EEG and Brain-Computer Interfaces
Functional Brain Connectivity Studies
Face Recognition and Perception
Publication details
- Journal / source
- IEEE Transactions on Neural Systems and Rehabilitation Engineering
- Publisher
- Institute of Electrical and Electronics Engineers (IEEE)
- Volume
- 34
- Pages
- 595–606
- Publication date
- 1 January 2026
- Language
- EN