STIG: encode multi-stage introduction logic as stage tokens so an LLM writes an entire Introduction in one inference

December 28, 20258 min

Overview

Decision SnapshotNeeds Validation

Method is practical and reproducible for similar academic paper domains; results are supported by automatic metrics, human eval, and an ablation, but generalization beyond conference-style CS introductions needs extra validation.

Citations0

Evidence Strength0.70

Confidence0.80

Risk Signals9

Trust Signals

Findings with numeric evidence: 4/4

Findings with evidence refs: 4/4

Results with explicit delta: 6/6

Reproducibility

Status: Code + data available

Open source: Partial

At A Glance

Cost impact: 60%

Production readiness: 60%

Novelty: 60%

Authors

Meicong Zhang, Tiancheng su, Guoxiu He

Links

Abstract / PDF

Why It Matters For Business

Convert multi-call agent pipelines into a single finetuned model with stage tokens to reduce API calls, cut token costs, and produce more structurally coherent section drafts for academic-style content.

Who Should Care

Summary TLDR

The paper proposes STIG, a simple finetuning method that encodes a writing workflow as explicit stage tokens so a single LLM call can produce an entire academic Introduction. Authors build a dataset from ~3,800 ACL papers (test on 1,176 ACL 2025 papers), add eight stage token pairs (outline + content per subsection), and finetune open models (Qwen2.5-7B, Llama3.1-8B). STIG yields better structural coherence and content coverage than multi-agent/outline pipelines, and uses tokens more efficiently (claimed 3.3× effective token rate vs. AutoSurvey). Code and dataset are provided in supplements.

Problem Statement

Agentic workflows split writing into many agent calls (outline → draft → integrate) which causes long reasoning chains, error accumulation, high token use, latency, and fragile designs that need manual reconfiguration across domains.

Main Contribution

Introduce STIG: parametric stage tokens that encode multi-stage introduction logic into the model so a single inference outputs all stages.

Create an annotated training corpus from ~3,800 ACL papers (1,176 test on ACL 2025) with eight stage labels (outline+content for four subsections).

Key Findings

STIG raises structural rationality (SR) substantially versus agentic baselines.

NumbersSR 0.832 (STIG) vs 0.658 (AutoSurvey) on Qwen2.5-7B

Practical UseIf your goal is clear subsection structure, finetune with stage tokens rather than chaining agents.

Evidence RefTable 1 (Qwen2.5-7B-Instruct SR columns)

STIG matches or improves semantic similarity and content coverage over baselines.

NumbersSS 0.977, CC 0.442 (Qwen2.5-7B STIG) vs CC 0.333 (AutoSurvey)

Practical UseYou can keep content fidelity while simplifying generation to a single call.

Evidence RefTable 1 (SS and CC columns)

Results

MetricValueBaselineDeltaSplit / DatasetEvidenceEvidence Ref
Semantic Similarity (SS)0.977 (Qwen2.5-7B STIG)0.966 (AutoSurvey)+0.011ACL 2025 test (1,176 papers)STIG SS 0.977 vs AutoSurvey 0.966Table 1
Structural Rationality (SR)0.832 (Qwen2.5-7B STIG)0.658 (AutoSurvey)+0.174 (17.4% absolute)ACL 2025 test (1,176 papers)SR 0.832 vs 0.658Table 1 and Section 5.1

What To Try In 7 Days

Extract 100 domain examples of introductions and add 4 subsection outlines + content pairs.

Add a small set of custom stage tokens (outline/content per subsection) and instruction-finetune an existing 7B instruction model.

Compare single-call generation vs your current agent pipeline on structure (SR) and token consumption.

Agent Features

Memory
Max context length up to 8K tokens during training/inference
Planning
Stage-guided generation (fixed ordered stages)Outline→content sequencing encoded in tokens
Tool Use
LoRALLaMA-Factory (finetuning)MinerU (parsing PDFs)ZeRO3 (memory optimization)
Frameworks
LLaMA-FactoryZeRO3LoRAMinerU
Architectures
Instruction-tuned LLM backbone (decoder-only transformer)Parametric stage tokens added to tokenizer
Collaboration
Replaces multi-agent orchestration with a single model; no external agent calls

Optimization Features

Token Efficiency
Claimed 3.3× effective generation rate vs AutoSurveyClaimed ~2× efficiency over Stage Writing FT
Infra Optimization
ZeRO3 for distributed finetuningA800 GPUs
Model Optimization
LoRA
System Optimization
Increases context window to 8K tokens
Training Optimization
Instruction finetuning with special stage tokensZeRO3 to optimize GPU memory8× A800 GPUs used
Inference Optimization
Single-inference end-to-end generation to avoid multiple API callsRemoves outlines and parses stage tokens after generation

Reproducibility

Code AvailableYes
Data AvailableYes
Open Source StatusPartial
LicenseUnknown

Risks & Boundaries

Limitations

Training uses ACL-domain introductions; may not generalize to other academic fields or non-conference formats.

Relies on high-quality annotations (outline↔content alignment); noisy labels will degrade stage learning.

When Not To Use

If you lack annotated intro-outline pairs in your domain.

When factual accuracy is critical and hallucination risk cannot be tolerated without heavy human verification.

Failure Modes

Model may hallucinate results or baselines when training data contains such fabrications.

Stage tokens could be misused, producing wrong subsection assignments if annotation patterns differ in new domains.

Core Entities

Models

Qwen2.5-7B-InstructLlama3.1-8B-InstructGPT-4oQwen2.5-32B-Instruct

Metrics

BERTScoreStructural Rationality (sentence-level misclassification rate)Content Coverage (SBERT-weighted sentence similarity)Perplexity (GPT-2 PPL)Quotation Constraint (QC)Human ranking

Datasets

ACL papers 2021-2025 (training corpus ~3,800)ACL 2025 Main Conference test set (1,176 papers)CVPR subset (102 papers) for generalization

Benchmarks

Structural Rationality (SR)Content Coverage (CC)BERTScore (SS)Perplexity (Narrative Quality NQ)

Context Entities

Models

AutoSurveyGPT (baseline prompts)Outline Writing pipeline

Datasets

ACL Anthology (source for parsing)Semantic Scholar API (for baseline abstracts)