首次实现高阶张量分解,高效构建对称性保持的基底
High-Rank Irreducible Cartesian Tensor Decomposition and Bases of Equivariant Spaces
- 通过路径矩阵与克莱布施-戈丹系数链式收缩,实现高阶张量分解
- 获得秩n=9的正交不变基底,计算时间从数分钟降至数秒
- 方法完全解析,适用于任意张量积与直和空间的对称设计
不可约笛卡尔张量(ICTs)在等变图神经网络及理论化学中具有关键作用,但高阶张量的对称保持线性算子设计极具挑战。尽管近年来已有针对n=5的显式分解(复杂度阶乘级),本文首次成功构造出秩n=9的ICT分解矩阵,利用路径矩阵结合克莱布施-戈丹矩阵按谱系方案进行链式收缩。证明路径矩阵拼接为笛卡尔张量积空间与球谐直和空间间的正交基变换矩阵。该方法避免使用行简化阶梯形算法,实现完全解析推导,显著提升任意秩正交ICT分解矩阵与等变基底的生成速度。进一步拓展至任意张量积与直和空间,支持自由对称设计。代码已开源,n=6~9的分解分别在28核CPU上耗时1秒、3秒、11秒和4分32秒完成。
原文摘要 · Abstract (English)
Irreducible Cartesian tensors (ICTs) play a crucial role in the design of equivariant graph neural networks, as well as in theoretical chemistry and chemical physics. Meanwhile, the design space of available linear operations on tensors that preserve symmetry presents a significant challenge. The ICT decomposition and a basis of this equivariant space are difficult to obtain for high-rank tensors. After decades of research, Bonvicini (2024) has recently achieved an explicit ICT decomposition for $n=5$ with factorial time/space complexity. In this work we, for the first time, obtain decomposition matrices for ICTs up to rank $n=9$ with reduced and affordable complexity, by constructing what we call path matrices. The path matrices are obtained via performing chain-like contractions with Clebsch-Gordan matrices following the parentage scheme. We prove and leverage that the concatenation of path matrices is an orthonormal change-of-basis matrix between the Cartesian tensor product space and the spherical direct sum spaces. Furthermore, we identify a complete orthogonal basis for the equivariant space, rather than a spanning set (Pearce-Crump, 2023b), through this path matrices technique. Our method avoids the RREF algorithm and maintains a fully analytical derivation of each ICT decomposition matrix, thereby significantly improving the algorithm's speed to obtain arbitrary rank orthogonal ICT decomposition matrices and orthogonal equivariant bases. We further extend our result to the arbitrary tensor product and direct sum spaces, enabling free design between different spaces while keeping symmetry. The Python code is available at https://github.com/ShihaoShao-GH/ICT-decomposition-and-equivariant-bases, where the $n=6,\dots,9$ ICT decomposition matrices are obtained in 1s, 3s, 11s, and 4m32s on 28-core Intel(R) Xeon(R) Gold 6330 CPU @ 2.00GHz, respectively.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。