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

Zhanghao Chen1, Yifei Sun1, Ruiquan Ge1* Wenjian Qin2, Cheng Pan3, Wenming Deng4, Zhou Liu4, Wenwen Min5, Ahmed Elazab6, Xiang Wan7, Changmiao Wang7*
1 Hangzhou Dianzi University, 2 Shenzhen Institute of Advanced Technology, Chinese Academy of Sciences, 3 Sanda University, 4 National Cancer Center/National Clinical Research Center for Cancer/Cancer Hospital&Shenzhen Hospital, Chinese Academy of Medical Sciences and Peking Union Medical College,
5 Yunnan University, 6 Shenzhen University,
7 Shenzhen Research Institute of Big Data
*Corresponding Authors
Accepted by IEEE ISBI 2024

What is bone suppression?

๐Ÿ”ป Try hovering the mouse over the image below and moving it horizontally!

The left side is a Chest X-Ray (CXR) image while the right side is a soft tissue image obtained by bone suppression using our BS-Diff model.

Bone suppression in CXR imaging refers to computational or imaging techniques that minimize or remove the visibility of bone structures from CXR images. The goal is to enhance the clarity of underlying soft tissues, particularly the lungs, to improve diagnostic accuracy for pulmonary conditions.

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.


Method

MY ALT TEXT

The overview framework of our proposed BS-Diff.


Highlights

๐ŸŒŸ To the best of our knowledge, this is the pioneering study that harnesses diffusion models for the generation of soft-tissue images from CXRs, thus addressing and overcoming the prevailing limitations of DES.

๐ŸŒŸ In our enhancement module, we introduce an innovative amalgamation of varied loss functions, devised to more effectively encapsulate spatial features and intricate texture details of images, while concurrently preserving their overall structures.

๐ŸŒŸ We have also assembled the most extensive dataset since 2010, comprising high-resolution paired images from 120 patients, which were collected in collaboration with our partner hospital. We anticipate making this dataset publicly available in the near future.

๐ŸŒŸ Through comprehensive experiments, comparative analyses, ablation studies, and clinical evaluations, we substantiate the superior performance of our proposed BS-Diff model in comparison to several high-performing bone suppression models.


Experiments

MY ALT TEXT

Visulization of different methods comparison on our dataset .

MY ALT TEXT Visualization of ablation studies on BS-Diff with and without enhancement module.

Clinical Evaluation of Image Quality Assessment

MY ALT TEXT

Image quality assessment of BS-Diff scored by three radiologists with various levels of experience.

BibTeX

@inproceedings{chen2024bs,
  title={Bs-diff: Effective bone suppression using conditional diffusion models from chest x-ray images},
  author={Chen, Zhanghao and Sun, Yifei and Ge, Ruiquan and Qin, Wenjian and Pan, Cheng and Deng, Wenming and Liu, Zhou and Min, Wenwen and Elazab, Ahmed and Wan, Xiang and others},
  booktitle={2024 IEEE International Symposium on Biomedical Imaging (ISBI)},
  pages={1--5},
  year={2024},
  organization={IEEE}
}