We gratefully acknowledge support from
the Simons Foundation and member institutions.
Full-text links:

Download:

Current browse context:

cs.CV

Change to browse by:

cs

References & Citations

DBLP - CS Bibliography

Bookmark

(what is this?)
CiteULike logo BibSonomy logo Mendeley logo del.icio.us logo Digg logo Reddit logo

Computer Science > Computer Vision and Pattern Recognition

Title: Lesion Harvester: Iteratively Mining Unlabeled Lesions and Hard-Negative Examples at Scale

Abstract: Acquiring large-scale medical image data, necessary for training machine learning algorithms, is frequently intractable, due to prohibitive expert-driven annotation costs. Recent datasets extracted from hospital archives, e.g., DeepLesion, have begun to address this problem. However, these are often incompletely or noisily labeled, e.g., DeepLesion leaves over 50% of its lesions unlabeled. Thus, effective methods to harvest missing annotations are critical for continued progress in medical image analysis. This is the goal of our work, where we develop a powerful system to harvest missing lesions from the DeepLesion dataset at high precision. Accepting the need for some degree of expert labor to achieve high fidelity, we exploit a small fully-labeled subset of medical image volumes and use it to intelligently mine annotations from the remainder. To do this, we chain together a highly sensitive lesion proposal generator and a very selective lesion proposal classifier. While our framework is generic, we optimize our performance by proposing a 3D contextual lesion proposal generator and by using a multi-view multi-scale lesion proposal classifier. These produce harvested and hard-negative proposals, which we then re-use to finetune our proposal generator by using a novel hard negative suppression loss, continuing this process until no extra lesions are found. Extensive experimental analysis demonstrates that our method can harvest an additional 9,805 lesions while keeping precision above 90%. To demonstrate the benefits of our approach, we show that lesion detectors trained on our harvested lesions can significantly outperform the same variants only trained on the original annotations, with boost of average precision of 7% to 10%. We open source our annotations at this https URL
Comments: 13 pages, 13 figures, to appear in IEEE Transactions on Medical Imaging
Subjects: Computer Vision and Pattern Recognition (cs.CV)
Journal reference: IEEE Transactions on Medical Imaging, 2020
DOI: 10.1109/TMI.2020.3022034
Cite as: arXiv:2001.07776 [cs.CV]
  (or arXiv:2001.07776v3 [cs.CV] for this version)

Submission history

From: Jinzheng Cai [view email]
[v1] Tue, 21 Jan 2020 21:09:49 GMT (1191kb,D)
[v2] Tue, 28 Jan 2020 05:09:07 GMT (998kb,D)
[v3] Mon, 23 Nov 2020 17:50:20 GMT (2281kb,D)

Link back to: arXiv, form interface, contact.