Hard Sample Aware Noise Robust Learning for Histopathology Image Classification
Chuang Zhu*
Wenkai Chen
Ting Peng
Ying Wang
Mulan Jin
[Paper]
[GitHub]


Abstract

Deep learning-based histopathology image classification is a key technique to help physicians in improving the accuracy and promptness of cancer diagno- sis. However, the noisy labels are often inevitable in the complex manual annotation process, and thus mislead the training of the classification model. In this work, we intro- duce a novel hard sample aware noise robust learning method for histopathology image classification. To distin- guish the informative hard samples from the harmful noisy ones, we build an easy/hard/noisy (EHN) detection model by using the sample training history. Then we integrate the EHN into a self-training architecture to lower the noise rate through gradually label correction. With the obtained almost clean dataset, we further propose a noise sup- pressing and hard enhancing (NSHE) scheme to train the noise robust model. Compared with the previous works, our method can save more clean samples and can be directly applied to the real-world noisy dataset scenario without using a clean subset. Experimental results demonstrate that the proposed scheme outperforms the current state- of-the-art methods in both the synthetic and real-world noisy datasets. The source code and data are available at https://github.com/bupt-ai-cz/HSA-NRL/. (The original WSIs are scanned at X20 objective magnification.)



Download Chaoyang Dataset

Provide your information to get access to the dataset, before this please check and agree the license.


License

This Chaoyang Dataset is made freely available to academic and non-academic entities for non-commercial purposes such as academic research, teaching, scientific publications, or personal experimentation. Permission is granted to use the data given that you agree to our license terms.



Citation

@article{zhuhard,
title={Hard Sample Aware Noise Robust Learning for Histopathology Image Classification},
author={Zhu, Chuang and Chen, Wenkai and Peng, Ting and Wang, Ying and Jin, Mulan},
journal={IEEE transactions on medical imaging}
}