高级搜索

留言板

尊敬的读者、作者、审稿人, 关于本刊的投稿、审稿、编辑和出版的任何问题, 您可以本页添加留言。我们将尽快给您答复。谢谢您的支持!

姓名
邮箱
手机号码
标题
留言内容
验证码

基于动态提示与知识蒸馏的轻量化多模态语义分割方法

沈博,  张志翔,  商正莹,  朱振宇,  杨刚

沈博, 张志翔, 商正莹, 朱振宇, 杨刚. 基于动态提示与知识蒸馏的轻量化多模态语义分割方法[J]. 电子与信息学报. doi: 10.11999/JEIT260729
引用本文: 沈博, 张志翔, 商正莹, 朱振宇, 杨刚. 基于动态提示与知识蒸馏的轻量化多模态语义分割方法[J]. 电子与信息学报. doi: 10.11999/JEIT260729
SHEN Bo, ZHANG Zhixiang, SHANG Zhengying, ZHU Zhenyu, YANG Gang. A Lightweight Multi-Modal Semantic Segmentation Method Based on Dynamic Prompt and Knowledge Distillation[J]. Journal of Electronics & Information Technology. doi: 10.11999/JEIT260729
Citation: SHEN Bo, ZHANG Zhixiang, SHANG Zhengying, ZHU Zhenyu, YANG Gang. A Lightweight Multi-Modal Semantic Segmentation Method Based on Dynamic Prompt and Knowledge Distillation[J]. Journal of Electronics & Information Technology. doi: 10.11999/JEIT260729

基于动态提示与知识蒸馏的轻量化多模态语义分割方法

doi: 10.11999/JEIT260729 cstr: 32379.14.JEIT260729
详细信息
    作者简介:

    沈博:男,副教授,研究方向为分布式计算。邮箱:shen@nwpu.edu.cn

    张志翔:男,硕士生,研究方向为计算机视觉

    商正莹:女,硕士生,研究方向为计算机视觉

    朱振宇:男,硕士生,研究方向为嵌入式计算

    杨刚:男,教授,研究方向为分布式计算

    通讯作者:

    沈博 shen@nwpu.edu.cn

  • 中图分类号: TP391.41

