LMSC-UNet: A Lightweight U-Net with Modified Skip Connections for Semantic Segmentation

Shrutika S. Sawant, Andreas Medgyesy, Sahana Raghunandan, Theresa Götz, Theresa Götz

2025

Abstract

U-Net, an encoder-decoder architecture is the most popular choice in the semantic segmentation field due to its ability to learn rich semantic features while handling enormous amounts of data. However, due to large number of parameters and slow inference, deploying U-Net on devices with limited computational resources such as mobile and embedded devices becomes challenging. To alleviate the above challenge, in this study, we propose an efficient, lightweight, and robust encoder-decoder architecture, LMSC-UNet for semantic segmentation that captures more comprehensive, contextual information and effectively learns rich semantic features. This lightweight architecture considerably reduces the number of trainable parameters, requiring sufficiently less memory space, training, and inference time. Skip connections in original U-Net fuse features from each encoder block to the corresponding decoder block. This simple skip connection reduces the semantic gap to some extent and may limit the segmentation performance. Therefore, we replace the skip connection from the second level of U-Net with a bottleneck residual block (BRB) which helps to enhance the final segmentation map by lessening the semantic gap between the features of decoder with the corresponding features of encoder. Extensive experiments on various segmentation datasets from diverse domains demonstrate the effectiveness of our proposed approach. The experimental results show that the compact model speeds up the inference process, while still maintaining the performance. When compared to the standard U-Net, LMSC-UNet has achieved 7× reduction in Floating Point Operations (FLOPs), and 34× reduction in model size, while maintaining the segmentation accuracy.

Paper Citation

in Harvard Style

Sawant S., Medgyesy A., Raghunandan S. and Götz T. (2025). LMSC-UNet: A Lightweight U-Net with Modified Skip Connections for Semantic Segmentation. In Proceedings of the 17th International Conference on Agents and Artificial Intelligence - Volume 2: ICAART; ISBN 978-989-758-737-5, SciTePress, pages 726-734. DOI: 10.5220/0013343800003890

in Bibtex Style

@conference{icaart25,
author={Shrutika Sawant and Andreas Medgyesy and Sahana Raghunandan and Theresa Götz},
title={LMSC-UNet: A Lightweight U-Net with Modified Skip Connections for Semantic Segmentation},
booktitle={Proceedings of the 17th International Conference on Agents and Artificial Intelligence - Volume 2: ICAART},
year={2025},
pages={726-734},
publisher={SciTePress},
organization={INSTICC},
doi={10.5220/0013343800003890},
isbn={978-989-758-737-5},
}

in EndNote Style

TY - CONF

JO - Proceedings of the 17th International Conference on Agents and Artificial Intelligence - Volume 2: ICAART
TI - LMSC-UNet: A Lightweight U-Net with Modified Skip Connections for Semantic Segmentation
SN - 978-989-758-737-5
AU - Sawant S.
AU - Medgyesy A.
AU - Raghunandan S.
AU - Götz T.
PY - 2025
SP - 726
EP - 734
DO - 10.5220/0013343800003890
PB - SciTePress