跳到正文
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 主赛道接收。

正文

View PDF HTML (experimental)

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