Equivariant Covariance Tensors: Guaranteed SPD Uncertainty for Tensor-Valued Geometric Learning
Ruihan Liu, Yu Ji, Jianbo Yu, Shifu Yan, Qingchao Jiang
Abstract
Tensor-valued prediction is fundamental to geometric deep learning, yet uncertainty quantification (UQ) for such outputs remains an open challenge. While E(3)-equivariant neural networks excel at point estimates, they lack rigorous confidence measures. We focus on symmetric rank-2 tensor prediction, where the target has six Kelvin-Mandel coordinates and full uncertainty is represented by a covariance matrix. We introduce a framework for E(3)-equivariant UQ, modeling the full predictive distribution where both mean and covariance preserve rotational symmetry. Our approach decomposes the covariance into irreducible representations . By mapping from the flat Lie algebra to the curved SPD manifold via matrix exponentiation, we strictly ensure positive-definite covariances while maintaining exact equivariance. Furthermore, we formulate a Log-Euclidean Equivariant Scoring Objective (LE-ESO)---a robust surrogate loss based on the Multivariate Laplace distribution---providing robustness to heavy-tailed errors and stable optimization. Validation on ModelNet40 inertia tensors and Materials Project dielectric tensors demonstrates that our method achieves competitive performance and provides physically consistent, symmetry-preserving uncertainty estimates with useful risk and OOD sensitivity.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext b66924ac-b2c3-46d1-a91a-b3b64b69538dBuilds on7
- Better Uncertainty Calibration via Proper Scores for Classification and BeyondSebastian G. Gruber, Florian BuettnerNeurIPS 2022 · 88 citations
- Tractable Function-Space Variational Inference in Bayesian Neural NetworksTim G. J. Rudner, Zonghao Chen, Yee Whye Teh, Yarin GalNeurIPS 2022 · 70 citations
- Emergent Equivariance in Deep EnsemblesJan E. Gerken, Pan KesselICML 2024 · 14 citations
- Bayesian Low-Rank Learning (Bella): A Practical Approach to Bayesian Neural NetworksBao Gia Doan, Afshar Shamsi, Xiao-Yu Guo, Arash Mohammadi et al.AAAI 2025 · 1 citation
- Revisiting the Canonicalization for Fast and Accurate Crystal Tensor Property PredictionHaowei Hua, Jingwen Yang, Wanyu Lin, Pan ZhouAAAI 2026 · 1 citation
Related papers
- GotenNet: Rethinking Efficient 3D Equivariant Graph Neural NetworksSarp Aykent, Tian XiaICLR 2025
- A Space Group Symmetry Informed Network for O(3) Equivariant Crystal Tensor PredictionKeqiang Yan, Alexandra Saxton, Xiaofeng Qian, Xiaoning Qian et al.ICML 2024 · 13 citations
- Representation Learning for Equivariant Inference with GuaranteesDaniel Felipe Ordonez Apraez, Vladimir Kostic, Alek Fröhlich, Vivien Brandt et al.ICML 2026
- Image to Sphere: Learning Equivariant Features for Efficient Pose PredictionDavid Klee, Ondrej Biza, Robert Platt, Robin WaltersICLR 2023 · 3 citations
- Equivariant Neural Networks for General Linear Symmetries on Lie AlgebrasChankyo Kim, Sicheng Zhao, Minghan Zhu, Tzu-Yuan Lin et al.ICML 2026 · 2 citations
