PRISM Lite: A lightweight model for interactive 3D placenta segmentation in ultrasound

TL;DR

This paper introduces PRISM Lite, a lightweight, real-time interactive 3D placenta segmentation model for ultrasound images that outperforms existing models in accuracy and speed, suitable for clinical use especially in low-resource settings.

Contribution

The paper presents a novel lightweight interactive segmentation model that leverages a fully automated model for initialization and achieves high accuracy with fewer parameters and faster inference.

Findings

Achieves superior segmentation performance compared to state-of-the-art models.

Uses significantly fewer parameters, enabling faster inference.

Robust to poor initial masks, suitable for clinical environments.

Abstract

Placenta volume measured from 3D ultrasound (3DUS) images is an important tool for tracking the growth trajectory and is associated with pregnancy outcomes. Manual segmentation is the gold standard, but it is time-consuming and subjective. Although fully automated deep learning algorithms perform well, they do not always yield high-quality results for each case. Interactive segmentation models could address this issue. However, there is limited work on interactive segmentation models for the placenta. Despite their segmentation accuracy, these methods may not be feasible for clinical use as they require relatively large computational power which may be especially prohibitive in low-resource environments, or on mobile devices. In this paper, we propose a lightweight interactive segmentation model aiming for clinical use to interactively segment the placenta from 3DUS images in real-time. The proposed model adopts the segmentation from our fully automated model for initialization and is designed in a human-in-the-loop manner to achieve iterative improvements. The Dice score and normalized surface Dice are used as evaluation metrics. The results show that our model can achieve superior performance in segmentation compared to state-of-the-art models while using significantly fewer parameters. Additionally, the proposed model is much faster for inference and robust to poor initial masks. The code is available at https://github.com/MedICL-VU/PRISM-placenta.