atlas

Research map / Large language models / Reasoning and chain of thought

Turning Dust into Gold: Distilling Complex Reasoning Capabilities from LLMs by Leveraging Negative Data

AAAI 2024 · Large language models › Reasoning and chain of thought · approach: Search and self-correction in reasoning · 8 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Proposes a three-step model specialization framework (negative assistant training, negative calibrated enhancement, adaptive self-consistency) that distills LLM chain-of-thought reasoning into LLaMA-7B using both positive and negative synthesized samples on the MATH dataset.

Problem

Distilling complex mathematical reasoning from LLMs into small models usually discards rationales with incorrect final answers, wasting knowledge in negative samples, especially on hard problems like MATH.

Approach

Train a frozen negative LoRA on incorrect rationales and fuse it into a positive LoRA via a corrected attention mechanism with signed weights in [-0.5, 0.5]; use the negative model as a calibration baseline for self-distillation via KL-based weighting; and train a ranking model on positive vs negative rationales to reweight self-consistency votes.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
add negative data via negative assistant training (dual LoRA) vs. positive-only CoT KDLLaMA 7B distilled from GPT-3.5 Turbo · MATH test setaverage accuracy: 6.81 vs 5.29Helped
add negative data via negative assistant training vs. positive-only CoT KDLLaMA 7B distilled from GPT-4 · MATH test setaverage accuracy: 6.83 vs 5.59Helped
directly mixing negative and positive data in trainingLLaMA 7B · MATH test setaverage accuracy: 3.03 vs 5.29 CoT KDHurt
remove negative data (dual LoRA on positive only) (ablation)NAT · MATH test setaverage accuracy: 6.58 vs 6.81Hurt
remove negative LoRA module (ablation)NAT · MATH test setaverage accuracy: 6.05 vs 6.81Hurt
replace corrected attention with gated integration (ablation)NAT · MATH test setaverage accuracy: 5.22 vs 6.81Hurt
restrict correction range to [0,1] (no negative direction) (ablation)NAT · MATH test setaverage accuracy: 6.30 vs 6.81Hurt
adaptive self-consistency (ranking model) instead of standard SCNAT and CoT KD models · MATH test setaverage accuracy: NAT 9.84 vs SC 9.25; CoT KD 8.25 vs 7.85Helped
SC with weighted sum (WS) instead of base SCNAT, CoT KD, NCE · MATH test setaverage accuracy: e.g. NAT 9.21 vs 9.25; NCE 9.64 vs 9.69No clear effect
negative calibrated enhancement (KL-weighted self-distillation) vs. KDLLaMA 7B with GPT-3.5 data · MATH test setaverage progress: avg progress 10% (0.66) over KDHelped
negative calibrated enhancement vs. NAT (with fewer parameters)LLaMA 7B · MATH test setaverage progress: 6.5% (0.44) over NATHelped
NCE vs KD with data augmentation on OOD generalizationLLaMA 7B trained on GSM8K · GSM8K, ASDiv, MultiArith, SVAMPaccuracy: GSM8K 41.93 vs 41.55; MultiArith 88.67 vs 88.05; SVAMP 51.50 vs 50.70Mixed

Surprising findings

Limitations

Experimental setup

Datasets: MATH GSM8K ASDiv MultiArith SVAMP StrategyQA

Benchmarks: MATH GSM8K ASDiv MultiArith SVAMP StrategyQA

Metrics: accuracy

Baselines: Fine-tune CoT KD MIX CL NT UL Self-Consistency with weighted sum PaLM GPT3

Builds on: chain-of-thought distillation self-consistency LoRA unlikelihood training self-distillation

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy