中文

Riiid! 答题正确性预测 Kaggle 竞赛:第四名方案综述

计算机与社会 2021-02-09 v1 人工智能 机器学习

摘要

本文介绍了我对 Riiid Labs(2020)在 Kaggle 上举办的“Riiid! 答题正确性预测”竞赛的解决方案,该方案得分为 0.817(AUC),在最终私有排行榜上排名第 4。它是一个单一的基于 transformer 的模型,深受 SAKT、SAINT 和 SAINT+ 等先前工作的启发。我认为起到关键作用的新颖要素包括时间感知注意力机制、输入序列嵌入与连续特征嵌入的拼接。

关键词

引用

@article{arxiv.2102.04250,
  title  = {Riiid! Answer Correctness Prediction Kaggle Challenge: 4th Place Solution Summary},
  author = {Duc Kinh Le Tran},
  journal= {arXiv preprint arXiv:2102.04250},
  year   = {2021}
}