loading page

PIXIE: A Novel Loss Function for Binary Semantic Segmentation
  • +4
  • Vansh Jain ,
  • Guruprasad Parasnis ,
  • Aaryan Purohit ,
  • Saher Thekedar ,
  • Anmol Chokshi ,
  • Kailas Devadkar ,
  • Pramod Bide
Vansh Jain
Author Profile
Guruprasad Parasnis
Sardar Patel Institute of Technology

Corresponding Author:[email protected]

Author Profile
Aaryan Purohit
Author Profile
Saher Thekedar
Author Profile
Anmol Chokshi
Author Profile
Kailas Devadkar
Author Profile
Pramod Bide
Author Profile

Abstract

This paper introduces PIXIE (Penalised Pixel Intersection Error), a novel loss function aimed at improving model performance through the penalization of error pixels and alignment of the predicted mask with the ground truth mask. We explore the effectiveness of PIXIE alongside U-Net and DeepLabV3 networks. In addition to novel loss functions, this paper conducts model selection studies to understand overestimations and underestimations. We conduct a comprehensive evaluation using diverse datasets spanning medical imaging (breast tumor, COVID-19, Brain MRI) and remote sensing domains (forest fire, water bodies satellite images). Performance metrics, such as Intersection over Union (IoU), Dice Coefficient (DC), Area Error Ratio (AER), precision, and recall are quantitatively assessed. Moreover, we compare PIXIE with traditional loss functions like Jacard loss, Focal loss, and Binary crossentropy. PIXIE demonstrates comparable performance to the traditional loss functions in certain metrics and outperforms them in others, establishing itself as a leading approach in achieving exceptional results across all examined segmentation datasets. These findings represent a substantial contribution to the practical evolution of semantic segmentation in computer vision, offering essential insights into the optimization of loss functions for the development of accurate and robust models.