2019 · 260 citations · 20 references
ComputingComputer ScienceArtificial IntelligenceComputer VisionMachine LearningActive LearningDeep LearningNeural NetworksOverfittingRepresentation LearningUnderfittingActive Learning With Deep Neural NetworksLabel Noise MitigationLabel Noise ModelingLabel Noise RobustnessLabelsRobust LossRobust Loss Functions
Prior methods for noisy labels rely on specially designed loss functions or networks, whereas O2U‑net offers an easy‑to‑implement alternative that remains effective. The paper introduces O2U‑net, a novel noisy‑label detection method for deep neural networks that requires no human annotations. O2U‑net works by repeatedly adjusting the network’s hyper‑parameters to cycle between over‑ and under‑fitting, recording each sample’s loss, and labeling samples with high normalized average loss as noisy, thereby integrating naturally with active learning. Experiments on multiple datasets demonstrate that O2U‑net achieves state‑of‑the‑art performance and adds flexibility for learning with noisy labels.
This paper proposes a novel noisy label detection approach, named O2U-net, for deep neural networks without human annotations. Different from prior work which requires specifically designed noise-robust loss functions or networks, O2U-net is easy to implement but effective. It only requires adjusting the hyper-parameters of the deep network to make its status transfer from overfitting to underfitting (O2U) cyclically. The losses of each sample are recorded during iterations. The higher the normalized average loss of a sample, the higher the probability of being noisy labels. O2U-net is naturally compatible with active learning and other human annotation approaches. This introduces extra flexibility for learning with noisy labels. We conduct sufficient experiments on multiple datasets in various settings. The experimental results prove the state-of-the-art of O2S-net.
20
Deep Residual Learning for Image Recognition
Kaiming He, Xiangyu Zhang, Shaoqing Ren et al. · 2016 · 222.1K citations · Full text
Prototypical Networks for Few-shot Learning
Jake Snell, Kevin Swersky, Richard S. Zemel · arXiv (Cornell University) · 2017 · 5.2K citations · Full text
Yoshua Bengio, Jérôme Louradour, Ronan Collobert et al. · 2009 · 5K citations · Full text
Active Learning Literature Survey
Burr Settles · Minds at UW (University of Wisconsin) · 2009 · 4.8K citations · Full text
Making Deep Neural Networks Robust to Label Noise: A Loss Correction Approach
Giorgio Patrini, Alessandro Rozza, Aditya Krishna Menon et al. · 2017 · 1.4K citations