研究人员开发了一种新的等变图神经网络算法，称为 Sobek，它通过消除物化特定于边的权重和消息的需要，显著提高了效率。这种流式方法将内存使用量减少了高达 99%，并将各种微分阶数和算子族的运行速度提高了高达 49.7 倍。Sobek 使得以前因内存限制而无法执行的工作负载得以运行，其性能优于 OpenEquivariance 等现有实现。 AI
影响 这种新算法可能能够训练和部署更大、更复杂的图神经网络模型，从而可能加速药物发现和材料科学等领域的研究。
排序理由 该集群包含一篇 arXiv 预印本，详细介绍了一种新的图神经网络方法。[lever_c_demoted from research: ic=1 ai=1.0]
- arXiv
- CUDA
- Equivariant Graph Neural Networks
- graphics processing unit
- Hugging Face
- OpenEquivariance
- Sobek
AI 生成摘要 · Google Gemini · 来自 1 个来源。 我们如何撰写摘要 →
