atlas

Research map / Graph learning / GNN architectures

How Does Message Passing Improve Collaborative Filtering?

NeurIPS 2024 · Graph learning › GNN architectures · approach: Graph rewiring and scalable message passing · 2 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Investigates why message passing helps collaborative filtering, finds gains mainly come from forward-pass neighbor representations and benefit low-degree users more, and proposes TAG-CF, a test-time single-step neighbor aggregation applied to matrix factorization embeddings.

Problem

Graph-based collaborative filtering uses message passing (MP) costly during training; it is unclear why MP helps CF and whether the explanations from graph learning tasks transfer to CF.

Approach

Decompose LightGCN's similarity into neighbor-information and gradient terms via ablations, and show BPR/DirectAU implicitly optimize a Laplacian smoothness term; based on this, apply a single MP step only at inference time to pretrained MF embeddings (TAG-CF), with an optional degree-cutoff variant (TAG-CF+).

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Add one-time test-time neighbor aggregation to MFMF trained with DirectAU · Six datasets, overall and low-degree usersNDCG@20: 3.2\% and 7.1\% for ENMF and MF overall; 4.6\% and 9.1\% for low-degree usersHelped
Add one-time test-time neighbor aggregation to MFMF trained with BPR · Anime, Yelp, Gowalla, BookNDCG@20: 6.3\% and 5.1\% average improvement on low-degree and overall NDCG respectivelyHelped
Use test-time neighbor aggregation instead of training-time aggregationCF recommendations compared with LightGCN · Anime, Yelp, Gowalla, BookNDCG@20: 1.2 average rankMixed
Remove gradients from the additional message-passing terms (ablation)LightGCN · Yelp-2018, Gowalla, Amazon-bookNDCG@20: 6.36 -> 6.16; 9.88 -> 9.87; 8.13 -> 7.80Hurt
Remove neighbor information from the test-time forward pass (ablation)LightGCN · Yelp-2018, Gowalla, Amazon-bookNDCG@20: 6.36 -> 4.71; 9.88 -> 6.95; 8.13 -> 6.95Hurt
Remove message passing in both forward and backward passes (ablation)LightGCN · Yelp-2018, Gowalla, Amazon-bookNDCG@20: 6.36 -> 6.09; 9.88 -> 9.83; 8.13 -> 7.75Hurt
Apply one-time aggregation only to low-degree nodes (ablation)TAG-CF · Amazon-book, Gowalla, Yelp-2018NDCG@20: 125\%, 17\%, and 11\%Helped
Increase low-degree-user upsampling from 100% to 300% (ablation)MF + TAG-CF on ML-1M · ML-1M, degree 40TAG-CF improvement in NDCG@20: 38.8\% -> 30.3\%Hurt
Use normalization m=n=-1/2 rather than tuned normalization (ablation)TAG-CF · Across datasetsrecommendation performanceMixed

Surprising findings

Limitations

Experimental setup

Datasets: Amazon-book Anime Gowalla Yelp-2018 MovieLens-1M Internal

Benchmarks: Amazon-book Anime Gowalla Yelp-2018 MovieLens-1M Internal

Metrics: NDCG@20 Recall@20 NDCG@10 Recall@10 running time

Baselines: MF ENMF UltraGCN LightGCN NGCF LightGCL SimGCL SGL

Builds on: LightGCN BPR DirectAU GCN

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy