全球速递！图表示学习新方法实现因果性的有效提升

(资料图片)

近日，中国科学院软件研究所天基综合信息系统重点实验室研究团队在图表示学习方法研究中取得进展。成果论文“面向跨粒度小样本学习的元注意力生成网络”被人工智能领域顶级学术会议AAAI接收并作为口头报告。论文首次提出一种新的图表示学习方法：鲁棒因果图表示学习（RCGRL）。该方法通过引入工具变量来消除图表示学习中的混杂因子，实现了图表示学习因果性的有效提升。

图表示学习旨在学习复杂图结构数据的特征表示，相比于图像或序列化数据的表示学习，图表示学习应用场景中的数据更加不规则，语义信息更加丰富，且常常会存在大量与任务无关的混杂因子，对任务进行干扰。如何去除这些混杂因子，只保留与任务相关的因果信息，是该研究团队希望解决的问题。

在这项研究中，团队遵循统计因果学的规律，提出了先基于神经网络自主生成的工具变量进行混杂因子消除，然后使用处理过的因果数据对模型进行训练，进而提升模型因果性的方法。团队还通过特定的损失函数对生成工具变量的神经网络以及进行表示学习的神经网络进行交替训练。训练后的模型能够细粒度的消除混杂因子对图表示提取的影响，并最大限度的保留因果信息。

研究团队将RCGRL方法与几种广泛使用的因果增强图表示学习方法相对比，实验结果表明RCGRL更具有优越性，在多种图表示学习任务精度上明显优于其它方法，有效提升了图表示学习的因果性。

相关论文信息：https://link.springer.com/article/10.1007/s11263-023-01760-7