A GitHub Project Recommendation Model Based on Self-Attention Sequence

Research output: Chapter in Book/Report/Conference proceedingConference contributionpeer-review

Abstract

In this paper, we propose a new GitHub recommendation model based on self-attention mechanism that considers user's historical operation sequence. It includes a project embedding layer, multiple encoder layers and a prediction layer. The main idea of our method is to add a position vector to the original project embedding vector to indicate the sequence information of the current project in the user's operation sequence. And considering that the next possible operation project of the user is largely determined by the previous project, model includes a residual connection to the encoder layer. Evaluated our method on a variety of large, real-world datasets, and it shows quantitatively that our outperforms alternative algorithms, especially on sparse datasets. The model can capture personalized dynamics and is able to make meaningful recommendations.

Original languageEnglish
Title of host publicationBDE 2021 - 2021 3rd International Conference on Big Data Engineering
PublisherAssociation for Computing Machinery
Pages110-116
Number of pages7
ISBN (Electronic)9781450389426
DOIs
StatePublished - 2021
Event3rd International Conference on Big Data Engineering, BDE 2021 - Virtual, Online, China
Duration: 29 May 202131 May 2021

Publication series

NameACM International Conference Proceeding Series

Conference

Conference3rd International Conference on Big Data Engineering, BDE 2021
Country/TerritoryChina
CityVirtual, Online
Period29/05/2131/05/21

Keywords

  • GitHub
  • Recommendation system
  • Self-attention
  • Sequence pattern

Fingerprint

Dive into the research topics of 'A GitHub Project Recommendation Model Based on Self-Attention Sequence'. Together they form a unique fingerprint.

Cite this