English

Generating large labeled data sets for laparoscopic image processing tasks using unpaired image-to-image translation

Machine Learning 2019-07-08 v1 Computer Vision and Pattern Recognition Machine Learning

Abstract

In the medical domain, the lack of large training data sets and benchmarks is often a limiting factor for training deep neural networks. In contrast to expensive manual labeling, computer simulations can generate large and fully labeled data sets with a minimum of manual effort. However, models that are trained on simulated data usually do not translate well to real scenarios. To bridge the domain gap between simulated and real laparoscopic images, we exploit recent advances in unpaired image-to-image translation. We extent an image-to-image translation method to generate a diverse multitude of realistically looking synthetic images based on images from a simple laparoscopy simulation. By incorporating means to ensure that the image content is preserved during the translation process, we ensure that the labels given for the simulated images remain valid for their realistically looking translations. This way, we are able to generate a large, fully labeled synthetic data set of laparoscopic images with realistic appearance. We show that this data set can be used to train models for the task of liver segmentation of laparoscopic images. We achieve average dice scores of up to 0.89 in some patients without manually labeling a single laparoscopic image and show that using our synthetic data to pre-train models can greatly improve their performance. The synthetic data set will be made publicly available, fully labeled with segmentation maps, depth maps, normal maps, and positions of tools and camera (http://opencas.dkfz.de/image2image).

Keywords

Cite

@article{arxiv.1907.02882,
  title  = {Generating large labeled data sets for laparoscopic image processing tasks using unpaired image-to-image translation},
  author = {Micha Pfeiffer and Isabel Funke and Maria R. Robu and Sebastian Bodenstedt and Leon Strenger and Sandy Engelhardt and Tobias Roß and Matthew J. Clarkson and Kurinchi Gurusamy and Brian R. Davidson and Lena Maier-Hein and Carina Riediger and Thilo Welsch and Jürgen Weitz and Stefanie Speidel},
  journal= {arXiv preprint arXiv:1907.02882},
  year   = {2019}
}

Comments

Accepted at MICCAI 2019

R2 v1 2026-06-23T10:13:18.990Z