arXiv cs.AI / cs.LG 最新· Vincent Jung·· 14 小时前AI 评分44
面向图生成的高效边感知注意力网络
Efficient Edge-aware Attention Network for Graph Generation
AI 导读
一项图生成研究提出高效边感知注意力层，通过头级分数调制注入结构信息，并用轻量残差机制保留依赖注意力分数的边更新。该层可基于 PyTorch 的 FlexAttention 实现，在生成质量与基线相近的同时将峰值 GPU 显存降低约 65%，并显著提升训练与采样吞吐。论文被 ICML 2026 的图基础模型研讨会与结构化概率推断及生成建模研讨会接收。
标题与摘要由 AI 辅助生成，事实与观点以原文为准。
来源：arXiv cs.AI / cs.LG 最新 · arxiv.org
本站标题、摘要与推荐理由由 AI 辅助生成；原文版权归来源所有。