README.md

Cosmic-Ray-Detection-via-Dictionary-Learning

This work presents a novel Dictionary Learning (DRL) based framework to detect Cosmic Ray (CR) hits that contaminate the astronomical images obtained through optical photometric surveys. The unique and distinguishable spatial signatures of CR hits compared to other actual astrophysical sources in the image motivated us to characterize the CR patches uniquely via their sparse representations obtained from a learned dictionary. Specifically, the dictionary is trained on images acquired from the Dark Energy Camera (DECam) observations. Next, the learned dictionary represents the CR and Non-CR patches (e.g., each patch with $11 \times 11$ pixel resolution) extracted from the original images. A Machine Learning (ML) classifier is then trained to classify the CR and Non-CR patches. Empirically, we demonstrate that the proposed DRL-based method can detect the CR hits at patch level and provide approximately $83%$ detection rates on test data from multiple photometric bands of the DECam with the Random Forest (RF) algorithm. Further, we used the coarse segmentation maps obtained from the classifier output to guide the deep-learning-based pixel-level CR segmentation models. The coarse maps are fed through a separate channel along with the contaminated image to detect the CR-induced pixels more accurately. We evaluated the performance of proposed DL-guided deep segmentation models over the baseline on test data from DECam. We demonstrate that the proposed method provides additional guidance to the baseline models regarding faster convergence rate and improves CR detection performance by $2%$ in the case of shallow models.

This work is accepted to the 30th European Signal Processing Conference (EUSIPCO) 2022, held in Belgrade, Serbia, from 29 August to 2 September 2022. The link for the paper is https://ieeexplore.ieee.org/document/9909810

Following are our major references from which we adopted and used the codes in this work.

1. We adopted Approximate KSVD from https://github.com/nel215/ksvd.
2. deepCR baseline model from https://github.com/profjsb/deepCR.