English

Multitask Brain Tumor Inpainting with Diffusion Models: A Methodological Report

Image and Video Processing 2023-04-03 v2 Computer Vision and Pattern Recognition Machine Learning

Abstract

Despite the ever-increasing interest in applying deep learning (DL) models to medical imaging, the typical scarcity and imbalance of medical datasets can severely impact the performance of DL models. The generation of synthetic data that might be freely shared without compromising patient privacy is a well-known technique for addressing these difficulties. Inpainting algorithms are a subset of DL generative models that can alter one or more regions of an input image while matching its surrounding context and, in certain cases, non-imaging input conditions. Although the majority of inpainting techniques for medical imaging data use generative adversarial networks (GANs), the performance of these algorithms is frequently suboptimal due to their limited output variety, a problem that is already well-known for GANs. Denoising diffusion probabilistic models (DDPMs) are a recently introduced family of generative networks that can generate results of comparable quality to GANs, but with diverse outputs. In this paper, we describe a DDPM to execute multiple inpainting tasks on 2D axial slices of brain MRI with various sequences, and present proof-of-concept examples of its performance in a variety of evaluation scenarios. Our model and a public online interface to try our tool are available at: https://github.com/Mayo-Radiology-Informatics-Lab/MBTI

Keywords

Cite

@article{arxiv.2210.12113,
  title  = {Multitask Brain Tumor Inpainting with Diffusion Models: A Methodological Report},
  author = {Pouria Rouzrokh and Bardia Khosravi and Shahriar Faghani and Mana Moassefi and Sanaz Vahdati and Bradley J. Erickson},
  journal= {arXiv preprint arXiv:2210.12113},
  year   = {2023}
}

Comments

17 pages, 7 figures

R2 v1 2026-06-28T04:12:09.713Z