• Sparse Edge PE: 一种基于FPGA的稀疏数据友好的CNN处理引擎设计

    Sparse Edge PE: An FPGA-based sparse activation friendly CNN PE design

    • 针对现有卷积神经网络加速器设计中对于输入激活固有稀疏性利用不足的问题,提出了一种基于FPGA的卷积神经网络加速器处理引擎(Processing Element,PE)设计方案。首先,通过硬件实现的Sobel边缘检测算法对输入图像进行预处理,有效提取边缘关键特征,显著提升了输入激活的稀疏度;其次,设计了改进的稀疏激活跳过机制,充分利用输入激活的稀疏特性,在提升系统吞吐量的同时显著降低了能耗;最后,提出了一种多bank一写多读混合乒乓读写机制,优化了卷积操作中输入激活的存储方案,实现了面积效率与吞吐量的最佳平衡。为验证方案的有效性,采用PyTorch框架训练ResNet卷积神经网络,在CIFAR-10和CIFAR-100数据集上进行了系统测试。实验结果表明,边缘检测预处理不会对网络分类精度产生显著影响。性能评估显示:与最新的工作相比,该设计能效比提升2.18倍,单位面积吞吐率提升84%,单位面积功耗降低42.9%,在计算效率和能耗效率方面均展现出显著优势。本研究为高性能低功耗神经网络加速器设计提供了新的技术路径,具有广阔的应用前景。

       

      Abstract: To address the issue of insufficient utilization of inherent sparsity in input activations within existing convolutional neural network (CNN) accelerator designs, this paper proposes a field-programmable gate array (FPGA)-based Processing Element (PE) design scheme for CNN accelerators. First, a hardware-implemented Sobel edge detection algorithm is employed to preprocess input images, effectively extracting key edge features and significantly enhancing the sparsity of input activations. Second, an improved sparse activation skipping mechanism is designed to fully exploit the sparsity characteristics of input activations, thereby improving system throughput while substantially reducing energy consumption. Furthermore, a multi-bank hybrid ping-pong read/write mechanism with one-write-multiple-read capability is proposed to optimize the storage scheme for input activations during convolution operations, achieving an optimal balance between area efficiency and throughput. To validate the effectiveness of the proposed scheme, a ResNet CNN is trained using the PyTorch framework and systematically evaluated on the CIFAR-10 and CIFAR-100 datasets. Experimental results demonstrate that the edge detection preprocessing does not significantly affect network classification accuracy. Performance evaluation reveals that, compared with the state-of-the-art designs, the proposed architecture achieves a 2.18× improvement in energy efficiency, an 84% increase in throughput per unit area, and a 42.9% reduction in power consumption per unit area, demonstrating significant advantages in both computational and energy efficiency. This study provides a novel technical pathway for the design of high-performance and low-power neural network accelerators, with broad application prospects.

       

    /

    返回文章
    返回