Explaining deep learning of galaxy morphology with saliency mapping
Explaining deep learning of galaxy morphology with saliency mapping
复制标题
用显着性映射解释星系形态的深度学习
DOI:
10.1093/mnras/stac368
复制
发表时间:
2022
影响因子:
4.8
通讯作者:
Bhambra P
中科院分区:
文献类型:
--
作者:
Bhambra P
We successfully demonstrate the use of explainable artificial intelligence (XAI) techniques on astronomical data sets 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 usingSmoothGrad(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 data sets available when training the models. We suggest that our XAI method can be used to extract other galactic features (such as the bulge-to-disc ratio) without needing to collect new data sets 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 data sets.