arXiv Computer Vision

DART: Scaling a Shape Representation Across Anatomical Classes and Imaging Modalities

arXiv Machine Learning
Aug 27

Unsupervised Anatomical Feature Learning via Diffusion Models: Enhanced Medical Image Segmentation with Denoising Diffusion Probabilistic Models

The paper introduces an unsupervised approach to medical image segmentation by training a Denoising Diffusion Probabilistic Model (DDPM) on 21 unlabeled abdominal CT scans to learn anatomical features. The encoder weights from the DDPM are transferred to a U‑Net for downstream segmentation on the BTCV multi‑organ dataset, resulting in a significant Dice score improvement for liver segmentation from 0.75 to 0.93. In low‑data regimes, diffusion‑pretrained models retain robust performance, achieving high Dice scores even with only 10% of labeled data.

By Akshat G, Divyansh Gupta, Shaleen Bhatnagar, Shilpa Ankalaki, Tusar Kanti Mishra
arXiv AI
Aug 25

SAS: Segment Anything Small for Ultrasound -- A Non-Generative Data Augmentation Technique for Robust Deep Learning in Ultrasound Imaging

The paper introduces Segment Anything Small (SAS), a data‑augmentation method that improves deep‑learning segmentation of small anatomical structures in ultrasound images. SAS uses two transformations: resizing and embedding organ thumbnails into a black background to vary organ scale, and adding noise to regions of interest to mimic tissue texture variability. Experiments on one internal and five external datasets show Dice score gains up to 0.35, with an average improvement of 0.16, and demonstrate that SAS enhances model robustness and generalizability without adding hallucinations or artifacts.

By Danielle L. Ferreira, Ahana Gangopadhyay, Hsi-Ming Chang, Ravi Soni, Gopal Avinash
arXiv Computer Vision
Aug 28

DALE-CT: Depth-Aware 2D Slice Encoders Learn an Anatomical World Model of Chest CT

DALE-CT introduces depth‑aware 2D slice encoders that learn an anatomical world model of chest CT scans without 3D or positional supervision. By sampling self‑supervised views across a physical $z$‑axis slab, the encoder captures how anatomy changes between neighboring slices, enabling it to recover slice ordering and distinguish slices by anatomy alone. The model, trained on a large 287k‑scan corpus, achieves state‑of‑the‑art performance on CT‑RATE and is released with full code and evaluation tools.

By Evan W. Damron, Mahmut S. Gokmen, Mitchell A. Klusty, Caroline N. Leach, Emily B. Collier, V. K. Cody Bumgardner
arXiv Computer Vision
Sep 4

Improving Clinical Target Volume Segmentation Accuracy using Anatomical Priors and Active Learning for the AGITG TOPGEAR Clinical Trial

The study explores how adding anatomical priors and active learning can improve the accuracy of deep learning models for segmenting the Clinical Target Volume (CTV) in gastric cancer radiotherapy. Using 100 retrospective CT scans, an nnU‑Net model trained on 10 expert‑contoured cases was enhanced with voxel‑wise anatomical prior maps and iterative active learning over four rounds. The combined approach raised the mean Dice Similarity Coefficient from 0.84 to 0.87, demonstrating that both techniques individually and together improve segmentation performance and generalizability.

By Phillip Chlap, Mark Lee, Trevor Leong, Matthew Field, Jason Dowling, Hang Min, Julie Chu, Jennifer Tan, Phillip K. Tran, Tomas Kron, Annette Haworth, Martin A. Ebert, Shalini K. Vinod, Lois Holloway
arXiv Computer Vision
Aug 31

Anatomy-Aware Promptable Segmentation with Online Interactive Training for AUTOPET V

The paper introduces an anatomy-aware, promptable segmentation model for whole-body lesion detection in FDG and PSMA PET/CT scans, tailored for the AUTOPET V challenge. The approach builds on nnU-Net, employing a two-stage training process: an initial pre-training phase for strong baseline segmentation and an online interactive phase that refines predictions using scribble prompts. Anatomical context is integrated via organ supervision with a shared head predicting both lesions and organs, reducing false positives, while a tracer classifier directs studies to either a combined FDG+PSMA model or a PSMA-specific model. Cross-validation results show that organ-supervised training yields the most stable performance, the interactive stage consistently improves Dice scores, and PSMA-specific training delivers the best tracer-wise results.

By Pablo Lozano-Jimenez, Sergio Romero-Tapiador, Ruben Tolosana
arXiv Machine Learning
Oct 15

Contrastive uncertainty learning for robust multi-structure segmentation in lower-pelvic MR

The paper introduces Contrastive Uncertainty Learning (CUL), a supervised framework that leverages voxel‑wise predictive uncertainty to guide representation learning for multi‑structure segmentation in lower‑pelvic MRI. Using Monte Carlo Dropout, CUL identifies uncertain voxels and shapes the embedding space so that low‑uncertainty, correctly predicted features become stable while high‑uncertainty or misclassified features are repelled, preventing feature collapse at challenging boundaries. Evaluated on 589 prostate MRI scans with a ViT‑based UperNet decoder, CUL achieves comparable Dice scores to a baseline while significantly reducing surface distance for most structures, and ablation studies confirm the necessity of both uncertainty‑guided sampling and the contrastive objective.

By Bilal Sidiqi, Yipei Wang, Wen Yan, Shaheer U. Saeed, Daniel C. Alexander, Yipeng Hu