ICML 2026poster0 citations

CSG: Cognitive Structure Generation for Intelligent Education

Hengnian Gu, Zhifu Chen, Yuxin Chen, Jin Zhou, Dongdai Zhou

Abstract

Cognitive structure (CS), a student's construction of concepts and inter-concept relations, has long been recognized as a foundational notion in psychology and intelligent education, yet remains largely unassessable in practice. Existing approaches such as knowledge tracing (KT) and cognitive diagnosis (CD) simplify and indirectly approximate CS, but they intertwine representation learning with prediction objectives, limiting generalization, interpretability, and reuse across tasks. To address this gap, we propose Cognitive Structure Generation (CSG), a task-agnostic framework that explicitly models CS through generative modeling. Based on educational theories, CSG first pretrains a Cognitive Structure Diffusion Probabilistic Model (CSDPM) and then applies reinforcement learning with SOLO-based hierarchical rewards to capture plausible patterns of cognitive development. By decoupling cognitive structure representation from downstream prediction, CSG produces interpretable and transferable cognitive structures that can be seamlessly integrated into diverse student modeling tasks. Experiments on five real-world datasets show that CSG yields more comprehensive representations, substantially improving performance while offering enhanced interpretability and modularity.

DiffusionRLTheoryBenchmark
BibTeX
@inproceedings{
gu2026csg,
title={{CSG}: Cognitive Structure Generation for Intelligent Education},
author={Hengnian Gu and Zhifu Chen and Yuxin Chen and Jin Peng Zhou and Dongdai Zhou},
booktitle={Forty-third International Conference on Machine Learning},
year={2026},
url={https://openreview.net/forum?id=PfarHkQG8e}
}