基于对抗因果图学习的可解释性节点分类预测方法

AITNT-国内领先的一站式人工智能新闻资讯网站
# 热门搜索 #
基于对抗因果图学习的可解释性节点分类预测方法
申请号:CN202511025215
申请日期:2025-07-24
公开号:CN120524163B
公开日期:2025-10-14
类型:发明专利
摘要
本申请提供了基于对抗因果图学习的可解释性节点分类预测方法,包括:搭建的预测模型包括冗余过滤模块和对抗因果图学习模块;冗余过滤模块和对抗因果图学习模块实现图信息瓶颈机制;冗余过滤模块采用两层图注意力网络GAT结构进行信息聚合,得到节点的嵌入;对抗因果图学习模块采用基于注意力机制的可学习子图采样器对节点的嵌入生成因果解释子图,基于因果增强机制的PGD对抗训练策略,对解释子图嵌入进行梯度扰动优化,生成对抗嵌入,通过多轮扰动迭代获得最终扰动解释子图嵌入;通过多目标损失联合优化,进行端到端的预测模型训练;训练完成后,将节点的嵌入输入至分类器输出预测结果。本申请显著提升了模型的结构透明度和可解释性。
技术关键词
分类预测方法 节点 过滤模块 联合损失函数 预测模型训练 邻居 采样器 冗余 分类器联合训练 网络 标签 多头注意力机制 邻域 矩阵 多层感知机 数据