AITOP
全部 AI 动态
AI 相关资讯全量信息流
全部博客资讯推文论文
全部模型产品行业论文技巧
标签：集合学习×
6月16日
20:46
6月12日
12:57
6月11日
15:28
15:23
15:07
6月4日
11:22
11:22arXiv cs.LG@Jose E. Escrig Molina, Baoquan Chen, Daniel Probst
精选
Graph Set Transformer (GST) 是一种专为图集（sets of graphs）学习设计的新型神经网络架构，解决了现有方法需先用GNN编码图嵌入、导致特征提取与集合上下文建模分离的瓶颈。GST在每一层交错进行节点级特征传播和图间上下文建模，并通过门控机制融合两类信息。在合成数据集和三个真实基准（原子反应中心识别、反应产率预测、图像分类）上，GST在相同参数量下优于DeepSets、SetTransformer等基线。消融实验表明，局部与集合上下文的交错融合是性能提升的关键。
论文图神经网络集合学习Graph Set Transformer架构创新图集
推荐理由：做图神经网络或集合学习的研究者，GST解决了图集任务中局部与全局信息割裂的痛点，代码已开源，值得复现对比。
原文