We gratefully acknowledge support from
the Simons Foundation and member institutions.
Full-text links:

Download:

Current browse context:

cs.IR

Change to browse by:

cs

References & Citations

DBLP - CS Bibliography

Bookmark

(what is this?)
CiteULike logo BibSonomy logo Mendeley logo del.icio.us logo Digg logo Reddit logo ScienceWISE logo

Computer Science > Information Retrieval

Title: Contrastive Learning with Bidirectional Transformers for Sequential Recommendation

Abstract: Contrastive learning with Transformer-based sequence encoder has gained predominance for sequential recommendation. It maximizes the agreements between paired sequence augmentations that share similar semantics. However, existing contrastive learning approaches in sequential recommendation mainly center upon left-to-right unidirectional Transformers as base encoders, which are suboptimal for sequential recommendation because user behaviors may not be a rigid left-to-right sequence. To tackle that, we propose a novel framework named \textbf{C}ontrastive learning with \textbf{Bi}directional \textbf{T}ransformers for sequential recommendation (\textbf{CBiT}). Specifically, we first apply the slide window technique for long user sequences in bidirectional Transformers, which allows for a more fine-grained division of user sequences. Then we combine the cloze task mask and the dropout mask to generate high-quality positive samples and perform multi-pair contrastive learning, which demonstrates better performance and adaptability compared with the normal one-pair contrastive learning. Moreover, we introduce a novel dynamic loss reweighting strategy to balance between the cloze task loss and the contrastive loss. Experiment results on three public benchmark datasets show that our model outperforms state-of-the-art models for sequential recommendation.
Comments: Accepted by CIKM 2022
Subjects: Information Retrieval (cs.IR)
Cite as: arXiv:2208.03895 [cs.IR]
  (or arXiv:2208.03895v3 [cs.IR] for this version)

Submission history

From: Hanwen Du [view email]
[v1] Mon, 8 Aug 2022 03:33:06 GMT (622kb,D)
[v2] Sun, 14 Aug 2022 02:52:08 GMT (622kb,D)
[v3] Wed, 17 Aug 2022 10:16:59 GMT (4940kb,D)

Link back to: arXiv, form interface, contact.