计算机科学 > 机器学习
[提交于 2024年12月31日
]
标题: 将图解释应用于操作符融合
标题: Applying Graph Explanation to Operator Fusion
摘要: 层融合技术对于提高深度神经网络(DNN)的推理效率至关重要,以便于部署。 融合旨在通过减少加速器的片上缓冲区和DRAM之间的数据传输来降低推理成本。 这是通过将多个操作如卷积和激活一起分组执行到单个执行单元——融合组中来实现的。 然而,片上缓冲区容量限制了融合组的大小,优化整个DNN的融合需要将其划分为多个融合组。 找到最佳组是一个复杂的问题,其中无效解的存在阻碍了传统搜索算法,并要求稳健的方法。 在本文中,我们将可解释的人工智能,特别是图解释技术(GET),引入到层融合中。 给定一个无效的融合组,我们确定对组无效性最负责的操作,然后利用这些知识通过一种贪心树-based算法递归地分割原始融合组,以最小化DRAM访问。 我们将我们的方案与常见算法结合,并在两种类型的层融合上优化DNN:行缓冲区深度优先(LBDF)和分支需求减少(BRR)。 实验表明,我们的方案在几个流行的和经典的卷积神经网络如ResNets和MobileNets上有效。 我们的方案在EfficientNet-B3上实现了超过20%的DRAM访问减少。
文献和引用工具
与本文相关的代码,数据和媒体
alphaXiv (什么是 alphaXiv?)
CatalyzeX 代码查找器 (什么是 CatalyzeX?)
DagsHub (什么是 DagsHub?)
Gotit.pub (什么是 GotitPub?)
Hugging Face (什么是 Huggingface?)
带有代码的论文 (什么是带有代码的论文?)
ScienceCast (什么是 ScienceCast?)
演示
推荐器和搜索工具
arXivLabs:与社区合作伙伴的实验项目
arXivLabs 是一个框架,允许合作伙伴直接在我们的网站上开发和分享新的 arXiv 特性。
与 arXivLabs 合作的个人和组织都接受了我们的价值观,即开放、社区、卓越和用户数据隐私。arXiv 承诺这些价值观,并且只与遵守这些价值观的合作伙伴合作。
有一个为 arXiv 社区增加价值的项目想法吗? 了解更多关于 arXivLabs 的信息.