Abstract:Traditional cosmic ray filtering algorithms used in X-ray imaging detectors aboard space telescopes perform event reconstruction based on the properties of activated pixels above a certain energy threshold, within 3x3 or 5x5 pixel sliding windows. This approach can reject up to 98% of the cosmic ray background. However, the remaining unrejected background constitutes a significant impediment to studies of low surface brightness objects, which are especially prevalent in the high-redshift universe. The main limitation of the traditional filtering algorithms is their ignorance of the long-range contextual information present in image frames. This becomes particularly problematic when analyzing signals created by secondary particles produced during interactions of cosmic rays with body of the detector. Such signals may look identical to the energy deposition left by X-ray photons, when one considers only the properties within the small sliding window. Additional information is present, however, in the spatial and energy correlations between signals in different parts of the frame, which can be accessed by modern machine learning (ML) techniques. In this work, we continue the development of an ML-based pipeline for cosmic ray background mitigation. Our latest method consist of two stages: first, a frame classification neural network is used to create class activation maps (CAM), localizing all events within the frame; second, after event reconstruction, a random forest classifier, using features obtained from CAMs, is used to separate X-ray and cosmic ray features. The method delivers >40% relative improvement over traditional filtering in background rejection in standard 0.3-10keV energy range, at the expense of only a small (<2%) level of lost X-ray signal. Our method also provides a convenient way to tune the cosmic ray rejection threshold to adapt to a user's specific scientific needs.

Cosmic-CoNN: A Cosmic Ray Detection Deep-Learning Framework, Dataset, and Toolkit

deepCR: Cosmic Ray Rejection with Deep Learning

A New Cosmic-Ray Rejection Routine for HST WFC3/UVIS via Label-free Training of deepCR

Interactive Segmentation and Visualization for Tiny Objects in Multi-megapixel Images

Detection, instance segmentation, and classification for astronomical surveys with deep learning ( deepdisc ): detectron2 implementation and demonstration with Hyper Suprime-Cam data

Supernovae Detection with Fully Convolutional One-Stage Framework.

Towards efficient machine-learning-based reduction of the cosmic-ray induced background in X-ray imaging detectors: increasing context awareness

Detection, Instance Segmentation, and Classification for Astronomical Surveys with Deep Learning (DeepDISC): Detectron2 Implementation and Demonstration with Hyper Suprime-Cam Data

DeepGhostBusters: Using Mask R-CNN to Detect and Mask Ghosting and Scattered-Light Artifacts from Optical Survey Images

An Observationally Driven Multifield Approach for Probing the Circum-Galactic Medium with Convolutional Neural Networks

Deep Neural Network closed-loop with raw data for optical resident space object detection

Deep Learning for the HAWC Gamma-Ray Observatory

A deep learning approach for detecting candidates of supernova remnants

Detection and Classification of Astronomical Targets with Deep Neural Networks in Wide Field Small Aperture Telescopes

Deep-HiTS: Rotation Invariant Convolutional Neural Network for Transient Detection

Deep learning techniques for Imaging Air Cherenkov Telescopes

Galaxy detection and identification using deep learning and data augmentation

CFC-Net: A Critical Feature Capturing Network for Arbitrary-Oriented Object Detection in Remote-Sensing Images

Deep Learning Cosmic Ray Transport from Density Maps of Simulated, Turbulent Gas

COSMIC: A Galaxy Cluster Finding Algorithm Using Machine Learning

RadioGalaxyNET: Dataset and Novel Computer Vision Algorithms for the Detection of Extended Radio Galaxies and Infrared Hosts