FlatFormer: A Flat Transformer Knowledge Tracing Model Based on Cognitive Bias Injection
Not on the map. Why?
Abstract (arXiv)
Knowledge Tracing (KT) models face a critical ``Performance-Complexity Trap'': capturing complex cognitive dynamics like learning sessions and memory decay typically requires deep hierarchical architectures, which incur prohibitive computational costs for real-time deployment. To resolve this, we propose FlatFormer, a streamlined architecture based on the novel design paradigm of ``Information Injection over Structural Stacking.'' Unlike parameter-heavy hierarchical models, FlatFormer leverages a standard flat Transformer augmented with two lightweight injection mechanisms: (i) a hybrid input encoding strategy combining learnable session identifiers with fixed sinusoidal step embeddings; and (ii) a pre-computed power-law bias integrated directly into attention logits to explicitly model the forgetting curve. Extensive experiments on four large-scale datasets (e.g., EdNet, Junyi) show that FlatFormer achieves state-of-the-art performance. For example, on the EdNet dataset, compared to the strongest hierarchical baseline (HiTSKT), its absolute AUC increased by 8.3%, while using less than 15% of parameters, and inference speed was about three times faster. These results validate that high cognitive fidelity does not necessitate architectural complexity.
Key facts. Cited by 1 paper in the map (data through 7 Oct 2026). Builds on 7 of its 21 references that are in the map. First posted on arXiv 7 Dec 2025.
Builds on
7 of 21 references.
References · 7
- 2017Attention Is All You Needcited by 59,932
- 2021Train Short, Test Long: Attention with Linear Biases Enables Input Length Extrapolationcited by 1,085
- 2015Deep Knowledge Tracingcited by 328
- 2020Context-Aware Attentive Knowledge Tracingcited by 172
- 2019A Self-Attentive model for Knowledge Tracingcited by 169
- 2019EdNet: A Large-Scale Hierarchical Dataset in Educationcited by 106
- 2020Towards an Appropriate Query, Key, and Value Computation for Knowledge Tracingcited by 79
Built on by
Citing papers per year: 2026: 1.
1 paper.