English

CorrectionLM: Self-Corrections with SLM for Dialogue State Tracking

Computation and Language 2024-10-25 v1

Abstract

Large language models (LLMs) have demonstrated self-improvement capabilities via feedback and refinement, but current small language models (SLMs) have had limited success in this area. Existing correction approaches often rely on distilling knowledge from LLMs, which imposes significant computation demands. In this work, we introduce CORRECTIONLM, a novel correction framework that enables SLMs to self-correct using in-context exemplars without LLM involvement. Applied to two dialogue state tracking (DST) tasks in low-resource settings, CORRECTIONLM achieves results similar to a state-of-the-art LLM at a small fraction of the computation costs.

Keywords

Cite

@article{arxiv.2410.18209,
  title  = {CorrectionLM: Self-Corrections with SLM for Dialogue State Tracking},
  author = {Chia-Hsuan Lee and Hao Cheng and Mari Ostendorf},
  journal= {arXiv preprint arXiv:2410.18209},
  year   = {2024}
}
R2 v1 2026-06-28T19:33:24.802Z