Dynamic Key-Value Memory Networks for Knowledge Tracing
Paper · 2017 · Jiani Zhang, Xingjian Shi, Irwin King et al.
Summary
The paper addresses knowledge tracing (KT): estimating how well a learner understands each underlying concept from their history of attempted exercises and whether each answer was right, then predicting success on the next exercise. Earlier approaches each have a weakness. Bayesian Knowledge Tracing handles every predefined concept in isolation with a binary hidden state. Deep Knowledge Tracing (an LSTM) compresses all knowledge into one hidden vector, which hides per-concept mastery.
The authors propose Dynamic Key-Value Memory Networks (DKVMN), a memory-augmented neural network with two parts:
- a fixed key matrix that holds latent concept representations
- a changeable value matrix that holds and revises the learner's mastery of each concept.
For each exercise, the model computes correlation weights against the keys, reads only the relevant mastery slots to make its prediction, and then writes updates back to those slots.
Across one synthetic and three real-world datasets, DKVMN consistently beat BKT, DKT and a single-matrix MANN baseline. It also uncovers exercise-to-concept groupings without expert labels and can visualize how a learner's mastery evolves. This makes it useful for adaptive tutoring and MOOCs.
Key points
- Architecture: a static key memory (concepts) paired with a dynamic value memory (per-concept mastery). This differs from standard MANNs, which use either one memory matrix or two static ones.
- Evaluation: On four KT datasets, three drawn from real-world sources and one synthetic, DKVMN achieved better results than both BKT and DKT. A plain MANN baseline used cosine-similarity reads and LRUA writes, with the usage decay γ fixed at 0.9.
- Concept discovery on synthetic-5: with memory size N = 50, exercises still fell into five concept groups. Assigning each exercise by its maximum correlation weight gave an adjusted mutual information of 0.879, and t-SNE-based clustering reached a perfect 1.0.
- Qualitative results on real data: exercises were grouped into 10 clusters. For example, ordering tasks for fractions, integers, decimals and real numbers landed in the same cluster.
- Knowledge state depiction: the model can trace individual concept states over time, illustrated with five concepts across 50 answered exercises. Planned future work includes adding exercise content to the embeddings and building hierarchical key-value memories.
Institutions
Fields, topics and concepts
📚 Shelves 4The field of science this work belongs to (OpenAlex fields).Computer Science · 19,503 worksArtificial Intelligence · 18,805 worksInformation Systems · 1,888 worksComputer Science Applications · 184 works
Cite
Cite as Jiani Zhang, Xingjian Shi, Irwin King and Dit-Yan Yeung. 2017. Dynamic Key-Value Memory Networks for Knowledge Tracing. Proceedings of the 26th International Conference on World Wide Web.
@inproceedings{zhang2017dynamic,
title = {{Dynamic Key-Value Memory Networks for Knowledge Tracing}},
author = {Jiani Zhang, Xingjian Shi, Irwin King and Dit-Yan Yeung},
booktitle = {Proceedings of the 26th International Conference on World Wide Web},
year = {2017},
doi = {10.1145/3038912.3052580},
eprint = {1611.08108},
archivePrefix = {arXiv},
url = {https://doi.org/10.1145/3038912.3052580}
}References
- A. Bordes, N. Usunier, S. Chopra, and J. Weston. Large-scale simple question answering with memory networks. arXiv preprint arXiv:1506.02075, 2015. find
- T. Chen, M. Li, Y. Li, M. Lin, N. Wang, M. Wang, T. Xiao, B. Xu, C. Zhang, and Z. Zhang. Mxnet: A flexible and efficient machine learning library for heterogeneous distributed systems. In In Neural Information Processing Systems, Workshop on Machine Learning Systems, 2015.
- A. T. Corbett and J. R. Anderson. Knowledge tracing: Modeling the acquisition of procedural knowledge. volume 4, pages 253–278. Springer, 1994. find
- R. S. d Baker, A. T. Corbett, and V. Aleven. More accurate student modeling through contextual estimation of slip and guess probabilities in bayesian knowledge tracing. In International Conference on Intelligent Tutoring Systems, pages 406–415. Springer, 2008. find
- M. Feng, N. Heffernan, and K. Koedinger. Addressing the assessment challenge with an online system that tutors as it assesses. User Modeling and User-Adapted Interaction, 19(3):243–266, 2009. find
- A. Graves, G. Wayne, and I. Danihelka. Neural turing machines. arXiv preprint arXiv:1410.5401, 2014. find
- A. Graves, G. Wayne, M. Reynolds, T. Harley, I. Danihelka, A. Grabska-Barwińska, S. G. Colmenarejo, E. Grefenstette, T. Ramalho, J. Agapiou, et al. Hybrid computing using a neural network with dynamic external memory. Nature, 538(7626):471–476, 2016. find
- E. Grefenstette, K. M. Hermann, M. Suleyman, and P. Blunsom. Learning to transduce with unbounded memory. In Advances in Neural Information Processing Systems, pages 1828–1836, 2015. find
- S. Hochreiter and J. Schmidhuber. Long short-term memory. Neural computation, 9(8):1735–1780, 1997. find
- A. Joulin and T. Mikolov. Inferring algorithmic patterns with stack-augmented recurrent nets. In Advances in Neural Information Processing Systems, pages 190–198, 2015. find
- M. Khajah, R. V. Lindsey, and M. C. Mozer. How deep is knowledge tracing? In Educational Data Mining 2016, 2016.
- K. R. Koedinger, R. S. Baker, K. Cunningham, A. Skogsholm, B. Leber, and J. Stamper. A data repository for the edm community: The pslc datashop. Handbook of educational data mining, 43, 2010. find
- A. Krizhevsky, I. Sutskever, and G. E. Hinton. Imagenet classification with deep convolutional neural networks. In Advances in Neural Information Processing Systems, pages 1097–1105, 2012.
- Y. LeCun, Y. Bengio, and G. Hinton. Deep learning. Nature, 521(7553):436–444, 2015. find
- L. v. d. Maaten and G. Hinton. Visualizing data using t-sne. Journal of Machine Learning Research, 9(11):2579–2605, 2008. find
- T. Mikolov, I. Sutskever, K. Chen, G. S. Corrado, and J. Dean. Distributed representations of words and phrases and their compositionality. In Advances in Neural Information Processing Systems, pages 3111–3119, 2013.
- A. Miller, A. Fisch, J. Dodge, A.-H. Karimi, A. Bordes, and J. Weston. Key-value memory networks for directly reading documents. arXiv preprint arXiv:1606.03126, 2016. find
- G. B. Orr and K.-R. Müller. Neural networks: tricks of the trade. Springer, 2003. find
- Z. A. Pardos and N. T. Heffernan. Modeling individualization in a bayesian networks implementation of knowledge tracing. In User Modeling, Adaptation, and Personalization, pages 255–266. Springer, 2010. find
- Z. A. Pardos and N. T. Heffernan. Kt-idem: Introducing item difficulty to the knowledge tracing model. In User Modeling, Adaption and Personalization, pages 243–254. Springer, 2011. find
- R. Pascanu, T. Mikolov, and Y. Bengio. On the difficulty of training recurrent neural networks. In Proceedings of The 30th International Conference on Machine Learning, pages 1310–1318, 2013. find
- C. Piech, J. Bassen, J. Huang, S. Ganguli, M. Sahami, L. J. Guibas, and J. Sohl-Dickstein. Deep knowledge tracing. In Advances in Neural Information Processing Systems, pages 505–513, 2015.
- J. Reye. Student modelling based on belief networks. International Journal of Artificial Intelligence in Education, 14(1):63–96, 2004. find
- A. Santoro, S. Bartunov, M. Botvinick, D. Wierstra, and T. Lillicrap. Meta-learning with memory-augmented neural networks. In Proceedings of The 33rd International Conference on Machine Learning, pages 1842–1850, 2016. find
- J. Schmidhuber. Deep learning in neural networks: An overview. Neural Networks, 61:85–117, 2015. find
- P. Steif and N. Bier. Oli engineering statics - fall 2011. Feb. 2014. find
- S. Sukhbaatar, J. Weston, R. Fergus, et al. End-to-end memory networks. In Advances in neural information processing systems, pages 2440–2448, 2015. find
- O. Vinyals, C. Blundell, T. Lillicrap, D. Wierstra, et al. Matching networks for one shot learning. In Advances in Neural Information Processing Systems, pages 3630–3638, 2016.
- P. Viola and W. M. Wells III. Alignment by maximization of mutual information. International Journal of Computer Vision, 24(2):137–154, 1997. find
- J. Weston, S. Chopra, and A. Bordes. Memory networks. International Conference on Learning Representations, 2015. find
- K. H. Wilson, X. Xiong, M. Khajah, R. V. Lindsey, S. Zhao, Y. Karklin, E. G. Van Inwegen, B. Han, C. Ekanadham, J. E. Beck, et al. Estimating student proficiency: Deep learning is not the panacea. In In Neural Information Processing Systems, Workshop on Machine Learning for Education, 2016. find
- X. Xiong, S. Zhao, E. G. Van Inwegen, and J. E. Beck. Going deeper with deep knowledge tracing. In Educational Data Mining 2016, 2016.
- M. V. Yudelson, K. R. Koedinger, and G. J. Gordon. Individualized bayesian knowledge tracing models. In Artificial intelligence in education, pages 171–180. Springer, 2013. find
Updated
This page in Markdown → https://md.research.elamur.ai/free/en/papers/zhang-2017-dynamic-key-value-memory-networks/