arXiv cs.LG· Samantha Chen, Jesse He, Coleman Clougherty, Gal Mishne, Chester Holtz·· 10 小时前AI 评分18
GraphPDHG:面向图鞍点问题的神经算法推理框架
Neural Algorithmic Reasoning for Graph Saddle Point Problems
AI 导读
研究者提出基于 Chambolle-Pock 原始-对偶混合梯度(PDHG)方法的消息传递框架 GraphPDHG,用于求解一般图鞍点问题。理论上该网络能通过模拟 PDHG 高效求解一类图鞍点问题,并可学习加速版 PDHG 算法;实验表明其作为二阶优化方法 SSNAL 的学习式热启动表现良好,且与 PDHG 对齐后比未对齐的 GNN 基线具有更强的规模泛化能力。
来源:arXiv cs.LG · arxiv.org