Continuous max-flow augmentation of self-supervised few-shot learning on SPECT left ventricles
Ádám István Szűcs, Béla Kári, Oszkár Pártos
TL;DR
This work tackles automatic left-ventricle segmentation in SPECT MPI under very small labeled datasets and varied acquisition geometries. It proposes a three-part approach: a 3D U-Net trained with self-supervised learning and heavy augmentation, a Continuous Max-Flow refinement to enforce global segmentation consistency, and kernel-density-based shape priors to accommodate non-Gaussian LV shapes, especially in hypoperfused cases. The integrated CMF-shape-prior framework yields 5–10% improvements over prior state-of-the-art methods on complete FOV volumes, including ill-conditioned cases, while highlighting wall-thickness artefacts as an area for further refinement. The approach is implemented in PyTorch with standard DL and optimization tools and validated on 102 patient datasets across multiple acquisition settings, demonstrating practical potential for diagnostic centers with limited labeled data.
Abstract
Single-Photon Emission Computed Tomography (SPECT) left ventricular assessment protocols are important for detecting ischemia in high-risk patients. To quantitatively measure myocardial function, clinicians depend on commercially available solutions to segment and reorient the left ventricle (LV) for evaluation. Based on large normal datasets, the segmentation performance and the high price of these solutions can hinder the availability of reliable and precise localization of the LV delineation. To overcome the aforementioned shortcomings this paper aims to give a recipe for diagnostic centers as well as for clinics to automatically segment the myocardium based on small and low-quality labels on reconstructed SPECT, complete field-of-view (FOV) volumes. A combination of Continuous Max-Flow (CMF) with prior shape information is developed to augment the 3D U-Net self-supervised learning (SSL) approach on various geometries of SPECT apparatus. Experimental results on the acquired dataset have shown a 5-10\% increase in quantitative metrics based on the previous State-of-the-Art (SOTA) solutions, suggesting a good plausible way to tackle the few-shot SSL problem on high-noise SPECT cardiac datasets.
