Graph neural networks (GNNs) have emerged as the state-of-the-art paradigm for collaborative filtering (CF). To improve the representation quality over limited labeled data, contrastive learning has attracted attention in recommendation and benefited graph-based CF model recently. However, the success of most contrastive methods heavily relies on manually generating effective contrastive views for heuristic-based data augmentation. This does not generalize across different datasets and downstream recommendation tasks, which is difficult to be adaptive for data augmentation and robust to noise perturbation. To fill this crucial gap, this work proposes a unified Automated Collaborative Filtering (AutoCF) to automatically perform data augmentation for recommendation. Specifically, we focus on the generative self-supervised learning framework with a learnable augmentation paradigm that benefits the automated distillation of important self-supervised signals. To enhance the representation discrimination ability, our masked graph autoencoder is designed to aggregate global information during the augmentation via reconstructing the masked subgraph structures. Experiments and ablation studies are performed on several public datasets for recommending products, venues, and locations. Results demonstrate the superiority of AutoCF against various baseline methods. We release the model implementation at https://github.com/HKUDS/AutoCF.
翻译:图神经网络是协同过滤中最先进的方法。为了提高有限标注数据的表示质量,对比学习在推荐领域引起了人们的关注,并最近在基于图的协同过滤模型中受益。然而,大多数对比方法的成功很大程度上取决于手动生成有效的对比视图来进行基于启发式数据增强。这在不同的数据集和下游推荐任务中不具有普适性,难以适应数据增强并对噪声扰动具有鲁棒性。为了填补这一重要差距,本文提出了一个统一的自动协同过滤 (AutoCF) 方法,用于自动推荐数据增强。具体而言,我们专注于具有可学习增强范式的生成式自我监督学习框架,该框架有助于自动提取重要的自我监督信号。为了增强表示判别能力,我们的掩码图自动编码器设计为通过重建掩码子图结构来聚合全局信息进行增强。我们在几个公共数据集上进行实验和消融研究,用于推荐产品、场所和位置等。结果表明,自动CF相对于各种基线方法具有优势。我们在https://github.com/HKUDS/AutoCF上发布了模型实现。