Denoising convolutional autoencoder for improving the classification performance based on noisy galaxy images
Wenda Fu · Applied and Computational Engineering · 2023
Images of galaxy objects are of great importance to the work of astronomers. Nowadays, the task of galaxy classification has been aided by Neural-Network-based classification models, who are powerful yet vulnerable to the attack of noisy images. In this research, RGB noises and bright spots simulating stars were generated, and a Convolutional-Neural-Network (CNN) based lightweight denoising image autoencoder was proposed. Firstly, a benchmark CNN classifier using DenseNet structure was trained on the Galaxy 10 DECals dataset, which consists of over 17, 000 RGB color galaxy images. Then, noisy images were generated by adding bright spots of different size and color simulating stars and applying gaussian RGB noises over the original images. The CNN autoencoder that consists of Convolutional layer in its encoder and Convolution Transpose layers in decoder was trained on the raw and noisy training data to learn effective galaxy image denoising. Finally, the effect of the autoencoder was evaluated by contrasting the performance of the CNN classifier over the noisy and denoised images. In contrast with being evaluated on the raw testing set, the CNN classifier’s accuracy dropped by 0.41 when tested on the generated noisy testing images, indicating the effectiveness of the attack of image noises. While after denoising with the proposed autoencoder, the classifier’s accuracy increased significantly by 0.37. Output denoised images also suggest that the autoencoder can effectively remove the applied bright spot and gaussian RGB noise, recreating the original shape of the galaxy.