Vesal, Sulaiman;
Gayo, Iani;
Bhattacharya, Indrani;
Natarajan, Shyam;
Marks, Leonard S;
Barratt, Dean C;
Fan, Richard E;
... Rusu, Mirabela; + view all
(2022)
Domain generalization for prostate segmentation in transrectal ultrasound images: A multi-center study.
Medical Image Analysis
, 82
, Article 102620. 10.1016/j.media.2022.102620.
Preview |
Text
domain_generalisation.pdf - Accepted Version Download (31MB) | Preview |
Abstract
Prostate biopsy and image-guided treatment procedures are often performed under the guidance of ultrasound fused with magnetic resonance images (MRI). Accurate image fusion relies on accurate segmentation of the prostate on ultrasound images. Yet, the reduced signal-to-noise ratio and artifacts (e.g., speckle and shadowing) in ultrasound images limit the performance of automated prostate segmentation techniques and generalizing these methods to new image domains is inherently difficult. In this study, we address these challenges by introducing a novel 2.5D deep neural network for prostate segmentation on ultrasound images. Our approach addresses the limitations of transfer learning and finetuning methods (i.e., drop in performance on the original training data when the model weights are updated) by combining a supervised domain adaptation technique and a knowledge distillation loss. The knowledge distillation loss allows the preservation of previously learned knowledge and reduces the performance drop after model finetuning on new datasets. Furthermore, our approach relies on an attention module that considers model feature positioning information to improve the segmentation accuracy. We trained our model on 764 subjects from one institution and finetuned our model using only ten subjects from subsequent institutions. We analyzed the performance of our method on three large datasets encompassing 2067 subjects from three different institutions. Our method achieved an average Dice Similarity Coefficient (Dice) of 94.0±0.03 and Hausdorff Distance (HD95) of 2.28 mm in an independent set of subjects from the first institution. Moreover, our model generalized well in the studies from the other two institutions (Dice: 91.0±0.03; HD95: 3.7 mm and Dice: 82.0±0.03; HD95: 7.1 mm). We introduced an approach that successfully segmented the prostate on ultrasound images in a multi-center study, suggesting its clinical potential to facilitate the accurate fusion of ultrasound and MRI images to drive biopsy and image-guided treatments.
| Type: | Article |
|---|---|
| Title: | Domain generalization for prostate segmentation in transrectal ultrasound images: A multi-center study |
| Location: | Netherlands |
| Open access status: | An open access version is available from UCL Discovery |
| DOI: | 10.1016/j.media.2022.102620 |
| Publisher version: | https://doi.org/10.1016/j.media.2022.102620 |
| Language: | English |
| Additional information: | This version is the author accepted manuscript. For information on re-use, please refer to the publisher’s terms and conditions. |
| Keywords: | Transrectal ultrasound, Gland segmentation, Deep learning, Prostate MRI, Targeted biopsy, Continual learning segmentation |
| UCL classification: | UCL UCL > Provost and Vice Provost Offices > UCL BEAMS UCL > Provost and Vice Provost Offices > UCL BEAMS > Faculty of Engineering Science UCL > Provost and Vice Provost Offices > UCL BEAMS > Faculty of Engineering Science > Dept of Med Phys and Biomedical Eng |
| URI: | https://discovery.ucl.ac.uk/id/eprint/10161193 |
Archive Staff Only
 |
View Item |
