Title: Convolutional neural networks on the HEALPix sphere: a pixel-based algorithm and its application to CMB data analysis

Abstract

We describe a novel method for the application of convolutional neural networks (CNNs) to fields defined on the sphere, using the Hierarchical Equal Area Latitude Pixelization scheme (HEALPix). Specifically, we have developed a pixel-based approach to implement convolutional and pooling layers on the spherical surface, similarly to what is commonly done for CNNs applied to Euclidean space. The main advantage of our algorithm is to be fully integrable with existing, highly optimized libraries for NNs (e.g., PyTorch, TensorFlow, etc.). We present two applications of our method: (i) recognition of handwritten digits projected on the sphere; (ii) estimation of cosmological parameter from simulated maps of the cosmic microwave background (CMB). The latter represents the main target of this exploratory work, whose goal is to show the applicability of our CNN to CMB parameter estimation. We have built a simple NN architecture, consisting of four convolutional and pooling layers, and we have used it for all the applications explored herein. Concerning the recognition of handwritten digits, our CNN reaches an accuracy of ~95%, comparable with other existing spherical CNNs, and this is true regardless of the position and orientation of the image on the sphere. For CMB-related applications, we tested the CNNmore » on the estimation of a mock cosmological parameter, defining the angular scale at which the power spectrum of a Gaussian field projected on the sphere peaks. We estimated the value of this parameter directly from simulated maps, in several cases: temperature and polarization maps, presence of white noise, and partially covered maps. For temperature maps, the NN performances are comparable with those from standard spectrum-based Bayesian methods. For polarization, CNNs perform about a factor four worse than standard algorithms. Nonetheless, our results demonstrate, for the first time, that CNNs are able to extract information from polarization fields, both in full-sky and masked maps, and to distinguish between E and B-modes in pixel space. Lastly, we have applied our CNN to the estimation of the Thomson scattering optical depth at reionization ( τ) from simulated CMB maps. Even without any specific optimization of the NN architecture, we reach an accuracy comparable with standard Bayesian methods. This work represents a first step towards the exploitation of NNs in CMB parameter estimation and demonstrates the feasibility of our approach.« less

Authors:

Krachmalnicoff, N. ^[1]; Tomasi, M. ^[2]

---

+ Show Author Affiliations

1. International School for Advanced Studies, Trieste (Italy); Inst. for Fundamental Physics of the Univ., Trieste (Italy)
2. Univ. degli Studi di Milano, Milano (Italy); Istituto Nazionale di Fisica Nucleare (INFN), Milano (Italy)

Publication Date:

2019-08-20

Research Org.:

Lawrence Berkeley National Laboratory (LBNL), Berkeley, CA (United States). National Energy Research Scientific Computing Center (NERSC)

Sponsoring Org.:

USDOE Office of Science (SC)

OSTI Identifier:

1577584

Resource Type:

Accepted Manuscript

Journal Name:

Astronomy and Astrophysics

Additional Journal Information:

Journal Volume: 628; Journal ID: ISSN 0004-6361

Publisher:

EDP Sciences

Country of Publication:

United States

Language:

English

Subject:

79 ASTRONOMY AND ASTROPHYSICS; Astronomy & Astrophysics; Abstract; Full HTML; PDF (2.726 MB); ePUB (6.803 MB); References; NASA ADS Abstract Service