BS-Diff: Effective Bone Suppression Using Conditional Diffusion Models from Chest X-Ray Images

Zhanghao Chen,Yifei Sun,Wenjian Qin,Ruiquan Ge,Cheng Pan,Wenming Deng,Zhou Liu,Wenwen Min,Ahmed Elazab,Xiang Wan,Changmiao Wang
2024-02-29
Abstract:Chest X-rays (CXRs) are commonly utilized as a low-dose modality for lung screening. Nonetheless, the efficacy of CXRs is somewhat impeded, given that approximately 75% of the lung area overlaps with bone, which in turn hampers the detection and diagnosis of diseases. As a remedial measure, bone suppression techniques have been introduced. The current dual-energy subtraction imaging technique in the clinic requires costly equipment and subjects being exposed to high radiation. To circumvent these issues, deep learning-based image generation algorithms have been proposed. However, existing methods fall short in terms of producing high-quality images and capturing texture details, particularly with pulmonary vessels. To address these issues, this paper proposes a new bone suppression framework, termed BS-Diff, that comprises a conditional diffusion model equipped with a U-Net architecture and a simple enhancement module to incorporate an autoencoder. Our proposed network cannot only generate soft tissue images with a high bone suppression rate but also possesses the capability to capture fine image details. Additionally, we compiled the largest dataset since 2010, including data from 120 patients with high-definition, high-resolution paired CXRs and soft tissue images collected by our affiliated hospital. Extensive experiments, comparative analyses, ablation studies, and clinical evaluations indicate that the proposed BS-Diff outperforms several bone-suppression models across multiple metrics. Our code can be accessed at <a class="link-external link-https" href="https://github.com/Benny0323/BS-Diff" rel="external noopener nofollow">this https URL</a>.
Image and Video Processing,Computer Vision and Pattern Recognition
What problem does this paper attempt to address?
The problem that this paper attempts to solve is to achieve effective bone suppression in chest X - ray images. Specifically, since approximately 75% of the lung area overlaps with bones in X - ray films, this hinders the detection and diagnosis of diseases. Although the existing dual - energy subtraction imaging technology is effective, it requires expensive equipment and exposes patients to a relatively high radiation dose. In addition, the existing deep - learning - based methods have deficiencies in generating high - quality images and capturing texture details, especially when dealing with pulmonary blood vessels. To overcome these problems, the paper proposes a new bone suppression framework BS - Diff, which combines the Conditional Diffusion Model (CDM) and a simple enhancement module, aiming to generate high - quality soft - tissue images with a high bone suppression rate and the ability to capture subtle image details.