arXiv · Machine Learning Theory· Samantha Chen, Jesse He, Coleman Clougherty, Gal Mishne, Chester Holtz·· 3 小时前AI 评分22
GraphPDHG:用神经算法推理求解图鞍点问题
Neural Algorithmic Reasoning for Graph Saddle Point Problems
AI 导读
研究者提出名为 GraphPDHG 的消息传递框架,基于 Chambolle-Pock 原始对偶混合梯度(PDHG)方法求解一般图鞍点问题。理论上该框架可通过模拟 PDHG 高效求解一类图鞍点问题,并能学习加速版 PDHG 算法。实验上以学习到的 warm start 结合二阶优化技术 SSNAL 验证性能,且与 PDHG 对齐比未对齐 GNN 基线具备更强规模泛化能力。
来源:arXiv · Machine Learning Theory · arxiv.org