Compound Loss Function of semantic segmentation models for imbalanced construction data

  • Chern, Wei-Chih (Department of Electrical and Computer Engineering, University of Dayton) ;
  • Kim, Hongjo (Department of Civil and Environmental Engineering, Yonsei University) ;
  • Asari, Vijayan (Department of Electrical and Computer Engineering, University of Dayton) ;
  • Nguyen, Tam (Department of Computer Science, University of Dayton)
  • Published : 2022.06.20

Abstract

This study presents the problems of data imbalance, varying difficulties across target objects, and small objects in construction object segmentation for far-field monitoring and utilize compound loss functions to address it. Construction site scenes of assembling scaffolds were analyzed to test the effectiveness of compound loss functions for five construction object classes---workers, hardhats, harnesses, straps, hooks. The challenging problem was mitigated by employing a focal and Jaccard loss terms in the original loss function of LinkNet segmentation model. The findings indicates the importance of the loss function design for model performance on construction site scenes for far-field monitoring.

Keywords

Acknowledgement

This research was conducted with the support of the "2021 Yonsei University Future-Leading Research Initiative (No.2021-22-0037)" and the "National RD Project for Smart Construction Technology (No.22SMIP-A158708-03)" funded by the Korea Agency for Infrastructure Technology Advancement under the Ministry of Land, Infrastructure and Transport, and managed by the Korea Expressway Corporation.