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