核心概念
本文提出一個在一般幾何上設計Riemannian多項式羅吉斯回歸(RMLR)的框架,只需要最小的幾何性質,從而展現廣泛的適用性。在對稱正定(SPD)流形上,我們系統地提出五個家族的SPD MLR,並在SO(n)上提出Lie MLR。大量實驗驗證了我們框架的有效性。
摘要
本文提出了一個通用的Riemannian多項式羅吉斯回歸(RMLR)框架,只需要最小的幾何性質,如Riemannian對數映射,就可以將Euclidean MLR推廣到流形上。這相比之前的方法,如依賴特定幾何性質的gyro MLR和平面MLR,具有更廣泛的適用性。
在對稱正定(SPD)流形上,作者系統地提出了五個家族的SPD MLR,涵蓋了之前的gyro SPD MLR和平面SPD MLR作為特例。這些SPD MLR基於不同的Riemannian度量,如仿射不變度量(AIM)、對數歐氏度量(LEM)、功率歐氏度量(PEM)、對數Cholesky度量(LCM)和Bures-Wasserstein度量(BWM)。作者還提出了Lie MLR,用於分類SO(n)上的旋轉矩陣,這是第一次將Euclidean MLR推廣到Lie群。
大量實驗表明,在SPD神經網絡和Lie神經網絡上,我們的RMLR一致優於非內在的LogEig MLR。特別是,我們的SPD MLR在人體動作識別任務上的性能提升可達14.23%,在EEG分類任務上提升4.46%。此外,我們的Lie MLR還可以提高訓練的穩定性。
統計資料
在SPDNet的人體動作識別任務上,我們的SPD MLR在2-Block和5-Block配置下的準確率分別提高了3.74%和2.4%。
在TSMNet的EEG分類任務上,我們的SPD MLR在inter-session和inter-subject實驗中的平衡準確率分別提高了2.6%和4.46%。
在RResNet的人體動作識別任務上,我們的SPD MLR在HDM05和NTU60數據集上的準確率分別提高了13.72%和8.41%。
引述
"我們的SPD MLR在人體動作識別任務上的性能提升可達14.23%,在EEG分類任務上提升4.46%。"
"我們的Lie MLR還可以提高訓練的穩定性。"