English

Improving Contrastive Learning with Model Augmentation

Machine Learning 2022-03-30 v1 Artificial Intelligence Information Retrieval

Abstract

The sequential recommendation aims at predicting the next items in user behaviors, which can be solved by characterizing item relationships in sequences. Due to the data sparsity and noise issues in sequences, a new self-supervised learning (SSL) paradigm is proposed to improve the performance, which employs contrastive learning between positive and negative views of sequences. However, existing methods all construct views by adopting augmentation from data perspectives, while we argue that 1) optimal data augmentation methods are hard to devise, 2) data augmentation methods destroy sequential correlations, and 3) data augmentation fails to incorporate comprehensive self-supervised signals. Therefore, we investigate the possibility of model augmentation to construct view pairs. We propose three levels of model augmentation methods: neuron masking, layer dropping, and encoder complementing. This work opens up a novel direction in constructing views for contrastive SSL. Experiments verify the efficacy of model augmentation for the SSL in the sequential recommendation. Code is available\footnote{\url{https://github.com/salesforce/SRMA}}.

Keywords

Cite

@article{arxiv.2203.15508,
  title  = {Improving Contrastive Learning with Model Augmentation},
  author = {Zhiwei Liu and Yongjun Chen and Jia Li and Man Luo and Philip S. Yu and Caiming Xiong},
  journal= {arXiv preprint arXiv:2203.15508},
  year   = {2022}
}

Comments

Preprint. Still under reivew

R2 v1 2026-06-24T10:30:01.284Z