ICMLJul, 2022
在集成重溢和分页的微型设备上训练神经网络
POET: Training Neural Networks on Tiny Devices with Integrated Rematerialization and Paging
Shishir G. Patil, Paras Jain, Prabal Dutta, Ion Stoica, Joseph E. Gonzalez
TL;DRPOET 算法可以实现在边缘设备上训练大型神经网络,同时减少能耗并不修改反向传播的数学正确性。在嵌入式设备的内存限制下,使用 POET 可以比当前边缘训练方法更节能地对 ResNet-18 和 BERT 进行微调。