Search papers, labs, and topics across Lattice.
This paper introduces TRIAGE, a novel interactive segmentation pipeline for lesion detection in whole-body PET/CT imaging that leverages a 3D STU-Net architecture pre-trained with masked autoencoding. The model is designed to efficiently integrate sparse user corrections while accounting for the significant variability in tracer distributions and physiological uptake patterns between FDG and PSMA studies. Key results demonstrate that TRIAGE's architecture, which includes a dedicated tracer classifier and curriculum-style training, enhances segmentation accuracy and robustness across diverse imaging scenarios.
Interactive lesion segmentation can now adaptively refine predictions in real-time, significantly improving accuracy in heterogeneous PET/CT datasets.
Interactive lesion segmentation in whole-body PET/CT requires a model to provide a strong initial prediction while also responding efficiently to sparse corrective scribbles during inference. This setting is particularly challenging because tracer distributions, physiological uptake patterns, lesion appearance, and acquisition characteristics differ substantially between FDG and PSMA studies. We present TRIAGE, Tracer-aware Refinement via Interactive Anatomy-Guided sEgmentation. The core backbone is a 3D STU-Net initialized through masked autoencoding pre-training with an asynchronous masking strategy, aiming to learn transferable anatomical and cross-modal representations before task-specific fine-tuning. In parallel, we train an auxiliary organ segmentation model whose predictions provide explicit anatomical context and help distinguish physiological uptake from malignant lesions. A dedicated tracer classifier first routes each study to an FDG- or PSMA-specific branch. Within each branch, a first-stage segmentation model consumes CT, PET, and organ context to generate an initial lesion mask. The initial prediction is then combined with cumulative foreground/background scribbles and refined by a second interactive segmentation network. The FDG and PSMA branches share the same overall processing pipeline but are trained independently to account for tracer-specific appearance and error modes. We additionally employ curriculum-style training and model ensembling to improve robustness across interaction steps and heterogeneous cohorts. Experiments are conducted using the official AutoPET V data and ten-fold split; quantitative results, ablations, and final test-set performance are left as placeholders to be completed after the challenge evaluation. Code: https://github.com/Liiiii2101/AUTOPET2026-MEDAI.
