双向GNN能量预测模型

误差低至2.6%

用图神经网络直接学习神经网络结构,比线性方法更准、更通用。

郭朝鹏 · Zhao-Jin, Zhong · Zexin, Zhang · 宋杰

Digital Communications and Networks 2024

参数信息

平均绝对百分比误差
2.60 %
平均绝对百分比误差
4.83 %

技术优势

预测更准

由于双向GNN能同时考虑每个层的输入和输出连接,捕捉到线性方法忽略的结构依赖。

泛化更强

在随机生成的数据集上训练,不依赖特定网络集合,因此对未见过的体系结构也能保持低误差。

误差4.83%

在近年10个典型CNN和7个NAS设计的CNN上评估,展示了跨架构的实用精度。

开源可用

代码已公开,研究者可直接复现并用于自己的设计流程。

应用场景