Feb, 2024

使用 BAM 进行图结构推断:引入双线性注意机制

TL;DR统计学和机器学习中,检测数据集中的依赖关系是一个核心挑战。我们提出了一种新颖的神经网络模型,用于监督式图结构学习,即学习观测数据与其底层依赖结构之间的映射关系。通过利用结构方程模型和使用随机生成的多元切比雪夫多项式来模拟训练数据,我们的方法在线性和各种类型的非线性依赖关系上都表现出稳健的可推广性。我们引入了一种新颖的双线性注意机制(BAM),用于显式处理依赖信息,其操作是基于转换数据的协方差矩阵水平,并尊重对称正定矩阵流形的几何性质。实证评估显示了我们的方法在检测广泛范围的依赖关系中的稳健性,在无向图估计方面表现出优势,并通过一种新颖的两步方法在完成部分有向无环图估计中取得竞争优势。