Corrosion segmentation based on the UNet architecture and the impact of loss functions

Abstract

Corrosion is a natural phenomenon that causes significant material and economic losses. Timely inspection of corroded assets is vital for preventing such losses. Visual inspection is a nondestructive method that is commonly used for corrosion detection. The ability of deep learning methods to capture corrosion visual information allows for the automation of the manual inspection process. The literature includes several examples of deep learning models to detect metal corrosion. In developing such models, loss functions are crucial and directly influence the models' performance. Although cross-entropy is a commonly used objective function for corrosion detection, systematic evaluation of alternative loss functions-and the performance gains they may unlock-has been largely neglected. This study uses eight loss functions, namely, cross-entropy loss, Focal loss, Unified Focal loss, Dice loss, Tversky loss, Focal Tversky loss, Combo loss, and Hybrid Focal loss, to train UNet-based models for corrosion segmentation and compares their performance. The results demonstrated that the model trained with the Dice loss achieved higher performance, with a mean Jaccard index value of 64.28%, while the Focal loss exhibited the lowest performance, with a mean Jaccard value of 57.20%. This study contributes to academia and practice by demonstrating the performance gains that can be achieved by tweaking the deep learning models using optimal loss functions.