Source-linked AI summary

Embarrassingly Shallow Autoencoders for Sparse Data

Harald Steck

arXiv:1905.03375v1cs.IRcs.LGstat.ML

TL;DR

Collaborative filtering needs effective methods for sparse implicit-feedback data, while recent deep models do not necessarily improve recommendation accuracy. The paper introduces easer, a linear autoencoder trained with a convex objective that has a closed-form solution. Despite its simplicity, easer outperformed competing approaches on most evaluated public datasets and enabled substantially faster training than Slim.

  • Problem

    The paper addresses collaborative filtering for sparse, typically binary implicit-feedback data, where recent deep approaches had not consistently improved recommendation accuracy.

  • Method

    easer is a linear autoencoder with zero-diagonal item-item weights, trained using square loss and L2 regularization in a convex objective with a closed-form solution.

  • Results

    easer outperformed various competing linear, nonlinear, deep, probabilistic, and neighborhood-based models on most evaluated public datasets.

  • Takeaways & Limitations

    The results support combining autoencoder reconstruction with neighborhood-style item relationships, including learned negative weights for item dissimilarities.

  • Takeaways & Limitations

    The interpretation of the closed-form solution as a precision-matrix estimate is limited to normally distributed, zero-mean variables, although the derivation permits other Gram matrices.

Abstract

from arXiv · show

Combining simple elements from the literature, we define a linear model that is geared toward sparse data, in particular implicit feedback data for recommender systems. We show that its training objective has a closed-form solution, and discuss the resulting conceptual insights. Surprisingly, this simple model achieves better ranking accuracy than various state-of-the-art collaborative-filtering approaches, including deep non-linear models, on most of the publicly available data-sets used in our experiments.

1 INTRODUCTION

The paper takes collaborative filtering's shallow-layer finding to its extreme by defining a linear autoencoder without hidden layers for sparse implicit-feedback data. Its closed-form training and strong ranking results support the value of this simple design.

  • 1 INTRODUCTION: easer is a linear autoencoder without a hidden layer that reconstructs users’ binary interaction vectors to predict recommended items.The input marks interacted items, while the output reproduces the input as recommendation scores.
  • 1 INTRODUCTION: The model has a closed-form convex training solution and can train several orders of magnitude faster than Slim when it fits into memory.The paper reports only a few lines of training code and much lower wall-clock time than Slim.
  • 1 INTRODUCTION: easer achieved competitive ranking accuracy and outperformed various deep, nonlinear, probabilistic, and neighborhood-based approaches on most public datasets.This is the paper’s principal empirical claim across the evaluated datasets.

2 MODEL DEFINITION

easer represents sparse implicit-feedback data with an item-item weight matrix and predicts each user’s item scores by multiplying the interaction vector by that matrix. Its zero diagonal prevents self-reconstruction and forces item relationships to drive prediction.

  • 2 MODEL DEFINITION: The data are represented by a sparse, typically binary matrix X, where positive entries indicate observed user-item interactions and zeros indicate none observed.X has one row per user and one column per item.
  • 2 MODEL DEFINITION: easer learns an item-item weight matrix B whose diagonal is constrained to zero.The zero diagonal forbids an item from directly reproducing itself across the input and output layers.
  • 2 MODEL DEFINITION: The predicted score S_u,j for item j and user u is computed from the dot product of user u’s interaction row and item j’s weight column.This makes each item score depend on the user’s interacted items and the corresponding weights.

3 MODEL TRAINING

The model uses a regularized square-loss objective with a zero-diagonal constraint, yielding a closed-form solution whose interpretation connects item prediction to precision-matrix structure. Training can operate on the item Gram matrix, reducing computational cost relative to Slim-style approaches.

  • Objective design: The square loss compares X with predicted scores S = XB and enables a closed-form training solution; L2 regularization leaves λ as the sole hyperparameter.Other losses might improve ranking accuracy but may increase computational cost.
  • Objective design: The constraint diag(B) = 0 prevents the trivial identity solution B = I and forces the model to generalize when reproducing inputs.The constraint was introduced in Slim and is described as crucial.
  • Closed-form solution: The constrained convex optimization problem is solved in closed form by incorporating the zero-diagonal equality constraint through Lagrange multipliers.The multipliers are selected so that the resulting estimate satisfies diag(B) = 0.
  • Closed-form solution: The learned off-diagonal weights derive from the regularized inverse matrix P̂, whose columns are divided by their diagonal elements; B̂ is generally asymmetric although P̂ is symmetric.The resulting solution obeys the zero-diagonal constraint.
  • Interpretation: The sufficient statistics are the item Gram matrix G = X^⊤X, which becomes an item co-occurrence matrix for sparse binary data and can be estimated accurately when counts are sufficiently large.This connection follows specifically from using the square loss.
  • Interpretation: Under the Gaussian zero-mean special case, the prediction rule matches conditional point prediction based on the precision matrix, making precision rather than covariance the conceptually correct similarity matrix.The precision-matrix interpretation is limited to normally distributed variables with zero mean.
  • Computational cost: Training requires only G rather than X and has complexity O(|I|^3) with basic inversion, independent of users and interactions when G is precomputed.This is lower than the corresponding Slim-style cost, helping explain the reduced training times observed for easer.

