Jul, 2020

使用进化图强化学习优化内存分配

TL;DR本文提出了一种基于进化图强化学习(EGRL)的方法,用于深度神经网络加速器的自动内存映射,该方法结合了图神经网络、强化学习和进化搜索等技术,并在 Intel NNP-I 芯片上进行了训练和验证,实验结果表明,EGRL 在 BERT、ResNet-101 和 ResNet-50 三种工作负载中表现良好,并相对于原生 NNP-I 编译器获得 28-78% 的加速。