arXiv:2509.22184cs.LGcs.AI2025-09被引 1

通过二次型学习群不变与等变函数,高效发现数据中的对称性。

Learning Equivariant Functions via Quadratic Forms

  • 从数据中学习对应二次型矩阵,构建带对称性先验的神经网络
  • 在多项式回归等任务中,精度和对称性发现能力优于基线方法
  • 适用于多输入向量的对称性建模,适合物理、粒子探测等场景

本文提出一种通过学习数据对应的二次型 $x^T A x$ 来发现群等变函数的方法。对于正交群,其保持特定二次型不变,我们利用该性质,在假设对称群为正交群的前提下,通过唯一对称矩阵及其对角形式引入归纳偏置,简化并提升模型效率。所得不变模型保持范数,等变模型可分解为范数不变与尺度不变模型的乘积(指群作用)。进一步扩展至多个输入向量的对角群作用情形,等变函数被分解为仅依赖归一化首向量的角度分量和依赖完整格拉姆矩阵的尺度不变分量,捕捉多输入间依赖关系的同时保留群对称性。在多项式回归、顶夸克标记、转动惯量矩阵预测等多个任务上验证,相比基线方法,本框架在对称性发现与等变函数学习上均表现更优。

原文摘要 · Abstract (English)

In this study, we introduce a method for learning group (known or unknown) equivariant functions by learning the associated quadratic form $x^T A x$ corresponding to the group from the data. Certain groups, known as orthogonal groups, preserve a specific quadratic form, and we leverage this property to uncover the underlying symmetry group under the assumption that it is orthogonal. By utilizing the corresponding unique symmetric matrix and its inherent diagonal form, we incorporate suitable inductive biases into the neural network architecture, leading to models that are both simplified and efficient. Our approach results in an invariant model that preserves norms, while the equivariant model is represented as a product of a norm-invariant model and a scale-invariant model, where the ``product'' refers to the group action. Moreover, we extend our framework to a more general setting where the function acts on tuples of input vectors via a diagonal (or product) group action. In this extension, the equivariant function is decomposed into an angular component extracted solely from the normalized first vector and a scale-invariant component that depends on the full Gram matrix of the tuple. This decomposition captures the inter-dependencies between multiple inputs while preserving the underlying group symmetry. We assess the effectiveness of our framework across multiple tasks, including polynomial regression, top quark tagging, and moment of inertia matrix prediction. Comparative analysis with baseline methods demonstrates that our model consistently excels in both discovering the underlying symmetry and efficiently learning the corresponding equivariant function.

对称性学习等变模型二次型神经网络

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。