arXiv:2605.21483astro-ph.COcs.LG2026-05

用对称性匹配方法提升宇宙星系速度重建精度,显著增强宇宙微波背景观测信噪比。

Velocityformer: Broken-Symmetry-Matched Equivariant Graph Transformers for Cosmological Velocity Reconstruction

论文配图:Velocityformer: Broken-Symmetry-Matched Equivariant Graph Transformers for Cosmological Velocity Reconstruction
图 1 · 摘自论文原文
  • 设计等变图变压器,匹配观测数据的破坏对称性以提升建模精度。
  • 在低精度模拟仅4个时仍达高精度,速度相关系数提升35%以上。
  • 无需微调即可跨几何、参数和星系样本零样本泛化,适合真实观测场景。

精确测量动力学太阳-泽尔多维奇(kSZ)效应——揭示大尺度重子物质分布的关键宇宙学探针——需要从光谱巡天中准确重构星系速度。kSZ测量的信噪比直接与重构速度和真实速度间的相关系数 $r$ 成正比。本文提出 Velocityformer,一种针对观测数据特定对称性破坏而设计的等变图变压器架构。尽管物理本质对平移和旋转具有等变性,但观测方向偏好导致对称性被打破。通过将模型的归纳偏置与数据的破缺对称性匹配,该方法在所有模型规模和训练体积下均一致提升性能:相比标准线性理论基线,$r$ 提升35%,且优于所有机器学习基线。通过结合基于物理的长波解进行条件建模,Velocityformer具备极高数据效率,仅需4个低保真度模拟即可训练至高精度,并实现输入几何、宇宙学参数与星系样本的零样本泛化。在高保真模拟星系目录上,相较于物理基线,$r$ 提升30%,直接转化为观测数据中相同倍数的信噪比增益。

原文摘要 · Abstract (English)

Precise measurement of the kinematic Sunyaev-Zel'dovich (kSZ) effect - a probe of the large-scale distribution of baryonic matter, a key observable for cosmological inference - requires accurate reconstruction of galaxy velocities from spectroscopic surveys. The signal-to-noise ratio (SNR) of kSZ measurements scales directly with the correlation coefficient $r$ between reconstructed and true velocities. We introduce Velocityformer, an equivariant graph transformer architecture designed to match the specific symmetry of the observational data. While the underlying physics is equivariant with respect to translations and rotations, observational effects break this symmetry due to the preferred line-of-sight direction. Matching the model's inductive bias to the data's broken symmetry consistently improves performance across all model sizes and training volumes, with Velocityformer improving $r$ by 35% over the standard linear theory baseline and outperforming ML baselines at every data volume. By matching the model's inductive bias to the data and conditioning on the physics-based long-wavelength solution, Velocityformer is highly data-efficient, training to high accuracy on as few as 4 low-fidelity simulations, and generalises zero-shot across input geometry, cosmological parameters, and galaxy sample. On high-fidelity simulated galaxy catalogues, this yields a 30% improvement in $r$ over the physical baseline, directly translating to the same SNR gain on observational data.

宇宙学图神经网络速度重建等变模型

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