Computer Science > Computer Vision and Pattern Recognition
[Submitted on 6 Mar 2021 (v1), revised 3 Mar 2022 (this version, v2), latest version 21 Aug 2022 (v6)]
Title:NVUM: Non-Volatile Unbiased Memory for Robust Medical Image Classification
View PDFAbstract:Real-world large-scale medical image analysis (MIA) datasets havethree challenges: 1) they contain noisy-labelled samples that affect training con-vergence and generalisation, 2) they usually have an imbalanced distribution ofsamples per class, and 3) they normally comprise a multi-label problem, wheresamples can have multiple diagnoses. Current approaches are commonly trainedto solve a subset of those problems, but we are unaware of methods that ad-dress the three problems simultaneously. In this paper, we propose a new trainingmodule called Non-Volatile Unbiased Memory (NVUM), which non-volatilitystores running average of model logits for a new regularization loss on noisymulti-label problem. We further unbias the classification prediction in NVUMupdate for imbalanced learning problem. We run extensive experiments to eval-uate NVUM on new benchmarks proposed by this paper, where training is per-formed on noisy multi-label imbalanced chest X-ray (CXR) training sets, formedby Chest-Xray14 and CheXpert, and the testing is performed on the clean multi-label CXR datasets OpenI and PadChest. Our method outperforms previous state-of-the-art CXR classifiers and previous methods that can deal with noisy labelson all evaluations.
Submission history
From: Fengbei Liu [view email][v1] Sat, 6 Mar 2021 07:42:36 UTC (1,452 KB)
[v2] Thu, 3 Mar 2022 07:11:22 UTC (590 KB)
[v3] Sun, 6 Mar 2022 09:45:16 UTC (1,451 KB)
[v4] Thu, 9 Jun 2022 07:24:59 UTC (579 KB)
[v5] Fri, 17 Jun 2022 10:15:57 UTC (579 KB)
[v6] Sun, 21 Aug 2022 05:56:50 UTC (184 KB)
References & Citations
Bibliographic and Citation Tools
Bibliographic Explorer (What is the Explorer?)
Litmaps (What is Litmaps?)
scite Smart Citations (What are Smart Citations?)
Code, Data and Media Associated with this Article
CatalyzeX Code Finder for Papers (What is CatalyzeX?)
DagsHub (What is DagsHub?)
Gotit.pub (What is GotitPub?)
Papers with Code (What is Papers with Code?)
ScienceCast (What is ScienceCast?)
Demos
Recommenders and Search Tools
Influence Flower (What are Influence Flowers?)
Connected Papers (What is Connected Papers?)
CORE Recommender (What is CORE?)
arXivLabs: experimental projects with community collaborators
arXivLabs is a framework that allows collaborators to develop and share new arXiv features directly on our website.
Both individuals and organizations that work with arXivLabs have embraced and accepted our values of openness, community, excellence, and user data privacy. arXiv is committed to these values and only works with partners that adhere to them.
Have an idea for a project that will add value for arXiv's community? Learn more about arXivLabs.