用RISC-V向量扩展加速嵌入式CNN推理,提升预处理和推理效率。
Flexible Vector Integration in Embedded RISC-V SoCs for End to End CNN Inference Acceleration
- 利用RISC-V Vector 1.0扩展,在缓存架构中灵活集成向量指令
- 图像预处理速度提升9倍,YOLOv3后层执行快3倍于CPU
- 适合资源受限的嵌入式AI系统,降低功耗且支持现代深度学习数据流
异构与领域专用架构在资源受限的嵌入式平台部署现代CNN方面展现出巨大潜力。定制硬件专注于计算最密集的CNN部分,如DLA(深度学习加速器)和NPU(神经网络处理器),可突破传统硅片缩放极限,缓解嵌入式SoC中的能效/性能权衡问题。高效使用这类专用加速器需合理的系统集成及编译/执行模型以实现异构架构中的均衡执行。本文聚焦硬件集成挑战,分析加速器在内存层次结构中的放置及与其他执行模块的邻近性。实验验证了运行时CNN执行及前后处理环节的性能瓶颈,而此前研究多仅关注加速器加速。本文借助RISC-V Vector 1.0扩展,展示其在适配缓存架构下的灵活性,有效缓解预处理瓶颈和CPU回退问题。结果表明,图像预处理最高提速9倍,YOLOv3回退层执行较CPU快3倍。实验还证明,基于RVV-1.0的编程模型可在加速器丰富的嵌入式SoC上实现计算与内存开销的平衡,同时比传统并行执行平台更省电。
原文摘要 · Abstract (English)
The emergence of heterogeneity and domain-specific architectures targeting deep learning inference show great potential for enabling the deployment of modern CNNs on resource-constrained embedded platforms. A significant development is the diversification of custom hardware solely targeting the most expensive parts of CNNs. DLAs (deep learning accelerators) and NPUs (neural processing units), among others, can overcome the approaching limits of traditional silicon scaling and provide a solution to the power/performance tradeoff within embedded SoCs. Efficient DSA utilization requires proper system integration and a compilation/execution model for balanced execution in these heterogeneous architectures. There is a critical need for proper system integration and an efficient compilation/execution model for balanced execution in these heterogeneous architectures. This work highlights the hardware integration challenges for efficiently placing these units within the memory hierarchy and correct proximity to other execution blocks. We experimentally verify performance bottlenecks in CNN execution and pre/post-processing at runtime, where previous attention has generally been given to accelerator speedup alone. This work takes advantage of the ratification of the RISC-V Vector 1.0 extension and demonstrates its potential as a flexible target within a well-suited cache hierarchy scheme to reduce pre-processing bottlenecks and CPU fallback processes. Our results show up to a 9x speedup of image pre-processing and YOLOv3 fallback layer execution by up to 3x compared to CPU. We demonstrate RVV-1.0 in exposing a flexible programming model that can enable a balanced computation and memory footprint on accelerator-rich embedded SoCs supporting modern deep-learning dataflows while consuming less power than traditional parallel execution platforms.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。