一种基于剪枝的图神经网络去相关性微调方法

AITNT
正文
推荐专利
一种基于剪枝的图神经网络去相关性微调方法
申请号:CN202411697508
申请日期:2024-11-26
公开号:CN119740603A
公开日期:2025-04-01
类型:发明专利
摘要
本公开提供一种基于剪枝的图神经网络去相关性微调方法。包括两个部分:模型精简模块和模型去相关模块;所述精简模块为通过随机移除冗余神经元来减少自监督预训练GNN中的模型冗余,以获取精简的图神经网络;所述模型去相关模块应用于微调阶段,学习用于分类损失的去相关权重,通过最小化重加权的部分交叉协方差矩阵调整fs和R的权重,鼓励节点嵌入在不同维度上的独立性,以减少嵌入表示间的冗余并使其更具信息量;对于图数据,将其输入到精简的图神经网络中以获得节点的嵌入表示,结合额外的预测层R预测每个节点的标签。
技术关键词
协方差矩阵 微调方法 冗余 预训练方法 模块 傅立叶 数据 标签 节点数 阶段 定义 变量
添加客服微信openai178,进AITNT官方交流群
驱动智慧未来:提供一站式AI转型解决方案
沪ICP备2023015588号