CPW-DICE: a novel center and pixel-based weighting for damage segmentation

Abstract

Reliable evaluation of damage in vehicles is a primary concern in the insurance industry. Consequently, solutions enhanced with Artificial Intelligence (AI) have become the norm. During the assessment, precise damage segmentation plays a crucial role. Dent is a type of damage that can commonly occur in vehicles. It is difficult to pinpoint and tends to blend in with the background. This paper proposes a novel loss function to improve dent segmentation accuracy in vehicle insurance claims. Centre and Pixel-based Weighted DICE (CPW-DICE) is a loss function that performs pixel-based weighting. The CPW-DICE aims to concentrate on the centre of the dent damage to lessen faulty segmentations. CPW-DICE generates a weight mask during training by employing ground truth (GT) and prediction masks. Simultaneously, the weight mask is incorporated into DICE loss. Experiments conducted on our comprehensive internal dataset show a 3% improvement in Intersection over Union (IoU) score for three state-of-the-art (SOTA) approaches compared to DICE loss. Finally, CPW-DICE is evaluated in similar tasks to demonstrate its benefits beyond car damage segmentation.