Enhancing Polyp Segmentation Generalizability by Minimizing Images' Total Variation

Mahmood Haithami*, Amr Ahmed, Iman Yi Liao*, Hamid Jalab

*Corresponding author for this work

Research output: Chapter in Book/Report/Conference proceedingConference proceeding (ISBN)peer-review

Abstract

Polyps are considered a precursor of colon cancer and early detection of polyps may help decrease mortality rate. Several deep learning models have been proposed to address the problem, however, with limited generalizability due to the scarcity of the current public datasets. To tackle the issue, researchers typically use data augmentation techniques or generative models to inflate training images, independent of a downstream learning task. In this paper, we propose a deep learning framework to jointly train an image transformation model with a segmentation model where the output of the former is the input of the latter. During training, the image transformation model generates variations of the input image at every epoch, implicitly increasing the training data size for the segmentation model. On the other hand, we design a total variational denoising cost for the image transformation model, which effectively ensures that a transformation applied to an input image works towards the segmentation and not any other random effects which may hurt the segmentation goal. The experimental results with different settings demonstrate that the proposed framework has consistently shown an improvement of approximately 1% to 10% polyp IoU on unseen test images.

Original languageEnglish
Title of host publication2023 IEEE International Symposium on Biomedical Imaging, ISBI 2023
PublisherIEEE Computer Society
Pages1-5
ISBN (Electronic)9781665473583
DOIs
Publication statusPublished - 1 Sept 2023
Event20th IEEE International Symposium on Biomedical Imaging, ISBI 2023 - Cartagena, Colombia
Duration: 18 Apr 202321 Apr 2023

Publication series

NameProceedings - International Symposium on Biomedical Imaging
Volume2023-April
ISSN (Print)1945-7928
ISSN (Electronic)1945-8452

Conference

Conference20th IEEE International Symposium on Biomedical Imaging, ISBI 2023
Country/TerritoryColombia
CityCartagena
Period18/04/2321/04/23

Keywords

  • generalizability
  • image transformation
  • Polyp
  • segmentation
  • total variation

Fingerprint

Dive into the research topics of 'Enhancing Polyp Segmentation Generalizability by Minimizing Images' Total Variation'. Together they form a unique fingerprint.

Cite this