Repository Feedback

Your feedback helps us improve the repository's content relevance and usability. Please share your thoughts to help us better serve researchers and practitioners.

Submit feedback

Submit a research study

Contribute to the repository:

Add a paper

Language Bottleneck Models: A Framework for Interpretable Knowledge Tracing and Beyond

Authors
Antonin Berthon,
Mihaela van der Schaar
Date
Publisher
arXiv
Accurately assessing student knowledge is critical for effective education, yet traditional Knowledge Tracing (KT) methods rely on opaque latent embeddings, limiting interpretability. Even LLM-based approaches generate direct predictions or summaries that may hallucinate without any accuracy guarantees. We recast KT as an inverse problem: learning the minimum natural-language summary that makes past answers explainable and future answers predictable. Our Language Bottleneck Model (LBM) consists of an encoder LLM that writes an interpretable knowledge summary and a frozen decoder LLM that must reconstruct and predict student responses using only that summary text. By constraining all predictive information to pass through a short natural-language bottleneck, LBMs ensure that the summary contains accurate information while remaining human-interpretable. Experiments on synthetic arithmetic benchmarks and the large-scale Eedi dataset show that LBMs rival the accuracy of state-of-the-art KT and direct LLM methods while requiring orders-of-magnitude fewer student trajectories. We demonstrate that training the encoder with group-relative policy optimization, using downstream decoding accuracy as a reward signal, effectively improves summary quality.
Study at a glance
  • Question: Can AI create clear, readable summaries of student knowledge that accurately predict their future answers?
  • Setting and data: Synthetic arithmetic data and the large-scale Eedi student dataset.
  • Method: Tests how well a new AI model predicts student answers compared with existing methods.
Why it matters

Accurately assessing student knowledge is critical for effective education. Yet traditional knowledge tracing relies on opaque embeddings that limit interpretability, and language models can hallucinate summaries or predictions without accuracy guarantees.

Key findings
  • On synthetic arithmetic benchmarks and the large-scale Eedi dataset, an AI model matched top student-tracking accuracy while needing far fewer student learning histories.
  • Restricting student tracking data to short, plain-language summaries ensured that student knowledge descriptions remained accurate and easily readable by humans.
  • Training the AI using its accuracy in predicting student responses as a reward effectively improved the quality of its written summaries.

Summary drafted with AI from the paper’s abstract and may contain errors. See the full paper for details.

What is the application?
Who is the user?