Hologram Noise Model for Data Augmentation and Deep Learning

Dániel Terbe,László Orzó,Barbara Bicsák,Ákos Zarándy
DOI: https://doi.org/10.3390/s24030948
IF: 3.9
2024-02-02
Sensors
Abstract:This paper introduces a noise augmentation technique designed to enhance the robustness of state-of-the-art (SOTA) deep learning models against degraded image quality, a common challenge in long-term recording systems. Our method, demonstrated through the classification of digital holographic images, utilizes a novel approach to synthesize and apply random colored noise, addressing the typically encountered correlated noise patterns in such images. Empirical results show that our technique not only maintains classification accuracy in high-quality images but also significantly improves it when given noisy inputs without increasing the training time. This advancement demonstrates the potential of our approach for augmenting data for deep learning models to perform effectively in production under varied and suboptimal conditions.
engineering, electrical & electronic,chemistry, analytical,instruments & instrumentation
What problem does this paper attempt to address?
### Problems the Paper Aims to Solve This paper aims to address the issue of performance degradation in deep learning models caused by image quality deterioration in long-term recording systems. Specifically, the researchers propose a new noise augmentation technique to improve the robustness of state-of-the-art deep learning models under poor image quality conditions, using the task of digital holographic microscopy (DHM) image classification. ### Main Research Content 1. **Noise Characteristics Analysis**: - The noise characteristics present in the hologram dataset used for the experiments were studied. 2. **Noise Model Establishment**: - A complex noise model based on empirical findings was proposed to generate realistic synthetic noise. 3. **Impact of Noise Augmentation on Neural Network Training**: - The impact of noise augmentation techniques on the effectiveness of neural network training was explored. ### Specific Methods and Results 1. **Noise Type Analysis**: - Researchers selected 70 noise-reconstructed holograms for statistical analysis, which only contained small particles (Sp) with uniform and consistent background noise. - Experiments showed that the noise intensity distribution followed a normal distribution with a standard deviation of approximately 0.07. - The noise exhibited spatial correlation, manifesting as isotropic diffusion rather than simple uncorrelated white noise. 2. **Noise Model Construction**: - An exponential decay function was used to fit the 1D empirical autocorrelation function to describe the spatial correlation of the noise. - Model parameters indicated that the noise's standard deviation was approximately 0.7, and the real and imaginary parts had similar noise characteristics. 3. **Application of Noise Augmentation Techniques**: - Synthetic noise was added to clean samples to generate noisy holograms. - Results showed that samples generated using colored noise augmentation were closer to actual noisy samples than those generated using white noise augmentation. 4. **Neural Network Training**: - Three different neural network models were trained: a baseline model, a white noise augmented model, and a colored noise augmented model. - On the test set, the colored noise augmented model exhibited better classification accuracy, especially under noisy input conditions. ### Summary This paper proposes a novel noise augmentation technique that effectively enhances the robustness and classification performance of deep learning models when dealing with low-quality images. The method's effectiveness was validated through its application to digital holographic microscopy images. This technique not only maintains classification accuracy on high-resolution images but also significantly improves classification performance under noisy input conditions without increasing training time.