arXiv:2410.11650cs.CVcs.AI2024-10被引 17

将Vision Transformer拆分到多设备运行,显著降低延迟与模型大小。

Efficient Partitioning Vision Transformer on Edge Devices for Distributed Inference

  • 将ViT模型按类别拆分为子模型,分担计算任务。
  • 在边缘设备上实现最高28.9倍的模型压缩和34.1倍的延迟降低。
  • 适合资源受限场景下高效部署复杂视觉模型的开发者。

深度学习模型正被越来越多地应用于资源受限的边缘设备以实现实时数据分析。近年来,Vision Transformer及其变体在多种计算机视觉任务中表现出色。然而,其高计算需求与低推理延迟要求给边缘设备部署带来挑战。为此,我们提出ED-ViT框架,可高效地将复杂Vision Transformer模型分割并在多个边缘设备上执行。方法将模型按类别划分为多个子模型,每个负责特定数据类别的处理。为进一步减少计算开销和推理延迟,引入基于类别的剪枝技术,缩小各子模型规模。在五个数据集上使用三种模型架构,并在真实边缘设备上实现验证,结果表明该方法显著降低边缘设备上的推理延迟,模型大小最大压缩28.9倍,同时保持与原始ViT相当的测试准确率。此外,与两种先进方法(部署CNN和SNN模型)对比,评估了准确率、推理时间及模型整体大小。全面评估证明了ED-ViT框架的有效性。

原文摘要 · Abstract (English)

Deep learning models are increasingly utilized on resource-constrained edge devices for real-time data analytics. Recently, Vision Transformer and their variants have shown exceptional performance in various computer vision tasks. However, their substantial computational requirements and low inference latency create significant challenges for deploying such models on resource-constrained edge devices. To address this issue, we propose a novel framework, ED-ViT, which is designed to efficiently split and execute complex Vision Transformers across multiple edge devices. Our approach involves partitioning Vision Transformer models into several sub-models, while each dedicated to handling a specific subset of data classes. To further reduce computational overhead and inference latency, we introduce a class-wise pruning technique that decreases the size of each sub-model. Through extensive experiments conducted on five datasets using three model architectures and actual implementation on edge devices, we demonstrate that our method significantly cuts down inference latency on edge devices and achieves a reduction in model size by up to 28.9 times and 34.1 times, respectively, while maintaining test accuracy comparable to the original Vision Transformer. Additionally, we compare ED-ViT with two state-of-the-art methods that deploy CNN and SNN models on edge devices, evaluating metrics such as accuracy, inference time, and overall model size. Our comprehensive evaluation underscores the effectiveness of the proposed ED-ViT framework.

边缘计算视觉Transformer模型压缩分布式推理

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