Nov, 2023

使用轨道距离最小化学习等变性的对称化方法

TL;DR我们提出了一个通用的框架,用于将任意神经网络架构进行对称化,并使其关于给定群体具有等变性。我们构建在 Kim 等人(2023)和 Kaba 等人(2023)的对称化建议的基础上,并通过用一个直观地测量群轨道之间距离的优化来取代将神经特征转换为群表示的方法来改进它们。这种改变使我们的方法适用于比这两个提议更广泛的矩阵群,如 Lorentz 群 O (1, 3)。我们在 SO (2) 图像分类任务上实验证明了我们方法的竞争力,同时也展示了在 O (1, 3) 任务上的更广泛性。我们的实现将在此 https URL 上提供访问。