Spatially regularized Leaky ReLU in dual space for CNN based image segmentation

Xiangyue Wang,Jun Liu
DOI: https://doi.org/10.3934/ipi.2024016
2024-04-18
Inverse Problems and Imaging
Abstract:The ReLU/Leaky-ReLU activation functions are widely used in convolutional neural network (CNN) architectures due to their simple implementation and outstanding performance. However, the current ReLU/Leaky-ReLU activation functions lack spatial priors among pixels during feature extraction. As a result, they may struggle to capture the critical piecewise semantic features that are essential for image segmentation. Drawing inspiration from the regional characteristics of human brain neurons, this paper presents a novel spatially regularized ReLU/Leaky-ReLU module for CNN-based image segmentation. To enhance the extraction of piecewise image features, we introduce a spatial regularization mechanism for ReLU/Leaky-ReLU activation. This is achieved by reinterpreting the activation function as a variational problem with a dual formulation. The proposed variational framework enables the integration of various successful variational priors, such as local and nonlocal spatial regularization, into the activation function design. For efficient regularization in ReLU/Leaky-ReLU, we develop a local and nonlocal threshold dynamics regularizer using difference of concave optimization. This novel activation module facilitates CNNs in extracting crucial piecewise and smooth deep features for image segmentation, while partially mitigating the influence of noise during the CNN test procedure. It provides a means of integrating traditional methods, such as the nonlocal regularization, into ReLU activation in CNNs. The effectiveness of the proposed activation module is demonstrated in the experimental results. Compared to baseline segmentation CNNs, the proposed approach yields a higher segmentation quality index (mIoU) and exhibits better generalization ability to noise.
mathematics, applied,physics, mathematical
What problem does this paper attempt to address?