AUGCON: automatic pipeline to generate diverse, multi-granularity SFT pairs from any corpus

May 26, 20247 min

Overview

Decision SnapshotNeeds Validation

Method is practical: uses open-source LLMs, reports compute/time, and shows gains on several benchmarks and human judgement. Ablations validate each module's value.

Citations1

Evidence Strength0.80

Confidence0.85

Risk Signals12

Trust Signals

Findings with numeric evidence: 3/3

Findings with evidence refs: 3/3

Results with explicit delta: 4/4

Reproducibility

Status: Code + data available

Open source: Yes

At A Glance

Cost impact: 60%

Production readiness: 60%

Novelty: 70%

Authors

Shanghaoran Quan

Links

Abstract / PDF / Code / Data

Why It Matters For Business

AUGCON creates high-quality, diverse SFT pairs automatically, lowering annotation costs and improving domain-adapted LLM performance for productized assistants and search/chat features.

Who Should Care

Summary TLDR

AUGCON is an automated pipeline that builds supervised fine-tuning (SFT) query–response pairs from a custom corpus. It (1) extracts multi-granularity questions via a Context-Split-Tree (CST) that recursively splits text and derives matching queries, (2) trains a contrastive scorer to rank and filter diverse queries, and (3) uses principle-driven self-alignment plus a self-improving in‑context example search to produce high-fidelity answers. AUGCON outperforms several prior context-driven generators on human evaluation (DailyM) and on four standard benchmarks, while being usable with open-source LLMs and released code/datasets.

Problem Statement

Creating high-quality, diverse query–response pairs from a private corpus is costly by hand. Existing automated methods produce redundant or single-granularity queries and lower-fidelity answers. We need an automatic, scalable way to generate multi-granularity, high-diversity SFT data that yields better fine-tuned models.

Main Contribution

Context-Split-Tree (CST): a recursive LLM-driven splitting method to derive queries that match different context granularities.

Contrastive scorer: train a lightweight scorer with contrastive learning to rank and filter queries for quality and diversity.

Key Findings

AUGCON improves accuracy on reading QA benchmarks compared to prior context-driven SFT methods.

NumbersSQuAD1.1 Acc 0.336 vs 0.314 (best baseline); TriviaQA 0.849 vs 0.825; DROP 0.350 vs 0.334; WebGLM-QA BS 0.924 vs 0.903

Practical UseUse AUGCON-generated SFT data to fine-tune LLMs for better QA performance on both short and long answers on evaluated benchmarks.

Evidence RefTable 1

AUGCON wins GPT-4 pairwise judgements vs other context-driven methods.

NumbersWins vs ETRC: 64.5% vs 35.5%; vs Context-Instruct: 60.3% vs 39.7%

Practical UseWhen judged by a high-quality LLM referee, AUGCON outputs are preferred; try AUGCON if you want higher judged quality for domain fine-tuning.

Evidence RefG.3 computation experiment / GPT-4 judge

Results

MetricValueBaselineDeltaSplit / DatasetEvidenceEvidence Ref
Accuracy0.336 ± 0.004 (AUGCON fine-tuned)Context-Instruct 0.314 ± 0.003+0.022SQuAD1.1Table 1 automatic evalTable 1
Accuracy0.849 ± 0.003 (AUGCON fine-tuned)Context-Instruct 0.825 ± 0.003+0.024TriviaQATable 1 automatic evalTable 1

What To Try In 7 Days

Run AUGCON on a small, high-value corpus (1–2k documents) and fine-tune an open-source chat model to compare against DAPT.

Use CST with a higher λ to prioritize macro-topic queries or a lower λ to harvest detailed Q&A and observe downstream task gains.

Open the released code and run the scorer + filtering pipeline to produce a compact, diverse SFT set and test model quality with a 100-query human sample.

Optimization Features

Token Efficiency

CST yields linear amount of questions relative to sentence units, controlling token generation via c

Infra Optimization
Single-node 8xA100 setup, DeepSpeed + ZeRO-2
Model Optimization
LoRA
System Optimization
Use of A100 80G multi-GPU node; generation throughput ~340 pairs/A100-hour
Training Optimization
DeepSpeed ZeRO-2 for memory efficiencyAdamW optimizer and 4 training epochs
Inference Optimization
vLLM for high-throughput LLM callsconcurrent requests (8 threads) to accelerate generation

Reproducibility

Code AvailableYes
Data AvailableYes
Open Source StatusYes
LicenseUnknown

Risks & Boundaries

Limitations

Quality depends on input context depth; shallow or noisy corpora yield weaker queries and answers.

Method requires substantial GPU resources for large corpora (reported 184 A100 hours to generate 120K pairs on DailyM).

When Not To Use

You have a tiny corpus with few coherent sentences—CST will yield limited value.

You lack GPU budget for generating and fine-tuning (AUGCON can be compute-heavy at scale).

Failure Modes

LLM hallucination during context split produces invalid subcontexts or nonsensical queries.

Scorer misranking retains low-quality queries if negative sample construction is insufficient.

Core Entities

Models

Qwen1.5-32B-ChatLlama3-70B-InstructLlama3-c-70BQwen1.5-c-32B

Metrics

AccuracyBERTScoreROUGE-LGPT-4 pairwise judgement

Datasets

SFTSQuAD1.1TriviaQADROPWebGLM-QA

Benchmarks

SQuAD1.1TriviaQADROPWebGLM-QADailyM test set

Context Entities

Models

GPT-4 (used as judge)

Datasets

Open-source magazine corpus (DailyM), public QA benchmarks