A Comparative Study of Noise Augmentation and Deep Learning Methods on Raman Spectral Classification of Contamination in Hard Disk Drive

Abstract

Deep neural networks have become state-of-the-art for many tasks in the past decade, especially Raman spectral classification. However, these networks heavily rely on a large collection of labeled data to avoid overfitting. Although labeled data is scarce in many application domains, there are techniques to help alleviate the problem, such as data augmentation. In this paper, we investigate one particular kind of data augmentation, noise augmentation that simply adds noise to input samples, for the Raman spectra classification task. Raman spectra yield fingerprint-like information about all chemical components but are prone to noise when the material's particles are small. We study the effectiveness of three noise models for noise augmen-tation in building a robust classification model, including noise from the background chemicals, extended multiplicative signal augmentation (EMSA), and statistical noises. In the experiments, we compared the performance of 11 popular deep learning models with the three noise augmentation techniques. The results suggest that RNN-based models perform relatively well with the increase in augmented data size compared to CNN-based models and that robust noise augmentation methods require characteristics of random variations. However, hyperparameter optimization is crucial for taking optimal advantage of noise augmentation.