English

Distillation Contrastive Decoding: Improving LLMs Reasoning with Contrastive Decoding and Distillation

Computation and Language 2024-08-26 v2 Artificial Intelligence Machine Learning

Abstract

We propose a straightforward approach called Distillation Contrastive Decoding (DCD) to enhance the reasoning capabilities of Large Language Models (LLMs) during inference. In contrast to previous approaches that relied on smaller amateur models or analysis of hidden state differences, DCD employs Contrastive Chain-of-thought Prompting and advanced distillation techniques, including Dropout and Quantization. This approach effectively addresses the limitations of Contrastive Decoding (CD), which typically requires both an expert and an amateur model, thus increasing computational resource demands. By integrating contrastive prompts with distillation, DCD obviates the need for an amateur model and reduces memory usage. Our evaluations demonstrate that DCD significantly enhances LLM performance across a range of reasoning benchmarks, surpassing both CD and existing methods in the GSM8K and StrategyQA datasets.

Keywords

Cite

@article{arxiv.2402.14874,
  title  = {Distillation Contrastive Decoding: Improving LLMs Reasoning with Contrastive Decoding and Distillation},
  author = {Phuc Phan and Hieu Tran and Long Phan},
  journal= {arXiv preprint arXiv:2402.14874},
  year   = {2024}
}

Comments

Under Review

R2 v1 2026-06-28T14:57:38.953Z