T_AI 115.3-2024 信息技术 神经网络表示与模型压缩 第3部分：图神经网络.docx

ICS35.040CCSL71

团体标准

T/AI115.3—2024

信息技术神经网络表示与模型压缩

第3部分：图神经网络

Informationtechnology-Neuralnetworkrepresentationandmodelcompression—

Part3:Graphneuralnetwork

2024-12-30发布2024-12-30实施

中关村视听产业技术创新联盟发布

T/AI115.3—2024

I

目次

前言 II

引言 III

1范围 1

2规范性引用文件 1

3术语和定义 1

4缩略语 3

5图神经网络表示与模型压缩概述 5

6图数据表示 6

6.1基本定义 6

6.2图数据类型定义 9

6.3图基本运算 10

6.4图基本任务 24

7图神经网络模型 25

7.1模型结构 25

7.2基础算子 28

7.3点级模型 76

7.4边级模型 97

7.5图级模型 103

8图神经网络压缩 108

8.1图数据压缩 108

8.2模型量化与剪枝 117

8.3模型蒸馏 119

8.4模型加速 124

9图神经网络计算框架 129

9.1基于深度学习平台的图神经网络计算框架 129

9.2图神经网络计算框架与第三方数据源接口 138

T/AI115.3-2024

II

前言

本文件按照GB/T1.1—2020《标准化工作导则第1部分：标准化文件的结构和起草规则》的规定起草。

本文件是T/AI115《信息技术神经网络表示与模型压缩》的第2部分。T/AI115已经发布了以下部分：

——第1部分：卷积神经网络；

——第3部分：图神经网络。

本文件由新一代人工智能产业技术创新战略联盟AI标准工作组提出。

本文件由中关村视听产业技术创新联盟归口。

本文件起草单位：北京大学、北京邮电大学、华为技术有限公司、蚂蚁金服（杭州）网络技术有限公司、北京百度网讯科技有限公司、鹏城实验室、吉林大学、西安交通大学、上海图灵天问智能科技有限公司、中国电子技术标准化研究院、阿里巴巴（中国）有限公司、北京海致星图科技有限公司、北京图数创智科技有限公司、杭州海康威视数字技术股份有限公司、上海上湖信息技术有限公司、中国科学院自动化研究所、中科南京人工智能创新研究院、中移动信息技术有限公司、东北大学、深圳市腾讯计算机系统有限公司、北京交通大学、清华大学、中国联合网络通信有限公司、腾讯科技(深圳)有限公司。

本文件主要起草人：石川、田永鸿、黄海、任宇翔、张志强、冯仕堃、陈光耀、杨博、裴红斌、刘忠雨、范深、沈游人、杨成、岳大帅、刘曜齐、耿赛赛、陈均泽、郝健翔、杨晋豫、文思源、张学仓、崔佳旭、王平辉、李慧园、陈泰乐、曾菊香、何径舟、黄正杰、黄世维、方泽阳、陈泽裕、于佃海、李伟彬、周俊、胡斌斌、韦绍玮、梁磊、敬斌、张翀、倪铭坚、彭佩玺、张文涛、盛则昂、黎洋、沈彧、杨智、崔斌、孙铭蔚、王鹏、刘洪、杨帆、邹磊、吴伟、周光煜、方超、陈诚、姜伟浩、王扬、王春平、陈磊、张一帆、赵士云、程健、陈卓、尚晶、李珍、张岩峰、蒋杰、陶阳宇、欧阳文、孙瑞鸿、耿阳李敖、李清勇、王鑫、朱文武、胡浩基、王化良、胡欣怡、杨雨泽、鲍薇、郑若琳、沈芷月、杨斌、程新洲、张伟民、赵海英、黄铁军、高文。

T/AI115.3—2024

III

引言

图神经网络已经在推荐系统、知识图谱、智能交通等领域得到广泛应用。基于图神经网络的深度学习算法及其衍生应用系统在许多图任务上取得了突破性的进展，但是其具体的实现仍然依赖于不同的算法平台。由于目前没有统一的图神经网络的表示标准，不同的算法平台采用了不同的图神经网络的表示和存储标准；不同图神经网络框架之间不能互操作与协同工作；图数据和图神经网络模型不能在不同平台上直接迁移使用；不同图神经网络框架对于图神经网络粒度定义不同，妨碍了硬件厂商对于图神经网络的加速和优化；对于新出现的算子，框架都需要进行重定义。本文件旨在提供图数据和图