Prostate Gleason score prediction via MRI using capsule network

Yuheng Li; Jing Wang; Mingzhe Hu; Pretesh Patel; Hui Mao; Tian Liu; Xiaofeng Yang

doi:10.1117/12.2653621

7 April 2023 Prostate Gleason score prediction via MRI using capsule network

Yuheng Li, Jing Wang, Mingzhe Hu, Pretesh Patel, Hui Mao, Tian Liu, Xiaofeng Yang

Proceedings Volume 12465, Medical Imaging 2023: Computer-Aided Diagnosis; 1246523 (2023) https://doi.org/10.1117/12.2653621
Event: SPIE Medical Imaging, 2023, San Diego, California, United States

Conference Poster

Abstract

Magnetic Resonance imaging (MRI) is a non-invasive modality for diagnosing prostate carcinoma (PCa) and deep learning has gained increasing interest in MR images. We propose a novel 3D Capsule Network to perform low grade vs high grade PCa classification. The proposed network utilizes Efficient CapsNet as backbone and consists of three main components, 3D convolutional blocks, depth-wise separable 3D convolution, and self-attention routing. The network employs convolutional blocks to extract high level features, which will form primary capsules via depth-wise separable convolution operations. A self-attention mechanism is used to route primary capsules to higher level capsules and finally a PCa grade is assigned. The proposed 3D Capsule Network was trained and tested using a public dataset that involves 529 patients diagnosed with PCa. A baseline 3D CNN method was also experimented for comparison. Our Capsule Network achieved 85% accuracy and 0.87 AUC, while the baseline CNN achieved 80% accuracy and 0.84 AUC. The superior performance of Capsule Network demonstrates its feasibility for PCa grade classification from prostate MRI and shows its potential in assisting clinical decision-making.

Citation Download Citation

Yuheng Li, Jing Wang, Mingzhe Hu, Pretesh Patel, Hui Mao, Tian Liu, and Xiaofeng Yang "Prostate Gleason score prediction via MRI using capsule network", Proc. SPIE 12465, Medical Imaging 2023: Computer-Aided Diagnosis, 1246523 (7 April 2023); https://doi.org/10.1117/12.2653621

ACCESS THE FULL ARTICLE

INSTITUTIONAL
Select your institution to access the SPIE Digital Library.

SELECT YOUR INSTITUTION

PERSONAL
Sign in with your SPIE account to access your personal subscriptions or to use specific features such as save to my library, sign up for alerts, save searches, etc.

PERSONAL SIGN IN

No SPIE Account? Create one

PURCHASE THIS CONTENT

SUBSCRIBE TO DIGITAL LIBRARY

50 downloads per 1-year subscription

Members: $195

Non-members: $335 ADD TO CART

25 downloads per 1 - year subscription

Members: $145

Non-members: $250 ADD TO CART

PURCHASE SINGLE ARTICLE

Includes PDF, HTML & Video, when available