SegDT: A Diffusion Transformer-Based Segmentation Model for Medical Imaging

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Bekhouche, Salah Eddine, Maroun, Gaby, Dornaika, Fadi, Hadid, Abdenour
Formato: Preprint
Publicado: 2025
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866912494645149696
author Bekhouche, Salah Eddine
Maroun, Gaby
Dornaika, Fadi
Hadid, Abdenour
author_facet Bekhouche, Salah Eddine
Maroun, Gaby
Dornaika, Fadi
Hadid, Abdenour
contents Medical image segmentation is crucial for many healthcare tasks, including disease diagnosis and treatment planning. One key area is the segmentation of skin lesions, which is vital for diagnosing skin cancer and monitoring patients. In this context, this paper introduces SegDT, a new segmentation model based on diffusion transformer (DiT). SegDT is designed to work on low-cost hardware and incorporates Rectified Flow, which improves the generation quality at reduced inference steps and maintains the flexibility of standard diffusion models. Our method is evaluated on three benchmarking datasets and compared against several existing works, achieving state-of-the-art results while maintaining fast inference speeds. This makes the proposed model appealing for real-world medical applications. This work advances the performance and capabilities of deep learning models in medical image analysis, enabling faster, more accurate diagnostic tools for healthcare professionals. The code is made publicly available at \href{https://github.com/Bekhouche/SegDT}{GitHub}.
format Preprint
id arxiv_https___arxiv_org_abs_2507_15595
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle SegDT: A Diffusion Transformer-Based Segmentation Model for Medical Imaging
Bekhouche, Salah Eddine
Maroun, Gaby
Dornaika, Fadi
Hadid, Abdenour
Computer Vision and Pattern Recognition
Medical image segmentation is crucial for many healthcare tasks, including disease diagnosis and treatment planning. One key area is the segmentation of skin lesions, which is vital for diagnosing skin cancer and monitoring patients. In this context, this paper introduces SegDT, a new segmentation model based on diffusion transformer (DiT). SegDT is designed to work on low-cost hardware and incorporates Rectified Flow, which improves the generation quality at reduced inference steps and maintains the flexibility of standard diffusion models. Our method is evaluated on three benchmarking datasets and compared against several existing works, achieving state-of-the-art results while maintaining fast inference speeds. This makes the proposed model appealing for real-world medical applications. This work advances the performance and capabilities of deep learning models in medical image analysis, enabling faster, more accurate diagnostic tools for healthcare professionals. The code is made publicly available at \href{https://github.com/Bekhouche/SegDT}{GitHub}.
title SegDT: A Diffusion Transformer-Based Segmentation Model for Medical Imaging
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2507.15595