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