Aberrant Crypt Foci (ACFs) are considered early markers of colorectal neoplasms, yet their segmentation remains challenging due to limited annotated data and large lesion variability. We propose a novel diffusion-based synthetic augmentation framework to generate realistic ACF images under diverse artefact conditions and shapes. By conditioning a denoising diffusion probabilistic model on multi-class masks (blood, dye, and waste), and depth cues, our method expands both morphological and contextual variability in small ACF datasets. We evaluate performance gains on five segmentation architectures: three convolutional networks (U-Net, U-Net++, and DeepLab V3+) and two models with transformer backbones (TransNetR, PVT CASCADE). In our experiments, the less complex CNN models achieve the most substantial boosts in test Dice (e.g., +22.1% for U-Net), while DeepLab V3+ sees an +8.4% gain. The transformer-backbone architectures also benefit, with improvements of +0.6% for TransNetR and +2.7% for PVT-CASCADE. Qualitative assessments confirm that the diffusion-generated images replicate the annotated artefacts and reduce the performance drop that typically occurs when training with limited real data. Our results indicate that CtrlEndoDiff improves ACF segmentation accuracy by adding visually realistic synthetic samples into the training process.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

CtrlEndoDiff: Diffusion-Based Synthetic Image Generation for Enhanced ACF Segmentation

  • Rayan Albokhari,
  • Bappaditya Debnath,
  • Darrien Henry,
  • Michael Johnson,
  • Adam Bailey,
  • James E. East,
  • Jens Rittscher

摘要

Aberrant Crypt Foci (ACFs) are considered early markers of colorectal neoplasms, yet their segmentation remains challenging due to limited annotated data and large lesion variability. We propose a novel diffusion-based synthetic augmentation framework to generate realistic ACF images under diverse artefact conditions and shapes. By conditioning a denoising diffusion probabilistic model on multi-class masks (blood, dye, and waste), and depth cues, our method expands both morphological and contextual variability in small ACF datasets. We evaluate performance gains on five segmentation architectures: three convolutional networks (U-Net, U-Net++, and DeepLab V3+) and two models with transformer backbones (TransNetR, PVT CASCADE). In our experiments, the less complex CNN models achieve the most substantial boosts in test Dice (e.g., +22.1% for U-Net), while DeepLab V3+ sees an +8.4% gain. The transformer-backbone architectures also benefit, with improvements of +0.6% for TransNetR and +2.7% for PVT-CASCADE. Qualitative assessments confirm that the diffusion-generated images replicate the annotated artefacts and reduce the performance drop that typically occurs when training with limited real data. Our results indicate that CtrlEndoDiff improves ACF segmentation accuracy by adding visually realistic synthetic samples into the training process.