Advancing weak lensing mass mapping with a mask-aware HEALPix transformer

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Wang, Yihe, Yu, Yu
Format: Preprint
Published: 2026
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866918411097866240
author Wang, Yihe
Yu, Yu
author_facet Wang, Yihe
Yu, Yu
contents We present HEALFormer, a transformer-based neural network architecture for weak gravitational lensing mass mapping that reconstructs convergence maps from incomplete and noisy shear observations on the celestial sphere. The model operates directly on the Hierarchical Equal Area isoLatitude Pixelization and employs learnable mask tokens to handle arbitrary survey geometries without requiring preprocessing. Through a progressive training strategy, HEALFormer efficiently processes high-resolution maps up to Nside = 1024 and demonstrates excellent performance across diverse survey footprints including KiDS, DES, DECaLS, and Planck. The model generalizes robustly to cosmological parameters beyond its training set, producing nearly unbiased reconstructions with superior noise suppression compared to traditional Kaiser-Squires and Wiener filter methods. Remarkably, HEALFormer exceeds the theoretical phase recovery limits of linear reconstruction methods at small scales, achieving a fundamental breakthrough in weak lensing analysis. The combination of computational efficiency, reconstruction accuracy, and adaptability to varying survey configurations makes HEALFormer well-suited for current and next-generation cosmological surveys. Code is available at GitHub.
format Preprint
id arxiv_https___arxiv_org_abs_2603_25471
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle Advancing weak lensing mass mapping with a mask-aware HEALPix transformer
Wang, Yihe
Yu, Yu
Instrumentation and Methods for Astrophysics
Cosmology and Nongalactic Astrophysics
We present HEALFormer, a transformer-based neural network architecture for weak gravitational lensing mass mapping that reconstructs convergence maps from incomplete and noisy shear observations on the celestial sphere. The model operates directly on the Hierarchical Equal Area isoLatitude Pixelization and employs learnable mask tokens to handle arbitrary survey geometries without requiring preprocessing. Through a progressive training strategy, HEALFormer efficiently processes high-resolution maps up to Nside = 1024 and demonstrates excellent performance across diverse survey footprints including KiDS, DES, DECaLS, and Planck. The model generalizes robustly to cosmological parameters beyond its training set, producing nearly unbiased reconstructions with superior noise suppression compared to traditional Kaiser-Squires and Wiener filter methods. Remarkably, HEALFormer exceeds the theoretical phase recovery limits of linear reconstruction methods at small scales, achieving a fundamental breakthrough in weak lensing analysis. The combination of computational efficiency, reconstruction accuracy, and adaptability to varying survey configurations makes HEALFormer well-suited for current and next-generation cosmological surveys. Code is available at GitHub.
title Advancing weak lensing mass mapping with a mask-aware HEALPix transformer
topic Instrumentation and Methods for Astrophysics
Cosmology and Nongalactic Astrophysics
url https://arxiv.org/abs/2603.25471