基于知识蒸馏的面向多元时序预测图神经网络压缩方法
摘要:
本发明公开了基于知识蒸馏的面向多元时序预测图神经网络压缩方法,包括处理基于图神经网络的多元时序预测模型的时间卷积层,将膨胀的扩展卷积变成固定的扩展卷积;创建并训练处一个效果好的深层网络;构建一个浅层的神经网络;对深层网络与浅层网络进行协同训练,得到压缩后的浅层网络。本发明首先实现了对基于图神经网络的多元时序预测模型的压缩,在边缘计算场景中,减少模型的计算资源与内存资源需求,使得模型能够顺利部署,提高物联网设备的管理效率。其次,在压缩目标模型的过程中,保证压缩不会使得模型的精确度损失过大,保证压缩后模型的预测效果。
0/0