近日,一篇題為《Discretizing Group-Convolutional Neural Networks for 3D Geometry in Feature Space》的論文引起了廣泛關注。該論文由Daniel Franzen等三位作者共同完成,於2026年5月14日提交至arXiv,全長11頁,包含7張圖表和2張表格。論文的核心貢獻在於提出了一種新穎的特徵空間採樣方法,旨在解決羣卷積神經網絡(Group-Convolutional Neural Networks, GCNNs)在處理三維幾何數據時面臨的計算成本高昂的問題。
GCNN是深度學習中將對稱性作為歸納偏置的重要工具。在每個線性層中,GCNN會對一個變換羣G進行密集採樣,並在不同姿態下關聯數據和濾波器(對於可操縱的GCNN,還需進行適當的抗混疊處理),以保持對G的等變性。然而,這種密集採樣導致濾波器需要應用於大量數據項,計算成本非常昂貴。即便僅考慮平移變換(即普通CNN的情況),成本也不容忽視;而當自由度增加時(如三維中的平移和旋轉),計算成本呈指數級增長,這嚴重阻礙了GCNN在實際應用中的推廣。
為了應對這一挑戰,作者提出了在特徵空間中進行採樣的方法。具體來説,他們不再對幾何空間進行密集採樣,而是根據特徵相似性選擇代表性的樣本,從而大幅減少需要處理的數據量。這種方法成功地將幾何分辨率與訓練和推理過程中的內存及計算成本解耦,為在計算效率和精度之間進行權衡提供了一條全新的途徑。
論文的主要實驗發現是,即使採用非常粗粒度的特徵空間採樣,分類精度仍然能夠保持得很好。這一結果意味着,研究人員可以基於幾何相似性進行預計算,從而顯著加速等變三維分類器的訓練過程。該研究為三維幾何深度學習中的計算效率問題提供了切實可行的解決方案,有望推動GCNN在計算機視覺、圖形學和機器學習等領域的實際應用。