vix.ing · top · new · best · stats · spec

Improving Cone-Beam CT Image Quality with Knowledge Distillation-Enhanced Diffusion Model in Imbalanced Data Settings

2024/09/19 by Joonil Hwang, Sangjoon Park, Hwang, Joonil +6
Engineering · Medicine · #Advanced X-ray and CT Imaging #Computer Vision and Pattern Recognition (cs.CV) #Digital Radiography and Breast Imaging #FOS: Computer and information sciences #Medical Imaging Techniques and Applications

paper · pdf · doi:10.48550/arxiv.2409.12539

openalex publication_date 2024/09/19 · openalex created_date 2024/10/25 · openalex updated_date 2026/07/28

Abstract

In radiation therapy (RT), the reliance on pre-treatment computed tomography (CT) images encounter challenges due to anatomical changes, necessitating adaptive planning. Daily cone-beam CT (CBCT) imaging, pivotal for therapy adjustment, falls short in tissue density accuracy. To address this, our innovative approach integrates diffusion models for CT image generation, offering precise control over data synthesis. Leveraging a self-training method with knowledge distillation, we maximize CBCT data during therapy, complemented by sparse paired fan-beam CTs. This strategy, incorporated into state-of-the-art diffusion-based models, surpasses conventional methods like Pix2pix and CycleGAN. A meticulously curated dataset of 2800 paired CBCT and CT scans, supplemented by 4200 CBCT scans, undergoes preprocessing and teacher model training, including the Brownian Bridge Diffusion Model (BBDM). Pseudo-label CT images are generated, resulting in a dataset combining 5600 CT images with corresponding CBCT images. Thorough evaluation using MSE, SSIM, PSNR and LPIPS demonstrates superior performance against Pix2pix and CycleGAN. Our approach shows promise in generating high-quality CT images from CBCT scans in RT.

Related