This is joint work with
Electrodes record the electrical activity of the brain.
Motor imagery: the subject imagines a movement and the signal should tell us which one.
A trial is a matrix $X\in\mathbb{R}^{C\times T}$, with $C$ channels and $T$ timestamps, carrying a label $y$: right hand, left hand, feet, tongue, rest.
Neighbouring electrodes see related signals. That redundancy is the structure we want to exploit.
The channels are correlated. Important information to predict the class: how pairs of electrodes co-vary over the trial.
Covariance matrices are structured features.
Not a vector space.
Covered by charts $\varphi:U\to\mathbb{R}^d$, with smooth transitions $\psi\circ\varphi^{-1}$. Curved globally, flat locally.
$T_p\mathcal{M}=\{\dot\gamma(0):\gamma(0)=p\}$, the velocities at $p$ of all curves through $p$. A vector space of dimension $d$, attached to $p$.
Joining two points of $\mathcal{M}$: the straight segment of $\mathbb{R}^n$ leaves the manifold at once.
$\exp_p(v)=\gamma(1)$ for the geodesic with $\dot\gamma(0)=v$, and $\log_p=\exp_p^{-1}$ where defined, with $d(p,q)=\lVert\log_p(q)\rVert_p$.
EEG encodes mental state. It is personal data, so raw trials never leave the site that recorded them.
Initialise $\theta_0$. Then repeat for $T$ communication rounds.
Now: $\theta\in\mathcal{M}$, a compact submanifold of $\mathbb{R}^{d\times p}$.
The ambient step leaves $\mathcal{M}$.
Keep only the part $\mathcal{M}$ can follow, the Riemannian gradient $\xi=-\eta\operatorname{grad}f_i(x)\in T_x\mathcal{M}$.
A retraction $R_x$ turns that direction into a point of $\mathcal{M}$.
Computationally cheap, geometry-preserving aggregations
A client sends a submission $\Delta^{(i)}$, built from $\theta_t$ and its local iterate. That submission is a known, deterministic function of its records.
The server observes $\Delta^{(i)}$. It also knows $\theta_t$, the model and the protocol, because it chose them.
So it can guess a record and compute the submission that guess would produce.
Then it minimises the distance between the two submissions, over the guess.
Reconstruction attacks Zhu et al., 2019 and membership inference Shokri et al., 2017 are exactly this, at scale.
The server follows the protocol exactly. It simply keeps everything it sees.
Honest, and curious.
Two databases over the same cohort of clients.
Record-adjacent, $D\bowtie D'$: one client's dataset differs by a single substituted record.
Everything else is untouched, including the other clients.
$E$: any set of outcomes the server may test.
Target: the whole transcript $\mathcal{V}_T$.
$\textcolor{#009E73}{\operatorname{clip}_C}(v) =v\,\min\bigl(1,\,C/\lVert v\rVert\bigr)$
$\mathcal{M}$ is a compact submanifold of $\mathbb{R}^{d\times p}$, carrying the inherited metric. .
| Scheme | $\textsc{Agg.Sub}(\theta_t,\theta_{t,\tau}^{(i)})$ | $\textsc{Agg.Out}(\theta_t,\Delta_t)$ |
|---|---|---|
| KFAvg | $\mathrm{Log}_{\theta_t}\bigl(\theta_{t,\tau}^{(i)}\bigr)$ | $\mathrm{Exp}_{\theta_t}(\Delta_t)$ |
| RetAvg | $\mathrm{R}^{-1}_{\theta_t}\bigl(\theta_{t,\tau}^{(i)}\bigr)$ | $\mathrm{R}_{\theta_t}(\Delta_t)$ |
| ProjAvgours | $\theta_{t,\tau}^{(i)}$ | $\Pi(\Delta_t)$ |
| RLAvgours | $P_{\theta_t}\bigl(\theta_{t,\tau}^{(i)}-\theta_t\bigr)$ | $\Pi(\theta_t+\Delta_t)$ |
| Parameter space | Metric | Privacy | Conv. rate |
|---|---|---|---|
| sphere, Stiefel, Grassmann | induced | ✓ | ✓ |
| flat, including $\log\mathcal{S}_n^{++}$ | Euclidean | ✓ | ✗ |
| $\mathcal{S}_n^{++}$ | affine-invariant | not runnable | |
| Method | Manifold | Privacy | Convergence regime | Geometric ops |
|---|---|---|---|---|
| RFedSVRGLi and Ma, 2022 | general | ✗ | multi-step with one client, or one step with all | $\mathrm{Exp},\mathrm{Log}$, transport |
| AGSHuang et al., 2024 | general | ✗ | multi-step, partial | $\mathrm{R}$, transport |
| PriRFedHuang et al., 2026 | general | ✓ | multi-step with one client, or one step with all | $\mathrm{R}$, $\mathrm{R}^{-1}$ |
| projection-basedZhang et al., 2024 · Wang et al., 2025 | compact | ✗ | multi-step, full | $\Pi$ only |
| oursDP-RFedProj | compact | ✓ | multi-step and partial | $\Pi$ only |
A compact convolutional network, applied to the raw trial instead of its covariance.
Riemannian federated learning
Li and Ma. Federated learning on Riemannian manifolds. 2022.
Huang et al. Riemannian federated learning via averaging gradient streams. 2024.
Zhang et al. Nonconvex federated learning on compact smooth submanifolds. 2024.
Wang et al. Zeroth-order federated learning on compact submanifolds. 2025.
Huang et al. PriRFed. 2026.
McMahan et al. Communication-efficient learning from decentralized data. 2017.
Privacy, geometry, EEG
Dwork et al. Calibrating noise to sensitivity in private data analysis. 2006.
Dong, Roth and Su. Gaussian differential privacy. 2022.
Han et al. Differentially private Riemannian optimization. 2024.
Zhu et al. Deep leakage from gradients. 2019.
Absil and Malick. Projection-like retractions on matrix manifolds. 2012.
Huang and Van Gool. A Riemannian network for SPD matrix learning. 2017.
Lawhern et al. EEGNet. 2018.
Chevallier et al. The largest EEG-based BCI reproducibility study. 2024.