Building Deep Networks on Grassmann Manifolds
About
Learning representations on Grassmann manifolds is popular in quite a few visual recognition tasks. In order to enable deep learning on Grassmann manifolds, this paper proposes a deep network architecture by generalizing the Euclidean network paradigm to Grassmann manifolds. In particular, we design full rank mapping layers to transform input Grassmannian data to more desirable ones, exploit re-orthonormalization layers to normalize the resulting matrices, study projection pooling layers to reduce the model complexity in the Grassmannian context, and devise projection mapping layers to respect Grassmannian geometry and meanwhile achieve Euclidean forms for regular output layers. To train the Grassmann networks, we exploit a stochastic gradient descent setting on manifolds of the connection weights, and study a matrix generalization of backpropagation to update the structured data. The evaluations on three visual recognition tasks show that our Grassmann networks have clear advantages over existing Grassmann learning methods, and achieve results comparable with state-of-the-art approaches.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Neurodevelopmental Disorder Diagnosis | ABIDE-I LEUVEN AAL116 atlas | Accuracy50.6 | 60 | |
| Brain Network Diagnosis | REST-meta-MDD site S1 | Accuracy52.8 | 52 | |
| Classification | Radar Signal Processing Dataset (5-fold CV) | Accuracy90.48 | 40 | |
| EEG signal classification | MAMEM-SSVEP-II | Accuracy61.23 | 29 | |
| Action Recognition | FPHA (5-fold) | Accuracy85.31 | 19 | |
| Action Recognition | NTU120 (5-fold) | Accuracy57.59 | 19 | |
| Action Recognition | HDM05 (5-fold) | Accuracy63.19 | 19 | |
| Brain Network Diagnosis | ABIDE (LOSO (Leave-One-Site-Out)) | NYU Performance56.3 | 13 | |
| Psychiatric Diagnosis Classification | REST-meta-MDD leave-one-site-out cross-validation rs-fMRI (test) | Classification Accuracy (S1)48.2 | 13 | |
| Video-based 3D action recognition | FPHA | Accuracy78.79 | 8 |