Improved Histology Image Classification under Label Noise Via Feature Aggregating Memory Banks
Published in 2022 IEEE 18th International Symposium on Biomedical Imaging (ISBI), 2022
Medical images are often subjectively labeled or weakly supervised, which introduces both closed-set and open-set label noise. Deep learning (DL) models underperform when the quality of supervisory labels degrades. We propose a unified framework for robust training of DL models that takes into account both closed-set and open-set noise, whereas the latter has been largely overlooked. We use a contrastive learning framework and feature aggregating memory banks to identify and increase the emphasis on clean training samples. Based on two publicly available datasets we demonstrate improved results for both synthetically introduced and naturally occurring (unknown) label noise.
Citation
N. C. Kurian, S. Varsha, A. Bajpai, S. Patel and A. Sethi, “Improved Histology Image Classification under Label Noise Via Feature Aggregating Memory Banks,” 2022 IEEE 19th International Symposium on Biomedical Imaging (ISBI), Kolkata, India, 2022, pp. 1-5, doi: 10.1109/ISBI52829.2022.9761682.