Conference material: "Proceedings of the International Conference on Computer Graphics and Vision “Graphicon” (19-21 September 2023, Moscow)"
Authors:Alimagadov K.A., Umnyashkin S.V.
Data Augmentation Based on Wavelet Filtration During Neural Network Training
The research is devoted to studying the influence of data augmentation by means of wavelet filtration on recognition accuracy of noised images by the neural network detector. It is proposed adding noise to samples of the training dataset and then processing them by Wiener filter in the domain of discrete wavelet transform (DWT). Generating of input training images uses the model applying white additive Gaussian noise to distort them. In addition to the Wiener filtering, we considered some thresholding methods in the domain of DWT for images denoising. Based on the obtained experimental data, we compare the results of the detector recognition for the different values of noise standard deviation. Also, we considered the case, when the neural network was trained on noised-augmented images without subsequent filtering. It is shown that the proposed approach allows higher accuracy of recognition to be achieved, than the augmentation, which doesn’t use noise suppression.
Data augmentation, machine learning, neural networks, wavelet filtration