Neural Network Pruning for Lightweight Metal Corrosion Image Segmentation Models

Abstract

Metal corrosion detection is essential for ensuring structural safety and minimizing economic losses. While deep learning (DL)-based image segmentation has improved corrosion detection accuracy and efficiency, its high computational demands hinder deployment on resource-constrained edge devices. This study investigates lightweight DL models for corrosion segmentation by applying structured pruning to reduce computational costs while maintaining accuracy. We evaluate five segmentation architectures (U-Net, U-Net++, FPN, LinkNet, and MA-Net) and three pruning strategies (linear, automated gradual pruning, and movement pruning) on two corrosion image datasets (NEA and SSCS). Detailed trade-off analysis between model size, computational cost (MAC), and segmentation performance (IoU) reveals that pruning up to 90% sparsity leads to a $\leq 10\%$ IoU drop on SSCS and $\leq 5\%$ on NEA, demonstrating that significant compression is possible with minimal accuracy loss. However, some architectures (e.g., LinkNet) and pruning strategies (e.g., movement pruning) show significant performance deterioration, suggesting that pruning effectiveness varies across models. These findings provide insights into optimizing corrosion segmentation models for efficient deployment on edge devices, balancing accuracy and resource constraints.