BoxShrink: From Bounding Boxes to Segmentation Masks

Michael Gröger, Vadim Borisov, Gjergji Kasneci

Research output: Chapter in Book/Report/Conference proceedingConference contributionpeer-review

5 Scopus citations

Abstract

One of the core challenges facing the medical image computing community is fast and efficient data sample labeling. Obtaining fine-grained labels for segmentation is particularly demanding since it is expensive, time-consuming, and requires sophisticated tools. On the contrary, applying bounding boxes is fast and takes significantly less time than fine-grained labeling, but does not produce detailed results. In response, we propose a novel framework for weakly-supervised tasks with the rapid and robust transformation of bounding boxes into segmentation masks without training any machine learning model, coined BoxShrink. The proposed framework comes in two variants – rapid-BoxShrink for fast label transformations, and robust-BoxShrink for more precise label transformations. An average of four percent improvement in IoU is found across several models when being trained using BoxShrink in a weakly-supervised setting, compared to using only bounding box annotations as inputs on a colonoscopy image data set. We open-sourced the code for the proposed framework and published it online.

Original languageEnglish
Title of host publicationMedical Image Learning with Limited and Noisy Data - 1st International Workshop, MILLanD 2022, Held in Conjunction with MICCAI 2022, Proceedings
EditorsGhada Zamzmi, Sameer Antani, Sivaramakrishnan Rajaraman, Zhiyun Xue, Ulas Bagci, Marius George Linguraru
PublisherSpringer Science and Business Media Deutschland GmbH
Pages65-75
Number of pages11
ISBN (Print)9783031167591
DOIs
StatePublished - 2022
Externally publishedYes
Event1st International Workshop on Medical Image Learning with Limited and Noisy Data, MILLanD 2022, held in conjunction with 25th International Conference on Medical Image Computing and Computer Assisted Intervention, MICCAI 2022 - Singapore, Singapore
Duration: 22 Sep 202222 Sep 2022

Publication series

NameLecture Notes in Computer Science (including subseries Lecture Notes in Artificial Intelligence and Lecture Notes in Bioinformatics)
Volume13559 LNCS
ISSN (Print)0302-9743
ISSN (Electronic)1611-3349

Conference

Conference1st International Workshop on Medical Image Learning with Limited and Noisy Data, MILLanD 2022, held in conjunction with 25th International Conference on Medical Image Computing and Computer Assisted Intervention, MICCAI 2022
Country/TerritorySingapore
CitySingapore
Period22/09/2222/09/22

Keywords

  • Colonoscopy
  • Deep neural networks
  • Segmentation
  • Weakly-supervised learning

Fingerprint

Dive into the research topics of 'BoxShrink: From Bounding Boxes to Segmentation Masks'. Together they form a unique fingerprint.

Cite this