Paired Image Generation with Diffusion-Guided Diffusion Models

TL;DR

This paper introduces a paired image generation method using diffusion-guided models to produce high-quality paired images and masks, improving data augmentation for breast cancer lesion segmentation in DBT images.

Contribution

It proposes a novel paired image generation approach with diffusion models that does not require external conditions, enhancing data quality and annotation availability.

Findings

Improved generation quality of lesion images and masks.

Enhanced segmentation performance with augmented data.

Alleviated annotation scarcity in breast cancer imaging.

Abstract

The segmentation of mass lesions in digital breast tomosynthesis (DBT) images is very significant for the early screening of breast cancer. However, the high-density breast tissue often leads to high concealment of the mass lesions, which makes manual annotation difficult and time-consuming. As a result, there is a lack of annotated data for model training. Diffusion models are commonly used for data augmentation, but the existing methods face two challenges. First, due to the high concealment of lesions, it is difficult for the model to learn the features of the lesion area. This leads to the low generation quality of the lesion areas, thus limiting the quality of the generated images. Second, existing methods can only generate images and cannot generate corresponding annotations, which restricts the usability of the generated images in supervised training. In this work, we propose a paired image generation method. The method does not require external conditions and can achieve the generation of paired images by training an extra diffusion guider for the conditional diffusion model. During the experimental phase, we generated paired DBT slices and mass lesion masks. Then, we incorporated them into the supervised training process of the mass lesion segmentation task. The experimental results show that our method can improve the generation quality without external conditions. Moreover, it contributes to alleviating the shortage of annotated data, thus enhancing the performance of downstream tasks. The source code is available at https://github.com/zhanghx1320/PIG.