PSMA-Hornet: Fully-automated, multi-target segmentation of healthy organs in PSMA PET/CT images

Prostate-specific membrane antigen (PSMA) PET imaging represents a valuable source of information reflecting disease stage, response rate, and treatment optimization options, particularly with PSMA radioligand therapy. Quantification of radiopharmaceutical uptake in healthy organs from PSMA images has the potential to minimize toxicity by extrapolation of the radiation dose delivery towards personalization of therapy. However, segmentation and quantification of uptake in organs requires labor-intensive organ delineations that are often not feasible in the clinic nor scalable for large clinical trials.

In this work we develop and test the PSMA Healthy organ segmentation network (PSMA-Hornet), a fully-automated deep neural net for simultaneous segmentation of 14 healthy organs representing the normal biodistribution of [¹⁸ F]DCFPyL on PET/CT images. We also propose a modified U-net architecture, a self-supervised pre-training method for PET/CT images, a multi-target Dice loss, and multi-target batch balancing to effectively train PSMA-Hornet and similar networks.

The study used manually-segmented [¹⁸ F]DCFPyL PET/CT images from 100 subjects, and 526 similar images without segmentations. The unsegmented images were used for self-supervised model pretraining. For supervised training, Monte-Carlo cross-validation was used to evaluate the network performance, with 85 subjects in each trial reserved for model training, 5 for validation, and 10 for testing. Image segmentation and quantification metrics were evaluated on the test folds with respect to manual segmentations by a nuclear medicine physician, and compared to inter-rater agreement. The model's segmentation performance was also evaluated on a separate set of 19 images with high tumor load.

With our best model, the lowest mean Dice coefficient on the test set was 0.826 for the sublingual gland, and the highest was 0.964 for liver. The highest mean error in tracer uptake quantification was 13.9% in the sublingual gland. Self-supervised pretraining improved training convergence, train-to-test generalization, and segmentation quality. In addition, we found that a multi-target network produced significantly higher segmentation accuracy than single-organ networks.

The developed network can be used to automatically obtain high-quality organ segmentations for PSMA image analysis tasks. It can be used to reproducibly extract imaging data, and holds promise for clinical applications such as personalized radiation dosimetry and improved radioligand therapy.