Apr, 2023

X-RLflow:神经网络子图转换的图形强化学习

TL;DR本文提出使用强化学习中一种无模型的代理,即基于图神经网络编码的目标计算图,并迭代输出一个经过转换的计算图的方式,来学习执行神经网络数据流图重写。我们展示了我们的方法在一系列深度学习模型上超越了现有的超优化系统,并在基于转换器式架构的模型上实现了高达40%的性能提升。