Sep, 2023

稀疏神经网络的硬件限制:稀疏屋顶线

TL;DR我们引入了稀疏屋顶线,一种用于评估神经网络中稀疏性能的视觉性能模型。该模型同时考虑了网络准确性、稀疏性和预测的推理加速度。通过使用一种新颖的分析模型预测稀疏网络性能,并通过在多个真实世界计算机视觉架构上对稀疏模式和程度进行验证,我们验证了预测的推理加速度。我们通过两个案例研究展示了我们模型的实用性和易用性:(1) 我们展示了机器学习研究人员如何预测未实现或未优化的块结构稀疏模式的性能,(2) 我们展示了硬件设计人员如何预测硬件中新的稀疏模式和稀疏数据格式的性能影响。在这两种情况下,稀疏屋顶线帮助性能专家确定具有最高性能潜力的稀疏区域。