English

PolypConnect: Image inpainting for generating realistic gastrointestinal tract images with polyps

Image and Video Processing 2022-06-01 v1 Computer Vision and Pattern Recognition Machine Learning

Abstract

Early identification of a polyp in the lower gastrointestinal (GI) tract can lead to prevention of life-threatening colorectal cancer. Developing computer-aided diagnosis (CAD) systems to detect polyps can improve detection accuracy and efficiency and save the time of the domain experts called endoscopists. Lack of annotated data is a common challenge when building CAD systems. Generating synthetic medical data is an active research area to overcome the problem of having relatively few true positive cases in the medical domain. To be able to efficiently train machine learning (ML) models, which are the core of CAD systems, a considerable amount of data should be used. In this respect, we propose the PolypConnect pipeline, which can convert non-polyp images into polyp images to increase the size of training datasets for training. We present the whole pipeline with quantitative and qualitative evaluations involving endoscopists. The polyp segmentation model trained using synthetic data, and real data shows a 5.1% improvement of mean intersection over union (mIOU), compared to the model trained only using real data. The codes of all the experiments are available on GitHub to reproduce the results.

Keywords

Cite

@article{arxiv.2205.15413,
  title  = {PolypConnect: Image inpainting for generating realistic gastrointestinal tract images with polyps},
  author = {Jan Andre Fagereng and Vajira Thambawita and Andrea M. Storås and Sravanthi Parasa and Thomas de Lange and Pål Halvorsen and Michael A. Riegler},
  journal= {arXiv preprint arXiv:2205.15413},
  year   = {2022}
}

Comments

6 pages