An Effective Label Noise Model for DNN Text Classification
This addresses the problem of label noise in text classification datasets for researchers and practitioners, offering a domain-specific solution that is incremental in adapting existing noise-handling methods to text data.
The paper tackles training deep neural networks for text classification with label noise by introducing a non-linear noise model layer into a CNN, which prevents overfitting to erroneous labels and improves sentence representations, achieving robustness even under extreme noise conditions.
Because large, human-annotated datasets suffer from labeling errors, it is crucial to be able to train deep neural networks in the presence of label noise. While training image classification models with label noise have received much attention, training text classification models have not. In this paper, we propose an approach to training deep networks that is robust to label noise. This approach introduces a non-linear processing layer (noise model) that models the statistics of the label noise into a convolutional neural network (CNN) architecture. The noise model and the CNN weights are learned jointly from noisy training data, which prevents the model from overfitting to erroneous labels. Through extensive experiments on several text classification datasets, we show that this approach enables the CNN to learn better sentence representations and is robust even to extreme label noise. We find that proper initialization and regularization of this noise model is critical. Further, by contrast to results focusing on large batch sizes for mitigating label noise for image classification, we find that altering the batch size does not have much effect on classification performance.