arrow
返回

RTLinearFormer: Semantic segmentation with lightweight linear attentions

delete2025-04-01
delete0
PRE
AI
Y
Yuhang Gu
C
Chong Fu *
W
Wei Song
X
Xingwei Wang
陈
陈俊鑫 (Junxin Chen)
DOI:10.1016/j.neucom.2025.129489delete
delete原文链接
delete分享
delete收藏
摘要

摘要

En 中文
Real-time semantic segmentation aims to balance networks' segmentation accuracy and inference speed. Recently, dual-resolution network shows its efficiency and effectiveness in this field. However, due to selfattention's O(N2) computational complexity, pure CNN-based networks still dominate in this architecture. In this paper, we propose RTLinearFormer, an efficient dual-resolution network with Transformer specially for real-time semantic segmentation, which achieves trade-off between segmentation accuracy and inference speed. To this end, our RTLinearFormer leverages lightweight channel-wise multi-scale attention with linear complexity, channel feature focusing and multi-scale features to extract global contextual information in low- resolution branch and uses cross-resolution ReLU-based linear attention with linear complexity to gather global contextual information derived from low-resolution branch with detailed information in high-resolution branch. Extensive experiments demonstrate the effectiveness and efficiency of our proposed RTLinearFormer in realtime semantic segmentation task. The model achieves 78.41% val mIoU and 66.7 FPS on Cityscapes using a single RTX 3090 GPU. When pretrained on Cityscapes, it achieves 77.4% test mIoU and 141.3 FPS on CamVid. It shows promising results on Cityscapes and CamVid, demonstrating superior performance over other stateof-the-art models. To further validate its generalization ability, we also test it on the ADE20K and COCOStuff datasets. It achieves 35.70% val mIoU and 107.3 FPS on ADE20K, and 31.15% test mIoU and 150.2 FPS on COCOStuff, setting new benchmarks for state-of-the-art performance.
Keyword:
Dual-resolution network
Lightweight channel-wise multi-scale attention
Cross-resolution ReLU-based linear attention
Real-time semantic segmentation

期刊

Neurocomputing 封面图
Neurocomputing
IF:
6.5
论文数:
2.5W
被引数:
6.5W

机构

D
Dalian University of Technology
学者数:
6.0W
论文数: 4.4W
被引数: 5.5W
N
northeastern university - china
学者数:
3.2W
论文数: 2.7W
被引数: 37
引用论文

引用论文

A Pilot Study of the Efficacy of the Unified Protocol for Transdiagnostic Treatment of Emotional Disorders in Treating Posttraumatic Psychopathology: A Randomized Controlled Trial
err2021-01-16
err0
errOAAI
errMeaghan L. O'Donnell; Winnie Lau; Katherine Chisholm; James Agathos; Jonathon Little; Sonia Terhaag; Rachel Brand; Andrea Putica; Alexander C. N. Holmes; Lynda Katona; Kim L. Felmingham; Kim Murray; Fardous Hosseiny; Matthew W. Gallagher
err分享
err收藏
err分享
err收藏
学者 查看更多内容