PUNet: a lightweight parallel U-Net architecture integrating Mamba–CNN for high-precision image segmentation
Real-time high-precision image segmentation on mobile and edge devices remains challenging due to the limited ability of traditional convolutional networks to model long-range dependencies and their high computational cost at high resolutions. We propose PUNet, a lightweight parallel U-Net variant that integrates depthwise separable convolutions (DSConv) with a structured state-space Mamba module in a dual-path encoder. The core component, the parallel structured state-space encoder, employs two branches to efficiently capture local spatial features (via DSConv) and model global semantic dependencies (via the visual Mamba layer). At the same time, a squeeze-and-excitation skip connection adaptively fuses multi-scale features. With only 0.26 M parameters and linear computational complexity, PUNet enables real-time inference on resource-constrained platforms. Experiments on the CamVid and CRACK500 datasets demonstrate superior performance, achieving validation Dice scores of 0.9208 and 0.7902, and mean Intersection-over-Union of 0.8643 and 0.6612, respectively, significantly outperforming other lightweight models.
Authors
- Xiaowei Li (ORCID: https://orcid.org/0000-0002-6548-9666)
- Zhaoyan Xie (ORCID: https://orcid.org/0009-0008-3571-1589)
- Hongyao Ma (ORCID: https://orcid.org/0000-0002-7705-8281)
- Sihao Wu (ORCID: https://orcid.org/0009-0005-7032-8903)
- Dayou Cui
Institutions
- Beijing University of Technology (CN)
- Shandong Jiaotong University (CN)
Publication Details
- Journal
- Scientific Reports
- Published
- 2025-11-06
- DOI
- https://doi.org/10.1038/s41598-025-22862-x
- Citations
- 1
- Primary Topic
- Advanced Neural Network Applications
- Type
- article
- Field-Weighted Citation Impact
- 1.28