DADA

<div align="center"> <h1>DADA</h1> <h3>Targeted False Positive Synthesis via Detector-guided Adversarial Diffusion Attacker for Robust Polyp Detection</h3> <br> <a href="http://sa.whut.edu.cn/yjspy/dsdw/202506/t20250613_614174.shtml">Quan Zhou</a>^{<span>1, &#42;</span>}, Gan Luo^{<span>1, &#42;</span>}, <a href="https://scholar.google.com/citations?user=rU2JxLIAAAAJ&hl=en">Qiang Hu</a>^{<span>2, &#8224;</span>}, Qingyong Zhang^{<span>1</span>}, Jinhua Zhang^{<span>3</span>}, Yinjiao Tian^{<span>3</span>}, Qiang Li^{<span>2</span>}, <a href="https://scholar.google.com/citations?user=LwQcmgYAAAAJ&hl=en">Zhiwei Wang</a>^{<span>2, &#8224;</span>} </br>

¹ WUT, ² WNLO, HUST, ³ Changzhou United lmaging Healthcare Surgical Technology Co., Ltd. <br> (<span>*</span>: equal contribution, <span>†</span>: corresponding author)

</div>

Overview

1. Inference Flow

In our work, we adopt a inpainting strategy and the specific inference flow is: Firstly you should have a real endoscopy image; Secondly you need to specify a inpainting region (a bounding box region); and Finally, our method can generate a high-value negative sample in the region you have formulated in this real image. To make it clear, we've made a GIF, which is shown as follows:

<p align="center"> <img src="figs/inference.gif"/ width=800> <br /> </p>

2. Introduction

To tackle the challenge of polyp detectors easy to produce false positive prediction, this work firstly (to the best of our knowledge) to apply adversarial diffusion framwork to generate high-value negative samples (e.g., polyp-like interferences, light spots, circular lumens and and other background objects). The core module is Detector-guided Adversarial Diffusion Attacker (DADA), which perturbs the denoising process to generate challenging false positives capable of confusing a well-trained polyp detector, thereby guiding the synthesis of high-value negative samples. The negative samples synthesized by our method can improve the performance of the detector more than other data data synthesis methods.

<p align="center"> <img src="figs/overview.jpg"/ width=800> <br /> </p>

Experimental Results

1. Comparison with State-of-the-arts

- Quantitative Comparisons

<p align="center"> <img src="figs/fig3.png"/ width=800> <br /> </p>

- Qualitative Comparisons

<p align="center"> <img src="figs/fig2.jpg"/ width=800> <br /> </p>

Quick Start

- Preliminaries

• Python 3.8+
• PyTorch 2.2+
• TorchVision corresponding to the PyTorch version
• NVIDIA GPU + CUDA

1. Install dependencies.

cd DADA
pip install -r requirements.txt

2. Data synthesis

We provide well-trained BG-De weights based on the public Kvasir dataset, which can be downloaded from this link, . Please place the weights in the BG-De_model folder.

Additionally, we also offer YOLOv5l weights trained on the public Kvasir dataset, which can be downloaded from this link. Please place the weights in the Detection_model folder.

  DADA
  ├── BG-De_model
  ├── Detection_model

After completing the above steps, you can generate negative samples by simply running Main.py:

python Main.py

Training

1. Prepare the datasets for BG-De.

Please place the dataset you want to train in the path ./datasets and ensure that the size of the images and masks is 256. The path structure should be as follows:

  DADA
  ├── datasets
  │   ├── images
  │   ├── masks

2.Train your own BG-De.

Please set the "state" parameter of modelConfig in Main.py to "train", and set parameters such as batch_size according to actual conditions.

python Main.py

Place the weights in the BG-De_model folder.

- Testing

Before running the tests, ensure an object detection model is ready. For this project, we utilize the YOLOv5l architecture. Place the weights in the Detection_model folder and set the "state" parameter of modelConfig in Main.py to "eval".

python Main.py

Acknowledgments

We acknowledge the outstanding open-source contributions from DDPM, YOLOv5, and DETR.

Citation

If you find our paper and code useful in your research, please consider giving us a star ⭐ and citing DADA by the following BibTeX entry.

@inproceedings{zhou2025targeted,
  title={Targeted False Positive Synthesis via Detector-Guided Adversarial Diffusion Attacker for Robust Polyp Detection},
  author={Zhou, Quan and Luo, Gan and Hu, Qiang and Zhang, Qingyong and Zhang, Jinhua and Tian, Yinjiao and Li, Qiang and Wang, Zhiwei},
  booktitle={International Conference on Medical Image Computing and Computer-Assisted Intervention},
  pages={593--602},
  year={2025},
  organization={Springer}
}