#图生成

高效边感知注意力网络：图生成模型的显存优化新方案

图生成模型常受限于图专用Transformer的显存和效率瓶颈。本文介绍一种高效边感知注意力网络，通过头级分数调制和轻量残差机制，借助PyTorch FlexAttention实现，在保持生成质量的同时将峰值GPU显存降低约65%，并显著提升训练与采样吞吐量。