arXiv:2410.02224cs.CV2024-10被引 3

轻量级网络融合卷积与Transformer,实现实时语义分割

Efficient Semantic Segmentation via Lightweight Multiple-Information Interaction Network

  • 设计轻量特征交互模块,高效融合局部与全局信息
  • 仅0.72M参数、11.74G FLOPs下达到72.0% mIoU
  • 适合部署在资源受限设备的实时分割场景

近年来,将卷积神经网络(CNN)的局部建模能力与Transformer的全局依赖捕捉优势结合,成为语义分割领域的热点。然而,高昂的计算开销和内存需求仍是其在实时场景中应用的主要障碍。本文提出轻量级多信息交互网络(LMIINet),在保持CNN与Transformer优势的同时,显著降低冗余计算与内存占用。其核心是轻量特征交互瓶颈(LFIB)模块,采用高效卷积增强上下文融合;同时改进扁平化Transformer,强化局部与全局特征交互以捕获细粒度语义。在LFIB与Transformer块中引入组合系数学习机制,进一步优化特征交互。大量实验表明,LMIINet在精度与效率间取得良好平衡:在Cityscapes测试集上,仅需0.72M参数和11.74G FLOPs,即可实现72.0% mIoU与100 FPS;在CamVid数据集上,达69.94% mIoU与160 FPS,均基于单张RTX2080Ti GPU。

原文摘要 · Abstract (English)

Recently, integrating the local modeling capabilities of Convolutional Neural Networks (CNNs) with the global dependency strengths of Transformers has created a sensation in the semantic segmentation community. However, substantial computational workloads and high hardware memory demands remain major obstacles to their further application in real-time scenarios. In this work, we propose a Lightweight Multiple-Information Interaction Network (LMIINet) for real-time semantic segmentation, which effectively combines CNNs and Transformers while reducing redundant computations and memory footprints. It features Lightweight Feature Interaction Bottleneck (LFIB) modules comprising efficient convolutions that enhance context integration. Additionally, improvements are made to the Flatten Transformer by enhancing local and global feature interaction to capture detailed semantic information. Incorporating a combination coefficient learning scheme in both LFIB and Transformer blocks facilitates improved feature interaction. Extensive experiments demonstrate that LMIINet excels in balancing accuracy and efficiency. With only 0.72M parameters and 11.74G FLOPs (Floating Point Operations Per Second), LMIINet achieves 72.0\% mIoU at 100 FPS (Frames Per Second) on the Cityscapes test set and 69.94\% mIoU (mean Intersection over Union) at 160 FPS on the CamVid test dataset using a single RTX2080Ti GPU.

语义分割轻量网络Transformer实时推理

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。