Skip to main navigation Skip to search Skip to main content

Increasing the Robustness of Deep Learning Models for Object Segmentation: A Framework for Blending Automatically Annotated Real and Synthetic Data

  • Artur Istvan Karoly
  • , Sebestyen Tirczka
  • , Huijun Gao
  • , Imre J. Rudas
  • , Peter Galambos*
  • *Corresponding author for this work
  • Óbuda University

Research output: Contribution to journalArticlepeer-review

Abstract

Recent problems in robotics can sometimes only be tackled using machine learning technologies, particularly those that utilize deep learning (DL) with transfer learning. Transfer learning takes advantage of pretrained models, which are later fine-tuned using smaller task-specific datasets. The fine-tuned models must be robust against changes in environmental factors such as illumination since, often, there is no guarantee for them to be constant. Although synthetic data for pretraining has been shown to enhance DL model generalization, there is limited research on its application for fine-tuning. One limiting factor is that the generation and annotation of synthetic datasets can be cumbersome and impractical for the purpose of fine-tuning. To address this issue, we propose two methods for automatically generating annotated image datasets for object segmentation, one for real-world and another for synthetic images. We also introduce a novel domain adaptation approach called filling the reality gap (FTRG), which can blend elements from real-world and synthetic scenes in a single image to achieve domain adaptation. We demonstrate through experimentation on a representative robot application that FTRG outperforms other domain adaptation techniques, such as domain randomization or photorealistic synthetic images, in creating robust models. Furthermore, we evaluate the benefits of using synthetic data for fine-tuning in transfer learning and continual learning with experience replay using our proposed methods and FTRG. Our findings indicate that fine-tuning with synthetic data can produce superior results compared to solely using real-world data.

Original languageEnglish
Pages (from-to)25-38
Number of pages14
JournalIEEE Transactions on Cybernetics
Volume54
Issue number1
DOIs
StatePublished - 1 Jan 2024

Keywords

  • Data annotation
  • deep learning (DL)
  • digital twin
  • mixed reality
  • robotics
  • synthetic data

Fingerprint

Dive into the research topics of 'Increasing the Robustness of Deep Learning Models for Object Segmentation: A Framework for Blending Automatically Annotated Real and Synthetic Data'. Together they form a unique fingerprint.

Cite this