Skip to main navigation Skip to search Skip to main content

Self Supervised Cerebrovascular Segmentation Using Shared Weight DDPM UNet with Adversarial Representation Learning

  • Dongguk University

Research output: Chapter in Book/Report/Conference proceeding › Conference contribution › peer-review

Abstract

Dense voxel-wise annotation for Time-of-Flight Magnetic Resonance Angiography (TOF-MRA) vessel segmentation is costly and often infeasible. We present a selfsupervised Denoising Diffusion Probabilistic Model (DDPM) UNet that eliminates the need for manual labels. A single shared encoder feeds three decoder heads-background inpainting, vessel segmentation, and angiogram synthesis-trained jointly via adversarial cycle-consistency on pseudo-masks generated by intensity thresholding. Pretrained on the Lausanne TOF-MRA Aneurysm Cohort and evaluated on the ADAM subset of COSTA, our model achieves Dice Similarity Coefficient (DSC) and Intersection Over Unions (IoU) scores of 0.6755 and 0.5286, respectively, rivalling fully supervised baselines. When fine-tuned with only one labeled slice, it attains a Dice of 0.6379 versus 0.3701 for a randomly initialized U-Net. This approach addresses the scarcity of annotated TOF-MRA datasets and paves the way for volumetric 3D diffusion extensions and task-specific attention modules to further enhance segmentation fidelity.

Original languageEnglish
Title of host publication2025 5th International Conference on Robotics, Automation, and Artificial Intelligence, RAAI 2025
PublisherInstitute of Electrical and Electronics Engineers Inc.
Pages267-272
Number of pages6
ISBN (Electronic)9798331558734
DOIs
StatePublished - 2025
Event2025 5th International Conference on Robotics, Automation, and Artificial Intelligence, RAAI 2025 - Singapore, Singapore
Duration: 18 Dec 2025 → 20 Dec 2025

Publication series

Name2025 5th International Conference on Robotics, Automation, and Artificial Intelligence, RAAI 2025

Conference

Conference2025 5th International Conference on Robotics, Automation, and Artificial Intelligence, RAAI 2025
Country/TerritorySingapore
CitySingapore
Period18/12/25 → 20/12/25

Keywords

  • adversarial network
  • cerebrovascular segmentation
  • DDPM
  • diffusion
  • self-supervised learning
  • vessel segmentation

Fingerprint

Dive into the research topics of 'Self Supervised Cerebrovascular Segmentation Using Shared Weight DDPM UNet with Adversarial Representation Learning'. Together they form a unique fingerprint.

Cite this