arXivDaily arXiv每日学术速递 周一至周五更新
北美退伍军人节(Veterans Day),12 日无更新
arXiv 2403.02308cs.CV
收藏

Vision-RWKV:基于RWKV类架构的高效可扩展视觉感知

Vision-RWKV: Efficient and Scalable Visual Perception with RWKV-Like Architectures

The Chinese University of Hong Kong(香港中文大学);Shanghai AI Laboratory(上海人工智能实验室);Fudan University(复旦大学);Nanjing University(南京大学);Tsinghua University(清华大学);SenseTime Research(商汤科技研究院)

Yuchen Duan, Weiyun Wang, Zhe Chen, Xizhou Zhu, Lewei Lu, Tong Lu, Yu Qiao, Hongsheng Li, Jifeng Dai, Wenhai Wang

AI总结:

本文提出Vision-RWKV,一种基于RWKV架构的高效视觉模型,通过降低空间聚合复杂度,实现高分辨率图像处理,在图像分类和密集预测任务上超越ViT,具有更快的速度和更低的内存占用。

摘要:

Transformer已经彻底改变了计算机视觉和自然语言处理领域,但其高计算复杂度限制了其在高清图像处理和长上下文分析中的应用。本文介绍了Vision-RWKV(VRWKV),这是一个从NLP领域使用的RWKV模型改编而来,并针对视觉任务进行了必要修改的模型。与视觉Transformer(ViT)类似,我们的模型旨在高效处理稀疏输入,展现出强大的全局处理能力,同时也能有效扩展,适应大规模参数和广泛的数据集。其独特优势在于降低了空间聚合复杂度,使其能够无缝处理高分辨率图像,无需进行窗口化操作。我们的评估表明,VRWKV在图像分类任务上超越了ViT的性能,并且在处理高分辨率输入时速度显著更快、内存占用更低。在密集预测任务中,它优于基于窗口的模型,同时保持了相当的速度。这些结果凸显了VRWKV作为视觉感知任务更高效替代方案的潜力。代码已发布在https://github.com/OpenGVLab/Vision-RWKV。

英文摘要:

Transformers have revolutionized computer vision and natural language processing, but their high computational complexity limits their application in high-resolution image processing and long-context analysis. This paper introduces Vision-RWKV (VRWKV), a model adapted from the RWKV model used in the NLP field with necessary modifications for vision tasks. Similar to the Vision Transformer (ViT), our model is designed to efficiently handle sparse inputs and demonstrate robust global processing capabilities, while also scaling up effectively, accommodating both large-scale parameters and extensive datasets. Its distinctive advantage lies in its reduced spatial aggregation complexity, which renders it exceptionally adept at processing high-resolution images seamlessly, eliminating the necessity for windowing operations. Our evaluations demonstrate that VRWKV surpasses ViT's performance in image classification and has significantly faster speeds and lower memory usage processing high-resolution inputs. In dense prediction tasks, it outperforms window-based models, maintaining comparable speeds. These results highlight VRWKV's potential as a more efficient alternative for visual perception tasks. Code is released at https://github.com/OpenGVLab/Vision-RWKV.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