4 RELATED WORK

The paper situates easer among autoencoders, Slim variants, and neighborhood methods, emphasizing its hidden-layer-free architecture, zero-diagonal constraint, dense weights, computational efficiency, and precision-matrix interpretation.

  • 4 RELATED WORK: easer is simultaneously presented as an autoencoder, a modified Slim model, and a neighborhood-based approach.
  • Autoencoders: Unlike deep collaborative-filtering models, easer has no hidden layer and forces each output item to use other input items for reconstruction.
  • Slim: Slim and its variants require substantially more computation because they solve |I| independent optimization problems, whereas easer requires one regression problem when its weight matrix fits in memory.
  • Slim: The zero-diagonal constraint is retained from Slim and is identified as crucial for improved ranking accuracy while preserving a closed-form solution.
  • Slim: Dropping non-negative weights improved ranking accuracy, while omitting L1 regularization improved computational efficiency; the resulting learned weight matrix is dense.
  • Neighborhood-based Approaches: Neighborhood methods typically use co-occurrence matrices heuristically, whereas easer uses the inverse data Gram matrix as a conceptually correct similarity matrix.

5 EXPERIMENTS

The experiments compare easer with linear, nonlinear, deep, probabilistic, and neighborhood-based recommenders across several public datasets and evaluation setups. Easer generally achieves stronger ranking accuracy while training much faster than Slim, with gains varying by dataset and recommendation emphasis.

  • Overall results: Easer obtained considerably better ranking accuracy than competing models on most evaluated datasets despite its simplicity.The comparison covered linear, nonlinear, deep, probabilistic, and neighborhood-based approaches.
  • Comparison to Slim: Easer outperformed Slim on all datasets, indicating benefits from removing non-negative-weight and L1-regularization constraints.Negative weights comprised about 60% of learned weights, and setting them to zero reduced accuracy toward Slim’s level.
  • Computational cost: Easer training took less than two minutes on Netflix and less than 20 minutes on MSD, versus about two weeks for Slim’s Netflix grid search.Easer has one L2 regularization hyperparameter, whereas Slim jointly optimizes L1 and L2 parameters.
  • Comparison to linear and deep models: On Netflix and MSD, easer significantly outperformed every competing linear, nonlinear, deep, or probabilistic model evaluated in [13].On MSD, improvements over the best competing model were 25% for Recall@20, 17% for Recall@50, and 23% for NDCG@100.
  • Dataset-dependent behavior: Easer’s relative advantage differed across datasets, suggesting flexibility to emphasize personally relevant or generally popular items.Its improvement was notable on MSD, while results on ML-20M and Netflix suggested greater focus on popular items.
  • Comparison to neighborhood-based approaches: Easer improved NDCG@10 by 0.0301 over item-item and 0.0289 over wmf, approximately twice the gains reported for ii-SVD-500.The comparison used the weak-generalization MovieLens 10M split from.

6 CONCLUSIONS

The paper concludes that its simple linear collaborative-filtering model combines autoencoder and neighborhood strengths while remaining efficient and accurate. Its results support inverse-Gram similarity, negative weights, and zero self-similarity as important design choices for sparse data.

  • The model combines autoencoder and neighborhood-based collaborative-filtering strengths in a simple linear formulation.
  • Training can save up to several orders of magnitude when the model fits into memory.
  • The conceptually correct neighborhood similarity matrix is based on the inverse of the data Gram matrix, unlike typical direct-Gram approaches.
  • Allowing negative weights was essential for ranking accuracy because the model can learn dissimilarities as well as similarities.
  • Ranking accuracy was on par with or better than various state-of-the-art approaches, including deep nonlinear and neighborhood-based models.
Loading 1905.03375v1…