Trusting Language Models in Education

Jogi Suda Neto,Li Deng,Thejaswi Raya,Reza Shahbazi,Nick Liu,Adhitya Venkatesh,Miral Shah,Neeru Khosla,Rodrigo Capobianco Guido
2023-08-08
Abstract:Language Models are being widely used in Education. Even though modern deep learning models achieve very good performance on question-answering tasks, sometimes they make errors. To avoid misleading students by showing wrong answers, it is important to calibrate the confidence - that is, the prediction probability - of these models. In our work, we propose to use an XGBoost on top of BERT to output the corrected probabilities, using features based on the attention mechanism. Our hypothesis is that the level of uncertainty contained in the flow of attention is related to the quality of the model's response itself.
Computation and Language,Artificial Intelligence
What problem does this paper attempt to address?
The problem this paper attempts to address is the confidence calibration of language models in educational question-answering systems. Specifically, although modern deep learning models perform well on question-answering tasks, they sometimes make mistakes. To avoid misleading students by presenting incorrect answers, it is necessary to calibrate the confidence (i.e., prediction probability) of these models. The paper proposes an XGBoost-based method to output corrected probabilities on top of BERT and uses features from the attention mechanism to improve calibration effectiveness. The authors hypothesize that the uncertainty in the attention flow is related to the quality of the model's responses.