A Lightweight Multi-Modal Semantic Segmentation Method Based on Dynamic Prompt and Knowledge Distillation

  • 摘要: 语义分割是计算机视觉中场景理解的重要组成部分,然而传统方法大多基于可见光图像,在低亮度、强光照与遮挡等场景中性能急剧下降。尽管多模态融合能有效增强模型鲁棒性,现有方法普遍采用双支路网络架构,大幅增加模型参数量,难以部署在算力受限的嵌入式平台。针对上述问题,本文提出基于多模态特征融合提示网络的语义分割模型(Multi-modal Feature Fusion Prompt Network for robust semantic segmentation, MFFPNet)。该模型采用单支路骨干架构,引入轻量级动态视觉提示模块,将热红外或深度信息编码为动态提示向量,通过注意力机制注入预训练骨干网络,实现跨模态信息的高效交互。在此基础上,结合知识蒸馏对模型进行轻量化压缩,在硬件方面分层解耦架构屏蔽异构性,并引入多线程异步流水线,实现模型运行时推理优化,提升嵌入式平台上的语义分割实时性。实验结果表明,MFFPNet仅需更新9.4 M参数,约占模型总参数量的10.4%,在MFNet数据集上平均交并比(mIoU)达到63.3%,较基准方法提高1.9%,在NYUDv2数据集上mIoU达到60.6%,方法在精度与轻量化之间取得了较优平衡;轻量化变体MFFPNet-S在MFNet数据集上,以教师模型约20.0%的总参数量,保留其约88.8%的性能;提出的运行时优化策略在三种异构硬件平台上将吞吐量提升了21.7%~40.9%,满足了主流商用嵌入式平台与国产自主平台的实时部署需求。
  • 图  1  多模态特征融合的语义分割模型MFFPNet结构

    图中左侧蓝色虚线框表示PMiT模块,内部依次包含4个PETB模块,右侧为FFP内部结构;红色模块代表需更新的可训练参数,蓝色模块代表冻结的预训练参数。

    图  2  MFNet数据集上的可视化分割结果

    图  3  NYUDv2数据集上的可视化分割结果

    图  4  基于自适应结构化蒸馏的整体框架

    图  5  特征分布适配器结构与特征对齐示意图

    图  6  基于置信度的自适应掩码生成策略

    图  7  异构嵌入式计算验证平台

    表  1  实验参数配置

    属性名称数据信息
    运行环境Ubuntu 22.04
    CUDA版本12.4
    显存容量48 GB GDDR6
    输入分辨率480×640
    Batch size6
    优化器SGD
    epoch1000
    初始学习率0.005(SGD)
    下载: 导出CSV

    表  2  在MFNet数据集上的精度对比

    Method Backbone mIoU(%) 总参数量(M) 训练参数量(M)
    MFNet[16] Custom CNN 39.7 - -
    RTFNet[7] ResNet 53.2 - -
    CMX-B2[6] MiT-B2 58.2 66.6 66.6
    CMX-B4[6] MiT-B4 59.7 139.9 139.9
    DPLNet[9] MiT-B5 58.3 88.6 7.15
    CRM-B[31] Swin-B 61.4 - -
    MixPrompt[10] MiT-B5 60.1 87.2 5.7
    Ours MiT-B5 63.3 90.4 9.4
    下载: 导出CSV

    表  3  在NYUDv2数据集上的指标对比

    Method Backbone 测试方式 mIoU (%) 总参数量(M) 训练参数量(M)
    TokenFusion-B3[12] MiT-B3 SS 54.2 45.9 45.9
    CMX-B5[6] MiT-B5 MS 56.9 181.1 181.1
    CMNeXt[20] MiT-B4 MS 56.9 119.6 119.6
    DFormer-L[32] DFormer-L MS 57.2 39.0 39.0
    DPLNet[9] MiT-B5 SS 58.3 88.6 7.2
    DPLNet[9] MiT-B5 MS 59.3 88.6 7.2
    MixPrompt[10] MiT-B5 MS 61.2 87.2 5.7
    Ours MiT-B5 SS 60.6 90.4 9.4
    下载: 导出CSV

    表  4  核心模块有效性消融实验

    模型配置 $ {P}_{\text{global}} $ $ {F}_{\text{FFP}} $ MFNet
    mIoU(%)
    NYUDv2
    mIoU(%)
    训练参数量(M)
    Baseline (Frozen MiT-B5 Backbone)[26] - - 40.5 48.4 -
    Baseline + Global Prompt √ - 45.1 55.0 ~0.1
    Baseline + Global Prompt + Simple Add √ - 58.6 57.4 2.6
    MFFPNet (Ours) √ √ 63.3 60.6 9.4
    下载: 导出CSV

    表  5  FFP内部融合策略消融实验

    融合策略描述mIoU(%)
    Element-wise Sum简单的逐元素相加融合56.5
    Concatenation通道拼接后卷积降维融合59.8
    Channel + Spatial Attn Only仅使用双路注意力加权60.5
    w/o Cross-Attention保留互补加权,不使用交叉注意力62.1
    c/s Attn + Cross-Attn (Ours)完整的$ {F}_{\text{FFP}} $模块63.3
    下载: 导出CSV

    表  6  FFP插入位置策略消融实验

    实验设置策略描述mIoU(%)
    $ {F}_{\text{FFP}} $插入位置Stage 1 only (仅浅层)57.6
    Stage 4 only (仅深层)59.2
    Stage 1 → 4 (All Stages)63.3
    训练范式对比Frozen Decoder
    (SegFormer-B5[26],冻结解码器)
    58.1
    Full Fine-tuning
    (SegFormer-B5[26],全量微调,不含提示机制)
    60.8
    下载: 导出CSV

    表  7  MFFPNet关键超参数敏感性分析(MFNet数据集)

    实验编号 参数设置 mIoU(%)
    E0-B5 λ1=0.5,λ2=0.5(默认) 63.3
    E1 λ1=0.3,λ2=0.7 62.8
    E2 λ1=0.7,λ2=0.3 62.8
    E0-B5 α=0.1(默认) 63.3
    E3 α=0 63.2
    E4 α=0.2 63.1
    下载: 导出CSV

    表  8  在MFNet数据集上的性能对比

    方法 总参数量(M) 训练参数量
    (M)
    FLOPs
    (G)
    帧率/
    (帧·s–1)
    mIoU
    (%)
    Teacher 90.4 9.4 122.8 10.0 63.3
    Student 18.1 18.1 19.6 23.9 56.2
    下载: 导出CSV

    表  9  NYUDv2数据集上的模型性能对比与增益分析

    模型 总参数量
    (M)
    FLOPs
    (G)
    帧率/
    (帧·s–1)
    mIoU
    (%)
    Teacher(MFFPNet) 90.4 122.8 9.83 60.6
    Student-B1(w/o KD) 18.05 19.6 25.06 38.6
    MFFPNet-S (MiT-B1) 18.05 19.6 25.59 42.9
    MFFPNet-S (MiT-B2) 29.33 29.1 20.09 48.5
    下载: 导出CSV

    表  10  不同蒸馏组件对模型性能的影响(MFNet数据集)

    模型配置$ {\mathcal{L}}_{\text{logit}} $$ {\mathcal{L}}_{\text{feat}} $$ {\mathcal{L}}_{\text{rel}} $$ {\mathcal{L}}_{\text{pair}} $mIoU(%)
    Student Baseline(MiT-B1)----38.6
    + Adaptive Logit Loss√---45.2
    + Feature Content Loss√√--51.6
    + Channel Relation Loss√√√-53.4
    MFFPNet-S(Ours)√√√√56.2
    MFFPNet(Teacher)----63.3
    下载: 导出CSV

    表  11  MFFPNet-S蒸馏权重敏感性分析(MFNet数据集)

    实验编号参数设置mIoU(%)
    E0-B1ρ=0.7(默认)56.21
    E5ρ=0.555.60
    E6ρ=0.956.18
    下载: 导出CSV

    表  12  异构平台模型推理性能对比

    模型配置 部署平台 量化精度 mIoU
    (%)
    帧率/
    (帧·s–1)
    MFFPNet(Teacher) Server(L40S) FP32 63.3 10.0
    MFFPNet-S (PyTorch) Server(L40S) FP32 56.2 23.9
    MFFPNet-S(TensorRT) Jetson Orin FP16 55.7 48.6
    MFFPNet-S (CANN) Atlas 200I FP16 51.5 31.5
    MFFPNet-S (RKNN) RK3588 Hybrid 49.4 28.6
    MFFPNet-S (RKNN) RK3588 INT8 41.1 31.2
    下载: 导出CSV

    表  13  多线程流水线优化效果对比

    硬件平台 运行模式 CPU利用率
    (%)
    GPU/NPU
    利用率(%)
    帧率/
    (帧·s–1)
    Jetson Orin NX 串行 35 42 48.6
    并行 68 95 68.5
    Atlas 200I DK A2 串行 55 38 31.5
    并行 92 65 41.2
    RK3588 串行 20 85 28.6
    并行 35 96 34.8
    下载: 导出CSV
  • [1] FENG Di, HAASE-SCHÜTZ C, ROSENBAUM L, et al. Deep multi-modal object detection and semantic segmentation for autonomous driving: Datasets, methods, and challenges[J]. IEEE Transactions on Intelligent Transportation Systems, 2021, 22(3): 1341–1360. doi: 10.1109/TITS.2020.2972974.
    [2] WANG Li, ZHANG Xinyu, SONG Ziying, et al. Multi-modal 3D object detection in autonomous driving: A survey and taxonomy[J]. IEEE Transactions on Intelligent Vehicles, 2023, 8(7): 3781–3798. doi: 10.1109/TIV.2023.3264658.
    [3] 梁燕, 李俊范, 邵凯, 等. 空间信息引导扩散的遥感图像域自适应语义分割[J]. 电子与信息学报, 2026, 48(7): 2829–2842. doi: 10.11999/JEIT260031.

    LIANG Yan, LI Junfan, SHAO Kai, et al. Spatial information-guided diffusion for domain adaptation semantic segmentation of remote sensing images[J]. Journal of Electronics & Information Technology, 2026, 48(7): 2829–2842. doi: 10.11999/JEIT260031.
    [4] WANG Changshuo, WANG Chen, LI Weijun, et al. A brief survey on RGB-D semantic segmentation using deep learning[J]. Displays, 2021, 70: 102080. doi: 10.1016/j.displa.2021.102080.
    [5] HAZIRBAS C, MA Lingni, DOMOKOS C, et al. FuseNet: Incorporating depth into semantic segmentation via fusion-based CNN architecture[C]. 13th Asian Conference on Computer Vision, Taipei, China, 2016: 213–228. doi: 10.1007/978-3-319-54181-5_14.
    [6] ZHANG Jiaming, LIU Huayao, YANG Kailun, et al. CMX: Cross-modal fusion for RGB-X semantic segmentation with transformers[J]. IEEE Transactions on Intelligent Transportation Systems, 2023, 24(12): 14679–14694. doi: 10.1109/TITS.2023.3300537.
    [7] SUN Yuxiang, ZUO Weixun, and LIU Ming. RTFNet: RGB-thermal fusion network for semantic segmentation of urban scenes[J]. IEEE Robotics and Automation Letters, 2019, 4(3): 2576–2583. doi: 10.1109/LRA.2019.2904733.
    [8] CAPRA M, BUSSOLINO B, MARCHISIO A, et al. Hardware and software optimizations for accelerating deep neural networks: Survey of current trends, challenges, and the road ahead[J]. IEEE Access, 2020, 8: 225134–225180. doi: 10.1109/ACCESS.2020.3039858.
    [9] DONG Shaohua, FENG Yunhe, YANG Qing, et al. Efficient multimodal semantic segmentation via dual-prompt learning[C]. 2024 IEEE/RSJ International Conference on Intelligent Robots and Systems, Abu Dhabi, United Arab Emirates, 2024: 14196–14203. doi: 10.1109/IROS58592.2024.10801872.
    [10] HAO Zhiwei, XIAO Zhongyu, GUO Jianyuan, et al. MixPrompt: Efficient mixed prompting for multimodal semantic segmentation[C]. The Thirty-ninth Annual Conference on Neural Information Processing Systems, San Diego, USA, 2025: 39255–39282. doi: 10.52202/085713-1173.
    [11] 毛秀华, 张强, 阮航, 等. 多任务协同的多模态遥感目标分割算法[J]. 电子与信息学报, 2024, 46(8): 3363–3371. doi: 10.11999/JEIT231267.

    MAO Xiuhua, ZHANG Qiang, RUAN Hang, et al. Multitask collaborative multi-modal remote sensing target segmentation algorithm[J]. Journal of Electronics & Information Technology, 2024, 46(8): 3363–3371. doi: 10.11999/JEIT231267.
    [12] WANG Yikai, CHEN Xinghao, CAO Lele, et al. Multimodal token fusion for vision transformers[C]. Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, New Orleans, USA, 2022: 12176–12185. doi: 10.1109/cvpr52688.2022.01187.
    [13] ZHANG Yang, XIONG Chenyun, LIU Junjie, et al. Spatial information-guided adaptive context-aware network for efficient RGB-D semantic segmentation[J]. IEEE Sensors Journal, 2023, 23(19): 23512–23521. doi: 10.1109/JSEN.2023.3304637.
    [14] 李晓玲, 陈后金, 李艳凤, 等. 多重关系感知的红外与可见光图像融合网络[J]. 电子与信息学报, 2024, 46(5): 2217–2227. doi: 10.11999/JEIT231062.

    LI Xiaoling, CHEN Houjin, LI Yanfeng, et al. Infrared and visible image fusion network with multi-relation perception[J]. Journal of Electronics & Information Technology, 2024, 46(5): 2217–2227. doi: 10.11999/JEIT231062.
    [15] DENG Fuqin, FENG Hua, LIANG Mingjian, et al. FEANet: Feature-enhanced attention network for RGB-thermal real-time semantic segmentation[C]. IEEE/RSJ International Conference on Intelligent Robots and Systems, Prague, Czech Republic, 2021: 4467–4473. doi: 10.1109/iros51168.2021.9636084.
    [16] HA Qishen, WATANABE K, KARASAWA T, et al. MFNet: Towards real-time semantic segmentation for autonomous vehicles with multi-spectral scenes[C]. 2017 IEEE/RSJ International Conference on Intelligent Robots and Systems, Vancouver, Canada, 2017: 5108–5115. doi: 10.1109/iros.2017.8206396.
    [17] ZHOU Wujie, LIN Xinyang, LEI Jingsheng, et al. MFFENet: Multiscale feature fusion and enhancement network for RGB–thermal urban road scene parsing[J]. IEEE Transactions on Multimedia, 2022, 24: 2526–2538. doi: 10.1109/TMM.2021.3086618.
    [18] LIU Jinfu, ZHOU Wujie, CUI Yueli, et al. GCNet: Grid-like context-aware network for RGB-thermal semantic segmentation[J]. Neurocomputing, 2022, 506: 60–67. doi: 10.1016/j.neucom.2022.07.041.
    [19] ZHOU Wujie, DONG Shaohua, XU Caie, et al. Edge-aware guidance fusion network for RGB-thermal scene parsing[C]. Proceedings of the 36th AAAI Conference on Artificial Intelligence, 2022: 3571–3579. doi: 10.1609/aaai.v36i3.20269. (查阅网上资料,未找到本条文献会议地,请确认).
    [20] ZHANG Jiaming, LIU Ruiping, SHI Hao, et al. Delivering arbitrary-modal semantic segmentation[C]. Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, Vancouver, Canada, 2023: 1136–1147. doi: 10.1109/cvpr52729.2023.00116.
    [21] 周国宇, 张菁, 闫伊, 等. 聚焦注意力与紧致特征融合Transformer的城市遥感影像语义分割[J]. 电子与信息学报, 2025, 47(12): 4790–4800. doi: 10.11999/JEIT250812.

    ZHOU Guoyu, ZHANG Jing, YAN Yi, et al. A focused attention and feature compact fusion transformer for semantic segmentation of urban remote sensing images[J]. Journal of Electronics & Information Technology, 2025, 47(12): 4790–4800. doi: 10.11999/JEIT250812.
    [22] HAO Zhiwei, XIAO Zhongyu, LUO Yong, et al. PrimKD: Primary modality guided multimodal fusion for RGB-D semantic segmentation[C]. Proceedings of the 32nd ACM International Conference on Multimedia, Melbourne, Australia, 2024: 1943–1951. doi: 10.1145/3664647.3681253.
    [23] FERROD R, DANTAS C F, DI CARO L, et al. Revisiting cross-modal knowledge distillation: A disentanglement approach for RGBD semantic segmentation[C]. Joint European Conference on Machine Learning and Knowledge Discovery in Databases, Porto, Portugal, 2025: 492–508. doi: 10.1007/978-3-032-06078-5_28.
    [24] ZHOU Wujie, XIAO Yuxiang, LIU Yuanyuan, et al. FIMKD: Feature-implicit mapping knowledge distillation for RGB-D indoor scene semantic segmentation[J]. IEEE Transactions on Artificial Intelligence, 2024, 5(12): 6488–6499. doi: 10.1109/TAI.2024.3452052.
    [25] GUO Xiaodong, ZHOU Wujie, and LIU Tong. Contrastive learning-based knowledge distillation for RGB-thermal urban scene semantic segmentation[J]. Knowledge-Based Systems, 2024, 292: 111588. doi: 10.1016/j.knosys.2024.111588.
    [26] XIE Enze, WANG Wenhai, YU Zhiding, et al. SegFormer: Simple and efficient design for semantic segmentation with transformers[C]. Proceedings of the 35th International Conference on Neural Information Processing Systems, 2021: 924. (查阅网上资料, 未找到本条文献会议地, 请确认).
    [27] SZEGEDY C, VANHOUCKE V, IOFFE S, et al. Rethinking the inception architecture for computer vision[C]. Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, Las Vegas, USA, 2016: 2818–2826. doi: 10.1109/cvpr.2016.308.
    [28] MILLETARI F, NAVAB N, and AHMADI S A. V-Net: Fully convolutional neural networks for volumetric medical image segmentation[C]. 2016 4th International Conference on 3D Vision, Stanford, USA, 2016: 565–571. doi: 10.1109/3dv.2016.79.
    [29] DENG Jia, DONG Wei, SOCHER R, et al. ImageNet: A large-scale hierarchical image database[C]. 2009 IEEE Conference on Computer Vision and Pattern Recognition, Miami, USA, 2009: 248–255. doi: 10.1109/cvpr.2009.5206848.
    [30] BOTTOU L. Large-scale machine learning with stochastic gradient descent[C]. Proceedings of 19th International Conference on Computational Statistics, Paris, France, 2010: 177–186. doi: 10.1007/978-3-7908-2604-3_16.
    [31] SHIN U, LEE K, KWEON I S, et al. Complementary random masking for RGB-thermal semantic segmentation[C]. 2024 IEEE International Conference on Robotics and Automation, Yokohama, Japan, 2024: 11110–11117. doi: 10.1109/icra57147.2024.10611200.
    [32] YIN Bowen, ZHANG Xuying, LI Zhongyu, et al. DFormer: Rethinking RGBD representation learning for semantic segmentation[C]. The 12th International Conference on Learning Representations, Vienna, Austria, 2024.
  • 加载中
图(7) / 表(13)
计量
  • 文章访问数:  10
  • HTML全文浏览量:  1
  • PDF下载量:  0
  • 被引次数: 0
出版历程
  • 修回日期:  2026-10-16
  • 录用日期:  2026-10-16
  • 网络出版日期:  2026-10-25

目录

    /

    返回文章
    返回