Track papers across labs, topics, and trends.
Search papers, labs, and topics across Lattice.
This paper introduces PU-UNet, a novel residual U-Net architecture that incorporates stable product-unit residual blocks to enhance medical image segmentation through explicit multiplicative feature modeling. By employing a combination of smooth positivity mapping and log-domain clipping, PU-UNet achieves significant improvements in segmentation accuracy, as evidenced by Dice scores of 0.942, 0.959, and 0.925 on benchmark datasets, while maintaining computational efficiency. The findings indicate that the integration of stable product-unit interactions is particularly beneficial in low-resolution stages, leading to a reduction in false-positive rates and improved performance over traditional U-Net variants.
Stable product-unit interactions in PU-UNet lead to a remarkable reduction in false-positive rates and significant improvements in segmentation accuracy without increasing computational costs.
Many dense prediction networks rely on additive feature transformations and model higher-order feature interactions only implicitly. Product units provide an explicit mechanism for multiplicative feature modeling, but their logarithmic--exponential formulation can cause numerical instability, which has limited their use in deep dense prediction networks. In this work, we propose Product-Unit U-Net (PU-UNet), a residual U-Net that integrates stable product-unit residual blocks into rich low-resolution stages for medical image segmentation. The proposed formulation combines smooth positivity mapping with log-domain clipping, enabling stable multiplicative feature learning with negligible computational overhead. On ISIC 2018, Kvasir-SEG, and BUSI, PU-UNet achieves Dice scores of 0.942, 0.959, and up to 0.925, respectively. Compared with a matched Residual U-Net baseline, PU-UNet consistently improves Dice and IoU while keeping parameters, FLOPs, and inference latency nearly unchanged, and reduces the image-level false-positive rate on normal BUSI cases from 0.077 to zero. Ablation studies suggest that the gains are associated with product-unit interactions, are strongest under low-resolution placement, and benefit from the proposed stabilization design. These results suggest that stable product-unit residual learning can be an effective way to enhance U-Net-style segmentation networks with explicit multiplicative interactions.
TU Munich