Method · Kernel and closed-form inner solvers
Distill-CF
Infinite Recommendation Networks: A Data-Centric Approach
Noveen Sachdeva, Mehak Preet Dhaliwal, Carole-Jean Wu, Julian McAuley
NeurIPS 2022 · first public 2022-06-03 · arXiv 2206.02626
In one paragraph
Uses the Neural Tangent Kernel equivalence of an infinitely-wide autoencoder (infinity-AE) to give recommendation-model training a closed-form solution, then builds Distill-CF on top of it with differentiable Gumbel-sampling to synthesize tiny, high-fidelity summaries of large, sparse user-item interaction matrices; reports 96-105% of infinity-AE's full-data performance using as little as 0.1% of the original interactions.
Where it sits
- Kernel and closed-form inner solvers (Bi-level performance matching)
- Setting: Other data types
Design choices
| Optimization regime | single-level |
| Prior / networks used | ntk-kernel |
Abstract (verbatim from arXiv)
We leverage the Neural Tangent Kernel and its equivalence to training infinitely-wide neural networks to devise $\infty$-AE: an autoencoder with infinitely-wide bottleneck layers. The outcome is a highly expressive yet simplistic recommendation model with a single hyper-parameter and a closed-form solution. Leveraging $\infty$-AE's simplicity, we also develop Distill-CF for synthesizing tiny, high-fidelity data summaries which distill the most important knowledge from the extremely large and sparse user-item interaction matrix for efficient and accurate subsequent data-usage like model training, inference, architecture search, etc. This takes a data-centric approach to recommendation, where we aim to improve the quality of logged user-feedback data for subsequent modeling, independent of the learning algorithm. We particularly utilize the concept of differentiable Gumbel-sampling to handle the inherent data heterogeneity, sparsity, and semi-structuredness, while being scalable to datasets with hundreds of millions of user-item interactions. Both of our proposed approaches significantly outperform their respective state-of-the-art and when used together, we observe 96-105% of $\infty$-AE's performance on the full dataset with as little as 0.1% of the original dataset size, leading us to explore the counter-intuitive question: Is more data what you need for better recommendation?
BibTeX (generated; prefer the venue's official entry)
@article{sachdeva2022infinite,
title = {Infinite Recommendation Networks: A Data-Centric Approach},
author = {Noveen Sachdeva and Mehak Preet Dhaliwal and Carole-Jean Wu and Julian McAuley},
journal = {NeurIPS 2022},
year = {2022}
}Nearby in Kernel and closed-form inner solvers
MMDD — Efficient Multi-modal Dataset Distillation via Analytic Parameter Matching
Deyu Bo, Xinchao Wang · ICML 2026notableVision–languagepaper ↗code ↗
Provable KRR DD — Provable and Efficient Dataset Distillation for Kernel Ridge Regression
Yilan Chen, Wei Huang, Tsui-Wei Weng · NeurIPS 2024notablepaper ↗
Rethinking Backdoor Attacks on Dataset Distillation: A Kernel Method Perspective
Ming-Yu Chung, Sheng-Yen Chou, Chia-Mu Yu et al. · ICLR 2024notablepaper ↗
KRR-ST — Self-Supervised Dataset Distillation for Transfer Learning
Dong Bok Lee, Seanie Lee, Joonho Ko et al. · ICLR 2024corePre-training & transferpaper ↗code ↗