Please wait a minute...
浙江大学学报(工学版)  2026, Vol. 60 Issue (9): 1934-1941    DOI: 10.3785/j.issn.1008-973X.2026.09.010
计算机技术、自动控制技术     
用于实时语义分割的数据高效轻量模型
李宏伟(),丁建阳,柴志雷*()
江南大学 人工智能与计算机学院,江苏 无锡 214122
Data-efficient lightweight model for real-time semantic segmentation
Hongwei LI(),Jianyang DING,Zhilei CHAI*()
School of Artificial Intelligence and Computer Science, Jiangnan University, Wuxi 214122, China
 全文: PDF(3944 KB)   HTML
摘要:

针对边缘设备实时语义分割任务对轻量化、高精度与高速度的要求,提出高效的轻量级模型U-SAM. 该模型基于改进的U-Net架构,在编码阶段设计多尺度空洞卷积融合模块,以增强特征表达能力并显著降低计算复杂度. 在解码阶段引入通道注意力与像素注意力机制,实现关键特征信息的有效建模与跨尺度融合. 在瓶颈部分融合自注意力机制与卷积操作,高效捕捉长距离依赖关系并提升模型泛化能力. 在小规模数据集CamVid上的实验表明,U-SAM仅需约1.3×106的参数量,即可达到70.2%的mIoU分割精度,推理速度达到69帧/s. U-SAM的综合性能显著优于多种主流轻量级实时语义分割模型,为自动驾驶、智能监控、遥感卫星等资源受限场景下的实时语义分割应用提供了高性能解决方案.

关键词: 语义分割实时性轻量模型小规模数据集空洞卷积注意力机制    
Abstract:

An efficient lightweight model, U-SAM, was proposed aiming at the requirement for lightweight, high-precision and high-speed performance in real-time semantic segmentation task on edge device. A multi-scale dilated convolution fusion module was designed in the encoder stage based on an improved U-Net architecture in order to enhance feature representation capability and significantly reduce computational complexity. Channel attention and pixel attention mechanism were introduced in the decoder stage in order to achieve effective modeling of key feature information and cross-scale fusion. Self-attention mechanism and convolution operation were integrated in the bottleneck section in order to efficiently capture long-range dependency and improve the generalization capability of the model. Experiments on the small-scale CamVid dataset demonstrated that U-SAM required only about 1.3×106 parameters in order to achieve segmentation accuracy of 70.2% mIoU, with inference speed reaching 69 frame/s. The overall performance of U-SAM is significantly superior to several mainstream lightweight real-time semantic segmentation models, providing a high-performance solution for real-time semantic segmentation application in resource-constrained scenarios such as autonomous driving, intelligent surveillance and remote sensing satellite.

Key words: semantic segmentation    real-time performance    lightweight model    small-scale dataset    dilated convolution    attention mechanism
收稿日期: 2025-11-29 出版日期: 2026-07-20
CLC:  TP 394  
基金资助: 国家自然科学基金资助项目(61972180);中央高校基本科研业务费资助项目(JUSRP202501072);无锡市科技发展基金资助项目(K20241027).
通讯作者: 柴志雷     E-mail: 6233114016@stu.jiangnan.edu.cn;zlchai@jiangnan.edu.cn
作者简介: 李宏伟(2002—),男,硕士生,从事人工智能研究. orcid.org/0009-0007-1337-0958. E-mail:6233114016@stu.jiangnan.edu.cn
服务  
把本文推荐给朋友
加入引用管理器
E-mail Alert
作者相关文章  
李宏伟
丁建阳
柴志雷

引用本文:

李宏伟,丁建阳,柴志雷. 用于实时语义分割的数据高效轻量模型[J]. 浙江大学学报(工学版), 2026, 60(9): 1934-1941.

Hongwei LI,Jianyang DING,Zhilei CHAI. Data-efficient lightweight model for real-time semantic segmentation. Journal of ZheJiang University (Engineering Science), 2026, 60(9): 1934-1941.

链接本文:

https://www.zjujournals.com/eng/CN/10.3785/j.issn.1008-973X.2026.09.010        https://www.zjujournals.com/eng/CN/Y2026/V60/I9/1934

