跳到正文
原文
arXiv cs.LG· Samantha Chen, Jesse He, Coleman Clougherty, Gal Mishne, Chester Holtz·· 27 天前AI 评分34

面向图鞍点问题的神经算法推理框架 GraphPDHG

Neural Algorithmic Reasoning for Graph Saddle Point Problems

AI 导读

研究人员提出基于 Chambolle-Pock 原始-对偶混合梯度(PDHG)算法的消息传递框架 GraphPDHG,用于求解通用图鞍点问题。理论证明该框架能有效模拟并学习加速 PDHG 算法,并在实验中成功作为二阶优化技术 SSNAL 的热启动。此外,与 PDHG 对齐的设计使其相比未对齐的 GNN 基线具备更强的尺寸泛化能力。

来源:arXiv cs.LG · arxiv.org