English

Last Query Transformer RNN for knowledge tracing

Computers and Society 2021-02-11 v1 Machine Learning

Abstract

This paper presents an efficient model to predict a student's answer correctness given his past learning activities. Basically, I use both transformer encoder and RNN to deal with time series input. The novel point of the model is that it only uses the last input as query in transformer encoder, instead of all sequence, which makes QK matrix multiplication in transformer Encoder to have O(L) time complexity, instead of O(L^2). It allows the model to input longer sequence. Using this model I achieved the 1st place in the 'Riiid! Answer Correctness Prediction' competition hosted on kaggle.

Keywords

Cite

@article{arxiv.2102.05038,
  title  = {Last Query Transformer RNN for knowledge tracing},
  author = {SeungKee Jeon},
  journal= {arXiv preprint arXiv:2102.05038},
  year   = {2021}
}

Comments

kaggle competition 'Riiid! Answer Correctness Prediction' 1st place solution

R2 v1 2026-06-23T22:59:37.414Z