Location-based Radiology Report-Guided Semi-supervised Learning for Prostate Cancer Detection

Alex Chen,Nathan Lay,Stephanie Harmon,Kutsev Ozyoruk,Enis Yilmaz,Brad J. Wood,Peter A. Pinto,Peter L. Choyke,Baris Turkbey
2024-06-18
Abstract:Prostate cancer is one of the most prevalent malignancies in the world. While deep learning has potential to further improve computer-aided prostate cancer detection on MRI, its efficacy hinges on the exhaustive curation of manually annotated images. We propose a novel methodology of semisupervised learning (SSL) guided by automatically extracted clinical information, specifically the lesion locations in radiology reports, allowing for use of unannotated images to reduce the annotation burden. By leveraging lesion locations, we refined pseudo labels, which were then used to train our location-based SSL model. We show that our SSL method can improve prostate lesion detection by utilizing unannotated images, with more substantial impacts being observed when larger proportions of unannotated images are used.
Computer Vision and Pattern Recognition,Machine Learning
What problem does this paper attempt to address?
The issue discussed in this paper is how to use location information from radiology reports to guide semi-supervised learning (SSL) for improving prostate cancer detection. Currently, the application of deep learning in prostate cancer detection is limited by the need for a large number of manually annotated images. The study proposes a new SSL method that reduces dependence on annotated images by automatically extracting lesion location information from radiology reports. This method refines pseudo-labels and trains a position-based SSL model, demonstrating that utilizing unlabeled images can enhance the performance of prostate lesion detection, especially when using a large number of unlabeled images. The paper showcases experimental results to demonstrate the advantages of their method in reducing false positive cases and improving detection accuracy.