从数据中自动发现连续对称性,构建可解释的神经网络架构。
Interpretable Discovery of One-parameter Subgroups: A Modular Framework for Elliptical, Hyperbolic, and Parabolic Symmetries
- 基于李代数设计分类型对称架构,按椭圆、双曲、抛物三类建模。
- 在合成与真实数据上准确恢复对称群参数,预测性能优异。
- 适合需要可解释性的物理建模与高能粒子识别任务。
我们提出一种模块化、数据驱动的框架,联合学习未知函数映射并发现支配数据的隐含单参数对称子群。不同于传统几何深度学习假设已知对称性,本方法直接从数据中识别相关连续子群。考虑单参数子群的广义类别,其具有典型的几何分类:椭圆、双曲和抛物。给定假设类别,框架实例化对应的对称发现架构,采用根据子群李代数结构设计的不变与等变表示层,并端到端学习精确的生成元参数。所获模型的不变性或等变性由构造保证,支持形式化证明,使对称性可追溯至架构中的可识别组件。该方法适用于广泛矩阵李群的单参数子群,包括 $SO(n)$、$SL(n)$ 及洛伦兹群。在合成与真实系统上的实验,涵盖转动惯量预测、双摆动力学及高能 extit{Top Quark Tagging},均展示出对称子群的准确恢复与强预测性能,覆盖紧致与非紧致情形。
原文摘要 · Abstract (English)
We propose a modular, data-driven framework for jointly learning unknown functional mappings and discovering the underlying one-parameter symmetry subgroup governing the data. Unlike conventional geometric deep learning methods that assume known symmetries, our approach identifies the relevant continuous subgroup directly from data. We consider the broad class of one-parameter subgroups, which admit a canonical geometric classification into three regimes: elliptical, hyperbolic, and parabolic. Given an assumed regime, our framework instantiates a corresponding symmetry discovery architecture with invariant and equivariant representation layers structured according to the Lie algebra of the subgroup, and learns the exact generator parameters end-to-end from data. This yields models whose invariance or equivariance is guaranteed by construction and admits formal proofs, enabling symmetry to be explicitly traced to identifiable components of the architecture. The approach is applicable to one-parameter subgroups of a wide range of matrix Lie groups, including $SO(n)$, $SL(n)$, and the Lorentz group. Experiments on synthetic and real-world systems, including moment of inertia prediction, double-pendulum dynamics, and high-energy \textit{Top Quark Tagging}, demonstrate accurate subgroup recovery and strong predictive performance across both compact and non-compact regimes.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。