arXiv:2501.03228v3 宣告类型: replace-cross 摘要：图神经网络（GNNs）通过其进行高阶表示平滑的能力，在协同推荐任务中展现了卓越的性能，有效捕捉了用户交互模式中的结构信息。然而，现有的GNN范式在处理大规模、嘈杂和现实世界数据集时面临着可扩展性和鲁棒性的巨大挑战。为了解决这些挑战，我们提出了LightGNN，这是一个轻量级且基于蒸馏的GNN剪枝框架，旨在大幅降低模型复杂性同时保持核心的合作建模能力。LightGNN框架引入了一个计算效率高的剪枝模块，该模块能够自适应地识别和移除冗余边和嵌入项，以实现模型压缩。该框架由一个耗资源友好的层次知识蒸馏目标引导，其中间层通过增强观察到的图来维持性能，特别是在高压缩率场景中。在公共数据集上的广泛实验表明，LightGNN具有显著的效果，不仅提高了计算效率，还提高了推荐准确性。值得注意的是，LightGNN在保持与更复杂的状态最先进baseline相当的性能的同时，实现了边数80%和嵌入项90%的减少。我们的LightGNN框架的实现可以在github仓库中获得：https://github.com/HKUDS/LightGNN。