CLJun 28, 2024

Calibrating LLMs with Preference Optimization on Thought Trees for Generating Rationale in Science Question Scoring

arXiv:2406.19949v225 citationsHas Code
AI Analysis

This addresses the need for explainable AI in educational assessment by improving rationale generation, though it is incremental as it builds on existing preference optimization methods.

The paper tackles the problem of generating faithful rationales for automated scoring systems in science questions, achieving a 38% improvement in assessment performance (QWK score) compared to prior work while producing higher-quality rationales.

Generating rationales that justify scoring decisions has been a promising way to facilitate explainability in automated scoring systems. However, existing methods do not match the accuracy of classifier-based methods. Plus, the generated rationales often contain hallucinated information. To address these issues, we propose a novel framework capable of generating more faithful rationales and, more importantly, matching performance with classifier-based black-box scoring systems. We first mimic the human assessment process by querying Large Language Models (LLMs) to generate a thought tree. We then summarise intermediate assessment decisions from each thought tree path for creating synthetic rationale data and rationale preference data. Finally, we utilise the generated synthetic data to calibrate LLMs through a two-step training process: supervised fine-tuning and preference optimization. Extensive experimental results demonstrate that our framework achieves a 38% assessment performance improvement in the QWK score compared to prior work while producing higher-quality rationales, as recognised by human evaluators and LLMs. Our work sheds light on the effectiveness of performing preference optimization using synthetic preference data obtained from thought tree paths. Data and code are available at https://github.com/lijiazheng99/thought_tree_assessment.

Code Implementations1 repo
Foundations

The foundational work for this paper's niche, ranked by how specifically the neighbourhood builds on it — not by global fame.

Your Notes