arXiv:cs.AI· Tomas Karella, Emily Shinkle, Alice Elisabeth Anastasia Allen, Pieter J. Swart, Nicholas Lubbers, Roxana Bujack·· 22 小时前
如何让 SE(2) 不变神经网络达到最优
On Making $\mathrm{SE}(2)$-Invariant Networks Optimal
AI 导读
研究基于经典不变量理论推导出可证明最优的 SE(2) 不变特征映射,在对称群下兼具最小性与信息保持,且无需改动非线性层或归一化层。作者据此提出一个即插即用模块,使标准卷积神经网络同时具备平移与旋转不变性。实验显示其精度与 SOTA 等变方法相当或更优,计算成本显著更低;论文被 NeurIPS 2026 主赛道接收。
正文
Abstract:Respecting a-priori known symmetries of underlying data is a key principle in designing efficient neural architectures. Convolutional neural networks are a canonical example, encoding translation equivariance. Extending this inductive bias to richer symmetry groups such as $\mathrm{SE}(2)$ in the most effective way possible is an ongoing architectural and computational challenge. Building on classical invariant theory, we derive invariant feature maps that are provably optimal: minimal and information-preserving with respect to the symmetry group. Our construction is designed to integrate seamlessly into standard architectures, requiring no modification to nonlinearities or normalization layers. As a concrete implementation, we introduce a plug-and-play module for convolutional neural networks that enforces $\mathrm{SE}(2)$ invariance, capturing both translations and rotations within standard architectures. The resulting models are both efficient and expressive. Experiments show competitive or improved accuracy compared to state-of-the-art equivariant methods at significantly lower computational cost. Finally, we provide a theoretical and empirical analysis of the relationship between equivariance and invariance, offering insight into their roles in controlling robustness and flexibility.
| Comments: | Accepted to the main track of NeurIPS 2026 |
| Subjects: | Machine Learning (cs.LG); Artificial Intelligence (cs.AI) |
| Cite as: | arXiv:2610.17424 [cs.LG] |
| (or arXiv:2610.17424v1 [cs.LG] for this version) | |
| https://doi.org/10.48550/arXiv.2610.17424 arXiv-issued DOI via DataCite (pending registration) |
Submission history
From: Tomas Karella [view email]
[v1]
Mon, 12 Oct 2026 21:38:16 UTC (2,916 KB)
来源:arXiv:cs.AI · arxiv.org