Jul, 2020
使用进化图强化学习优化内存分配
Optimizing Memory Placement using Evolutionary Graph Reinforcement Learning
Shauharda Khadka, Estelle Aflalo, Mattias Marder, Avrech Ben-David, Santiago Miret...
TL;DR本文提出了一种基于进化图强化学习(EGRL)的方法,用于深度神经网络加速器的自动内存映射,该方法结合了图神经网络、强化学习和进化搜索等技术,并在 Intel NNP-I 芯片上进行了训练和验证,实验结果表明,EGRL 在 BERT、ResNet-101 和 ResNet-50 三种工作负载中表现良好,并相对于原生 NNP-I 编译器获得 28-78% 的加速。