Abstract
Purpose:
Ultrasound-guided biopsy plays a major role in prostate cancer (PCa) detection, yet is limited by a high rate of false negatives and low diagnostic yield of the current systematic, non-targeted approaches. Developing machine learning models for accurately identifying cancerous tissue in ultrasound would help sample tissues from regions with higher cancer likelihood. A plausible approach for this purpose is to use individual ultrasound signals corresponding to a core as inputs and consider the histopathology diagnosis for the entire core as labels. However, this introduces significant amount of label noise to training and degrades the classification performance. Previously, we suggested that histopathology-reported cancer involvement can be a reasonable approximation for the label noise.
Methods:
Here, we propose an involvement-based label refinement (iLR) method to correct corrupted labels and improve cancer classification. The difference between predicted and true cancer involvements is used to guide the label refinement process. We further incorporate iLR into state-of-the-art methods for learning with noisy labels and predicting cancer involvement.
Results:
We use 258 biopsy cores from 70 patients and demonstrate that our proposed label refinement method improves the performance of multiple noise-tolerant approaches and achieves a balanced accuracy, correlation coefficient, and mean absolute error of 76.7%, 0.68, and 12.4, respectively.
Conclusions:
Our key contribution is to leverage a data-centric method to deal with noisy labels using histopathology reports, and improve the performance of prostate cancer diagnosis through a hierarchical training process with label refinement.
Similar content being viewed by others
Notes
The Iverson bracket [P] returns 1 if P is true; 0 otherwise.
References
Sung H, Ferlay J, Siegel RL, Laversanne M, Soerjomataram I, Jemal A, Bray F (2021) Global cancer statistics 2020: globocan estimates of incidence and mortality worldwide for 36 cancers in 185 countries. CA Cancer J Clin 71(3):209–249
Pinto PA, Chung PH, Rastinehad AR, Baccala AA, Kruecker J, Benjamin CJ, Xu S, Yan P, Kadoury S, Chua C, Locklin JK, Turkbey B, Shih JH, Gates SP, Buckner C, Bratslavsky G, Linehan WM, Glossop ND, Choyke PL, Wood BJ (2011) Magnetic resonance imaging/ultrasound fusion guided prostate biopsy improves cancer detection following transrectal ultrasound biopsy and correlates with multiparametric magnetic resonance imaging. J Urol 186(4):1281–1285
Scheenen TW, Rosenkrantz AB, Haider MA, Fütterer JJ (2015) Multiparametric magnetic resonance imaging in prostate cancer management: current status and future perspectives. Invest Radiol 50(9):594–600
Drost F-JH, Osses D, Nieboer D, Bangma CH, Steyerberg EW, Roobol MJ, Schoots IG (2020) Prostate magnetic resonance imaging, with or without magnetic resonance imaging-targeted biopsy, and systematic biopsy for detecting prostate cancer: a cochrane systematic review and meta-analysis. Eur Urol 77(1):78–94
Noble JA, Boukerroui D (2006) Ultrasound image segmentation: a survey. IEEE Trans Med Imaging 25(8):987–1010
Feleppa E, Porter C, Ketterling J, Dasgupta S, Ramachandran S, Sparks D (2007) Recent advances in ultrasonic tissue-type imaging of the prostate. Acoustical Imaging, 331–339
Moradi M, Mahdavi SS, Nir G, Mohareri O, Koupparis A, Gagnon L-O, Fazli L, Casey RG, Ischia J, Jones EC, Goldenberg SL, Salcudean SE (2014) Multiparametric 3D in vivo ultrasound vibroelastography imaging of prostate cancer: Preliminary results. Med Phys 41(7):073505
Sedghi A, Pesteie M, Javadi G, Azizi S, Yan P, Kwak JT, Xu S, Turkbey B, Choyke P, Pinto P, Wood B, Rohling R, Abolmaesumi P, Mousavi P (2019) Deep neural maps for unsupervised visualization of high-grade cancer in prostate biopsies. Int J Comput Assist Radiol Surg 14(6):1009–1016
Javadi G, To MNN, Samadi S, Bayat S, Sojoudi S, Hurtado A, Chang S, Black P, Mousavi P, Abolmaesumi P (2020) Complex cancer detector: Complex neural networks on non-stationary time series for guiding systematic prostate biopsy. In: International conference on medical image computing and computer-assisted intervention, pp 524–533. Springer
Sedghi A, Mehrtash A, Jamzad A, Amalou A, Wells WM, Kapur T, Kwak JT, Turkbey B, Choyke P, Pinto P, Wood B, Xu S, Abolmaesumi P, Mousavi P (2020) Improving detection of prostate cancer foci via information fusion of mri and temporal enhanced ultrasound. Int J Comput Assist Radiol Surg 15(7):1215–1223
Javadi G, Samadi S, Bayat S, Pesteie M, Jafari MH, Sojoudi S, Kesch C, Hurtado A, Chang S, Mousavi P, Black P, Abolmaesumi P (2020) Multiple instance learning combined with label invariant synthetic data for guiding systematic prostate biopsy: a feasibility study. Int J Comput Assist Radiol Surg 15(6):1023–1031
Javadi G, Samadi S, Bayat S, Sojoudi S, Hurtado A, Chang S, Black P, Mousavi P, Abolmaesumi P (2021) Characterizing the uncertainty of label noise in systematic ultrasound-guided prostate biopsy. In: 2021 IEEE 18th International symposium on biomedical imaging (ISBI), pp 424–428. IEEE
Cao K, Chen Y, Lu J, Arechiga N, Gaidon A, Ma T (2021) Heteroskedastic and imbalanced deep learning with adaptive regularization. In: International conference on learning representations
Zhang Z, Sabuncu MR (2018) Generalized cross entropy loss for training deep neural networks with noisy labels. In: 32nd Conference on neural information processing systems (NeurIPS), pp 8792–8802
Ma X, Huang H, Wang Y, Romano S, Erfani S, Bailey J (2020) Normalized loss functions for deep learning with noisy labels. In: International conference on machine learning, pp 6543–6553. PMLR
Han B, Yao Q, Yu X, Niu G, Xu M, Hu W, Tsang IW, Sugiyama M (2018) Co-teaching: robust training of deep neural networks with extremely noisy labels. In: Proceedings of the 32nd international conference on neural information processing systems, pp 8536–8546
Wu P, Zheng S, Goswami M, Metaxas D, Chen C (2020) A topological filter for learning with label noise. In: Advances in neural information processing systems
Zhou X, Liu X, Jiang J, Gao X, Ji X (2021) Asymmetric loss functions for learning with noisy labels. In: International conference on machine learning, pp 12846–12856. PMLR
Lin T-Y, Goyal P, Girshick R, He K, Dollár P (2017) Focal loss for dense object detection. In: Proceedings of the IEEE international conference on computer vision, pp 2980–2988
Sensoy M, Kaplan L, Kandemir M (2018) Evidential deep learning to quantify classification uncertainty. In: Proceedings of the 32nd international conference on neural information processing systems, pp 3183–3193
Ginsburg B, Castonguay P, Hrinchuk O, Kuchaiev O, Lavrukhin V, Leary R, Li J, Nguyen H, Zhang Y, Cohen JM (2019) Stochastic gradient methods with layer-wise adaptive moments for training of deep networks. arXiv preprint arXiv:1905.11286
Ghosh A, Kumar H, Sastry P (2017) Robust loss functions under label noise for deep neural networks. In: Proceedings of the AAAI conference on artificial intelligence, vol. 31, pp 1919–1925
Acknowledgements
We thank NSERC and CIHR for supporting our research.
Author information
Authors and Affiliations
Ethics declarations
Conflict of interest:
All authors confirm that there are no known conflicts of interest with this publication.
Ethical approval:
All procedures performed in studies involving human participants were in accordance with the ethical standards of the institutional and/or national research committee and with the 1964 Helsinki Declaration and its later amendments or comparable ethical standards.
Informed consent:
Informed consent was obtained from all individual participants included in the study.
Additional information
Publisher's Note
Springer Nature remains neutral with regard to jurisdictional claims in published maps and institutional affiliations.
Rights and permissions
About this article
Cite this article
To, M.N.N., Fooladgar, F., Javadi, G. et al. Coarse label refinement for improving prostate cancer detection in ultrasound imaging. Int J CARS 17, 841–847 (2022). https://doi.org/10.1007/s11548-022-02606-2
Received:
Accepted:
Published:
Issue Date:
DOI: https://doi.org/10.1007/s11548-022-02606-2