Current browse context:
astro-ph.IM
Change to browse by:
References & Citations
Astrophysics > Instrumentation and Methods for Astrophysics
Title: Explaining deep learning of galaxy morphology with saliency mapping
(Submitted on 15 Oct 2021 (v1), last revised 14 Feb 2022 (this version, v2))
Abstract: We successfully demonstrate the use of explainable artificial intelligence (XAI) techniques on astronomical datasets in the context of measuring galactic bar lengths. The method consists of training convolutional neural networks on human classified data from Galaxy Zoo in order to predict general galaxy morphologies, and then using SmoothGrad (a saliency mapping technique) to extract the bar for measurement by a bespoke algorithm. We contrast this to another method of using a convolutional neural network to directly predict galaxy bar lengths. These methods achieved correlation coefficients of 0.76 and 0.59, and root mean squared errors of 1.69 and 2.10 respective to human measurements. We conclude that XAI methods outperform conventional deep learning in this case, which could be reasonably explained by the larger datasets available when training the models. We suggest that our XAI method can be used to extract other galactic features (such as the bulge-to-disk ratio) without needing to collect new datasets or train new models. We also suggest that these techniques can be used to refine deep learning models as well as identify and eliminate bias within training datasets.
Submission history
From: Prabh Bhambra [view email][v1] Fri, 15 Oct 2021 18:00:25 GMT (1383kb,D)
[v2] Mon, 14 Feb 2022 12:54:11 GMT (2314kb,D)
Link back to: arXiv, form interface, contact.