atlas

Research map / Architectures and efficiency / Attention and transformers

StreamingDialogue: Prolonged Dialogue Learning via Long Context Compression with Minimal Losses

NeurIPS 2024 · Architectures and efficiency › Attention and transformers · approach: Sparse attention for long-context LLMs · 3 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 StreamingDialogue, which compresses dialogue history into End-of-Utterance "conversational attention sinks" with attention masking and two self-learning objectives (SMR and LMR) to enable prolonged dialogue with lower cost than dense attention.

Problem

Standard LLMs struggle with long multi-turn dialogues due to quadratic attention cost and loss of historical information (consistency) when using sparse or sliding-window attention.

Approach

Observes that End-of-Utterance tokens aggregate attention like StreamingLLM's initial-token sinks; caches only the first token, EoU sink tokens, and the most recent two utterances, and trains with short-memory reconstruction (SMR) and long-memory reactivation (LMR) objectives under a custom attention mask to make EoU tokens compress utterance content.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Use conv-attn sink attention instead of StreamingLLM attentionLlama-2-7B StreamingDialogue · MSC testBLEU-avg: 16.76 -> 19.33Helped
Use conv-attn sink attention instead of StreamingLLM attentionLlama-2-7B StreamingDialogue · MSC testROUGE-L: 14.21 -> 15.86Helped
Use conv-attn sink attention instead of StreamingLLM attentionLlama-2-7B StreamingDialogue · PersonaChat testDistinct-2: 32.64% to 37.23%Helped
Use conv-attn sink attention instead of StreamingLLM attention, without trainingLlama-2-7B-Chat · MSC testBLEU-avg: 20.16 -> 20.19Helped
Use conv-attn sink attention instead of StreamingLLM attention, without trainingLlama-3-8B-Instruct · MSC testBLEU-avg: 16.48 -> 16.77Helped
Use conv-attn sink attention instead of StreamingLLM attention, without trainingMistral-7B · MSC testBLEU-avg: 12.75 -> 13.33Helped
Add SMR and LMR (ablation)StreamingDialogue · MSC testBLEU-avg: 17.32 -> 19.33Helped
Add SMR and LMR (ablation)StreamingDialogue · MSC testROUGE-L: 10.25 -> 15.86Helped
Add SMR and LMR (ablation)StreamingDialogue · MSC testDistinct-3: 46.15 -> 50.27Helped
Add SMR (ablation)StreamingDialogue · MSC testBLEU-avg: 18.87 -> 19.33Helped
Add SMR (ablation)StreamingDialogue · MSC testROUGE-L: 15.66 -> 15.86Helped
Add LMR (ablation)StreamingDialogue · MSC testBLEU-avg: 18.25 -> 19.33Helped

Surprising findings

Limitations

Experimental setup

Datasets: PersonaChat MSC (Multi-Session Chat) Topical-Chat MultiWOZ

Benchmarks: PersonaChat MSC Topical-Chat MultiWOZ

Metrics: perplexity BLEU ROUGE Distinct USL-H Dial-M C score

Baselines: Dense attention Local attention Big Bird StreamingLLM MemBART HRED VHRED

Builds on: StreamingLLM Longformer Big Bird MemBART

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