一种基于信息熵的图神经网络自适应丢弃方法及系统

AITNT
正文
推荐专利
一种基于信息熵的图神经网络自适应丢弃方法及系统
申请号:CN202510819465
申请日期:2025-06-19
公开号:CN120337986B
公开日期:2025-08-19
类型:发明专利
摘要
本发明公开了一种基于信息熵的图神经网络自适应丢弃方法及系统,涉及图神经网络随机丢弃策略技术领域,包括:接收节点的特征向量,对节点的特征向量进行Softmax归一化处理,并基于信息熵来计算节点对应的熵值,对节点对应的熵值进行归一化,并与预设的全局最大丢弃率相乘,得到节点的个性化丢弃概率;通过PyTorch Geometric框架中边索引的机制对节点的个性化丢弃概率进行映射,得到映射边,基于映射边进行采样,得到伯努利掩码;基于伯努利掩码和节点的个性化丢弃概率进行缩放,得到扰动后的消息矩阵,将扰动后的消息矩阵输入至预先建立的图神经网络模型内,从而实现图神经网络的自适应丢弃。
技术关键词
信息熵 丢弃方法 消息 神经网络模型 节点特征 矩阵 Softmax函数 邻居 索引 源节点 处理器 采样模块 存储器 机制 代表 框架 超参数 终端设备 元素
添加客服微信openai178,进AITNT官方交流群
驱动智慧未来:提供一站式AI转型解决方案
沪ICP备2023015588号