图 1  U-SAM的整体结构
图 2  Encoder的结构
图 3  Decoder的结构
图 4  多头自注意力机制
图 5  多尺度融合模块
模型Np/106Dice/%IoU/%
Unet++[15]9.287.7880.72
Unet[16]31.288.6081.58
Attention UNet[17]31.488.6681.68
ConvUNeXt[18]3.589.8183.19
U-Lite[14]0.990.3983.63
U-SAM(本文方法)0.5190.2483.70
1.3290.7983.90
表 1  不同模型在ISIC2018数据集上的性能对比
类别模型输入尺寸主干网络GPUFLOPs/109Np/106v/(帧·s?1)mIoU/%
大型模型SegNet[19]360×480VGG-16Titan X106.5929.506055.6
DFANet[20]720×960XceptionTitan X7.8012064.7
ICNet[21]720×960PSPNet-50Titan X26.502867.1
MLFNet[22]720×960ResNet-34Titan XP13.035769.0
MGSeg[23]736×960ResNet-18Titan XP13.3012772.7
轻量模型ENet[24]360×480NoTitan X2.120.369651.3
NDNet[25]360×480NoTitan X1.160.507857.2
DABNet[26]360×480No1080 Ti3.490.7613666.4
FDDWNet[27]360×480No2080 Ti4.580.807966.9
LARNet[28]360×480No2080Ti0.9520467.1
BiLevelNet[29]360×480No30900.7068.2
FBSNet[8]360×480No2080 Ti7.970.6212068.9
MSCFNet[30]360×480NoTitan XP1.1511069.3
LETNet[31]360×480No30904.840.9520070.5
U-SAM(本文方法)360×480No4060 Ti4.431.326970.2
720×960No4060 Ti17.791.324470.4
表 2  不同模型在CamVid数据集上的性能对比
模型v/(帧·s?1)Mp/106Ec/(W·h)
FDDWNet[27]412414.01
FBSNet[8]224115.25
LETNet[31]284384.32
U-SAM(本文方法)693423.84
表 3  相同环境下的性能对比
图 6  不同网络在CamVid数据集上的可视化结果
图 7  可视化结果的局部放大
图 8  ISIC2018的特征可视化结果
模块结构Np/106v/(帧·s?1)mIoU/%
编码Dilation=[2,3,5,9]1.386769.3
Dilation=[2,3,5]1.327068.8
解码No CA+PA1.2911064.6
瓶颈No Transformer+CNN1.137765.9
Only Transformer1.267166.6
Only CNN1.197366.3
U-SAM[2,3,5,9]+[2,3,5]、
CA+PA、Transformer+CNN
1.326970.2
表 4  各模块在CamVid数据集上的消融研究
图 9  U-SAM图像分割在喷墨印刷中的应用
1 何淼楹, 崔宇超 面向自动驾驶的交通场景语义分割[J]. 计算机应用, 2021, 41 (Suppl.1): 25- 30
HE Miaoying, CUI Yuchao Semantic segmentation of traffic scenes for autonomous driving[J]. Journal of Computer Applications, 2021, 41 (Suppl.1): 25- 30
doi: 10.11772/j.issn.1001-9081.2020071114
2 熊彬, 张双德 基于改进PSPNet的卫星遥感图像建筑物语义分割算法[J]. 遥感信息, 2023, 38 (4): 73- 79
XIONG Bin, ZHANG Shuangde Semantic segmentation algorithm for buildings in satellite remote sensing images based on improved PSPNet[J]. Remote Sensing Information, 2023, 38 (4): 73- 79
3 HE K, ZHANG X, REN S, et al. Deep residual learning for image recognition [C]//Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition. Las Vegas: IEEE, 2016: 770–778.
4 LONG J, SHELHAMER E, DARRELL T. Fully convolutional networks for semantic segmentation [C]// Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition. New York: IEEE, 2015: 3431–3440.
5 YANG M, YU K, ZHANG C, et al. DenseASPP for semantic segmentation in street scenes [C]//Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition. Salt Lake City: IEEE, 2018: 3684–3692.
6 WANG W, XIE E, LI X, et al PVT v2: improved baselines with Pyramid Vision Transformer[J]. Computational Visual Media, 2022, 8 (3): 415- 424
doi: 10.1007/s41095-022-0274-8
7 GAO R. Rethinking dilated convolution for real-time semantic segmentation [EB/OL]. (2021-11-18) [2025-03-28]. https://arxiv.org/abs/2111.09957.
8 GAO G, XU G, LI J, et al FBSNet: a fast bilateral symmetrical network for real-time semantic segmentation[J]. IEEE Transactions on Multimedia, 2022, 25: 3273- 3283
doi: 10.1109/tmm.2022.3157995
9 ZHOU Q, WANG L, GAO G, et al Boundary-guided lightweight semantic segmentation with multi-scale semantic context[J]. IEEE Transactions on Multimedia, 2024, 26: 7887- 7900
doi: 10.1109/TMM.2024.3372835
10 VASWANI A, SHAZEER N, PARMAR N, et al Attention is all you need[J]. Advances in Neural Information Processing Systems, 2017, 30: 6000- 6010
doi: 10.1007/978-3-031-84300-6_13
11 DOSOVITSKIY A, BEYER L, KOLESNIKOV A, et al. An image is worth 16x16 words: transformers for image recognition at scale [EB/OL]. (2020-10-22) [2025-03-23]. https://arxiv.org/abs/2010.11929.
12 HUANG Z, WANG X, HUANG L, et al. CCNet: criss-cross attention for semantic segmentation [C]//Proceedings of the IEEE/CVF International Conference on Computer Vision. Seoul: IEEE, 2020: 603–612.
13 CHEN L, FU Y, GU L, et al Frequency-aware feature fusion for dense image prediction[J]. IEEE Transactions on Pattern Analysis and Machine Intelligence, 2024, 46 (12): 10763- 10780
doi: 10.1109/TPAMI.2024.3449959
14 DINH B D, NGUYEN T T, TRAN T T, et al. 1M parameters are enough? A lightweight CNN-based model for medical image segmentation [C]//Proceedings of the Asia Pacific Signal and Information Processing Association Annual Summit and Conference. Taipei: IEEE, 2023: 1279–1284.
15 ZHOU Z, RAHMAN SIDDIQUEE M M, TAJBAKHSH N, et al. UNet++: a nested U-Net architecture for medical image segmentation [C]//Deep Learning in Medical Image Analysis and Multimodal Learning for Clinical Decision Support. Cham: Springer, 2018: 3–11.
16 RONNEBERGER O, FISCHER P, BROX T. U-Net: convolutional networks for biomedical image segmentation [C]//Medical Image Computing and Computer-Assisted Intervention. Cham: Springer, 2015: 234–241.
17 OKTAY O, SCHLEMPER J, LE FOLGOC L, et al. Attention U-Net: learning where to look for the pancreas [EB/OL]. (2018-04-11) [2025-03-25]. https://arxiv.org/abs/1804.03999.
18 HAN Z, JIAN M, WANG G G ConvUNeXt: an efficient convolution neural network for medical image segmentation[J]. Knowledge-Based Systems, 2022, 253: 109512
doi: 10.1016/j.knosys.2022.109512
19 BADRINARAYANAN V, KENDALL A, CIPOLLA R SegNet: a deep convolutional encoder-decoder architecture for image segmentation[J]. IEEE Transactions on Pattern Analysis and Machine Intelligence, 2017, 39 (12): 2481- 2495
doi: 10.1109/TPAMI.2016.2644615
20 LI H, XIONG P, FAN H, et al. Dfanet: deep feature aggregation for real-time semantic segmentation [C]// Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition. New York: IEEE, 2019: 9522–9531.
21 ZHAO H, QI X, SHEN X, et al. Icnet for real-time semantic segmentation on high-resolution images [C]// Proceedings of the European Conference on Computer Vision. Cham: Springer, 2018: 405–420.
22 FAN J, WANG F, CHU H, et al MLFNet: multi-level fusion network for real-time semantic segmentation of autonomous driving[J]. IEEE Transactions on Intelligent Vehicles, 2022, 8 (1): 756- 767
doi: 10.1109/wcnc61545.2025.10978834
23 HE J, LIANG S, WU X, et al MGSeg: multiple granularity-based real-time semantic segmentation network[J]. IEEE Transactions on Image Processing, 2021, 30: 7200- 7214
doi: 10.1109/TIP.2021.3102509
24 PASZKE A, CHAURASIA A, KIM S, et al. ENet: a deep neural network architecture for real-time semantic segmentation [EB/OL]. (2016-06-07) [2025-03-06]. https://arxiv.org/abs/1606.02147.
25 LI S, YAN Q, ZHOU X, et al NDNet: spacewise multiscale representation learning via neighbor decoupling for real-time driving scene parsing[J]. IEEE Transactions on Neural Networks and Learning Systems, 2024, 35 (6): 7884- 7898
doi: 10.1109/TNNLS.2022.3221745
26 LI G, YUN I, KIM J, et al. DABNet: depth-wise asymmetric bottleneck for real-time semantic segmentation [EB/OL]. (2019-07-25) [2025-02-15]. https://arxiv.org/abs/1907.11357.
27 LIU J, ZHOU Q, QIANG Y, et al. FDDWNet: a lightweight convolutional neural network for real-time semantic segmentation [C]//2020 IEEE International Conference on Acoustics, Speech and Signal Processing. Barcelona: IEEE, 2020: 2373–2377.
28 HU X, XU S, JING L Lightweight attention-guided redundancy-reuse network for real-time semantic segmentation[J]. IET Image Processing, 2023, 17 (9): 2649- 2658
doi: 10.1049/ipr2.12816
29 吴马靖, 张永爱, 林珊玲, 等 基于BiLevelNet的实时语义分割算法[J]. 光电工程, 2024, 51 (5): 240030
WU Majing, ZHANG Yongai, LIN Shanling, et al Real-time semantic segmentation algorithm based on BiLevelNet[J]. Opto-Electronic Engineering, 2024, 51 (5): 240030
doi: 10.12086/oee.2024.240030
30 ZHANG T, ZHUANG Y, WANG G, et al Multiscale semantic fusion-guided fractal convolutional object detection network for optical remote sensing imagery[J]. IEEE Transactions on Geoscience and Remote Sensing, 2022, 60: 5608720
doi: 10.1109/tgrs.2021.3108476
[1] 杨恒,王韶涵,董青,赵科渊,杨明亮. 基于注意力增强的跨模态多层融合网络的物体位姿估计[J]. 浙江大学学报(工学版), 2026, 60(9): 1872-1880.
[2] 火久元,姜灏,常琛. 基于改进EAST的高速公路交通标识文本检测[J]. 浙江大学学报(工学版), 2026, 60(9): 1881-1889.
[3] 杨军,卯恒睿,党吉圣. 基于多模态知识对齐的双分支点云语义分割算法[J]. 浙江大学学报(工学版), 2026, 60(9): 1924-1933.
[4] 赵明华,吕雨萱,吕佳豪,陈逸飞,石程,胡静. 基于多尺度外观运动融合的视频异常检测[J]. 浙江大学学报(工学版), 2026, 60(8): 1730-1738.
[5] 陈广秋,任天蓉,段锦,黄丹丹. 结合边缘辅助与多级特征融合的跨模态语义分割算法[J]. 浙江大学学报(工学版), 2026, 60(8): 1782-1791.
[6] 刘慧霞,王荣景,高宇轩,曹猛,盛苏英,马友鹏. 基于自适应神经网络的无信号交叉口车辆无死锁调度算法[J]. 浙江大学学报(工学版), 2026, 60(8): 1832-1840.
[7] 徐恺蔚,KHIZER BIN TALIBHafiz,曹衍龙,许源平,许志杰,宋景春. 基于光流和卷积视觉Transformer的轻量级微表情识别[J]. 浙江大学学报(工学版), 2026, 60(7): 1381-1391.
[8] 杨艳春,李佳龙. 基于多路协同与空谱先验的高光谱与多光谱图像融合[J]. 浙江大学学报(工学版), 2026, 60(7): 1427-1437.
[9] 张乃洲,赵云超,曹薇,张啸剑. 基于多视图跨模态特征融合的图像描述生成[J]. 浙江大学学报(工学版), 2026, 60(6): 1205-1212.
[10] 李云红,张琪琪,陈锦妮,陈伟重,苏雪平,梁成名. 基于生成对抗网络和坐标注意力机制的文本生成图像算法[J]. 浙江大学学报(工学版), 2026, 60(6): 1213-1220.
[11] 李国燕,于威,梅玉鹏,张明辉,王新强. 全局局部特征融合的遥感图像建筑物提取[J]. 浙江大学学报(工学版), 2026, 60(5): 1100-1108.
[12] 李子祥,陆克成,蔡海兵,解伟帅,张广东. 基于触觉仿生技术的黑暗环境混凝土裂缝检测[J]. 浙江大学学报(工学版), 2026, 60(5): 915-925.
[13] 宋耀莲,彭驰,唐菁敏,赵宣植,虞贵财. 基于融合注意力机制的光学遥感图像小目标检测算法[J]. 浙江大学学报(工学版), 2026, 60(4): 763-771.
[14] 万刚,王小波,石纲,叶德震,朱思思,司帆. 基于特征细化与注意力增强重构的水下图像增强算法[J]. 浙江大学学报(工学版), 2026, 60(4): 800-811.
[15] 陈文强,冯琳越,王东丹,顾玉磊,赵轩. 融合动态风险图与多变量注意力机制的车辆轨迹预测模型[J]. 浙江大学学报(工学版), 2026, 60(3): 455-467.