Auto-Fill Knowledge Specialist (Qwen3-8B)

The knowledge specialist of Auto-Fill: Learning to Predict Missing Values Accurately with Specialist Language Models (PVLDB 19(11), 2026 — arXiv:2607.19847). Given a table with one cell marked [MISSING], this model answers directly, without chain-of-thought, and is meant for cells whose value is a matter of world knowledge or of patterns visible in the table (entity attributes, codes, names, dates).

Auto-Fill runs three specialists in parallel — knowledge, reasoning and coding — calibrates their confidences with isotonic regression and returns the most confident answer (or abstains). Sibling specialists: Reasoning · Coding.

Model details

Base model Qwen/Qwen3-8B
Training full-parameter SFT with ms-swift, DeepSpeed ZeRO-3, bf16
Data 30,000 examples: direct (table, value) pairs — one cell masked per table, the original value as target.
Hyper-parameters 2 epochs, lr 1e-5, cosine schedule, weight decay 0.1, max length 40,960, effective batch size 16
Hardware 4× A100 80 GB
Confidence signal the log-probability of the generated value tokens (obtained from vLLM logprobs). Calibrated with isotonic regression fitted on a held-out validation split (calibrators.json).
Decoding used in the paper temperature 0.1, max new tokens 4096, default Qwen3 chat template

Training tables come from public sources only (spreadsheets crawled from a search-engine index, public BI models, Wikipedia, nationalarchives.gov.uk, GitHub CSV/Parquet files); one cell per table is masked and its original value is the target.

Prompt and output format

The table is serialized as a Markdown pipe table (pandas.DataFrame.to_markdown(index=False, tablefmt="pipe")) with the cell to fill written as [MISSING]. The user message is exactly (see autofill/utils/prompts.py):

Please fill in the missing value in the input table. The missing value is denoted by '[MISSING]'. Please return the value filled in JSON format: {"value": "filled_value"}.

Input Table:
<markdown table>

Expected output: {"value": "<filled value>"} — no reasoning text.

Usage

With the code repository (recommended) — runs the full ensemble on one table:

python inference/run_specialists.py \
    --table          /path/to/table.csv \
    --knowledge_path lyrain2001/Auto-Fill-Qwen3-8B-Knowledge \
    --reasoning_path lyrain2001/Auto-Fill-Qwen3-8B-Reasoning \
    --coding_path    lyrain2001/Auto-Fill-Qwen3-8B-Coding \
    --calibrators    checkpoints/calibrators.json \
    --gpu_ids        0,1,2

or this specialist alone on the benchmark:

python inference/run_benchmark.py --mode knowledge --model_path lyrain2001/Auto-Fill-Qwen3-8B-Knowledge \
    --dataset Gov-CSV --benchmark Auto-Fill-Benchmark/sample200 --gpu_ids 0

Minimal vLLM example

import pandas as pd
from vllm import LLM, SamplingParams

llm = LLM(model="lyrain2001/Auto-Fill-Qwen3-8B-Knowledge", dtype="bfloat16", max_model_len=40960)
table = pd.read_csv("table.csv", dtype=str).to_markdown(index=False, tablefmt="pipe", disable_numparse=True)
prompt = PROMPT + table   # PROMPT = the user message above, up to and including "Input Table:\n"
text = llm.get_tokenizer().apply_chat_template(
    [{"role": "user", "content": prompt}], tokenize=False, add_generation_prompt=True)
out = llm.generate([text], SamplingParams(temperature=0.1, max_tokens=4096))
print(out[0].outputs[0].text)

Results

Recall@Precision=0.9 on the Auto-Fill benchmark (200 cases per dataset; from the paper's specialist ablation):

Pub-XLS Pub-BI Pub-Wiki Gov-CSV Git-Parquet Ent-CSV* Ent-XLS* Pub-Web Rel-AR Rel-FD Rel-ST Mean
Knowledge specialist alone 0.335 0.580 0.210 0.450 0.565 0.570 0.640 0.235 0.550 0.865 0.960 0.542
Auto-Fill (all three + ensemble) 0.525 0.615 0.285 0.500 0.590 0.585 0.660 0.275 0.990 0.890 0.995 0.628

* Ent-CSV / Ent-XLS are proprietary enterprise datasets that are not part of the public benchmark.

Limitations

  • Trained and evaluated on English-language tables with one missing cell per table; tables were serialized with at most 40,960 tokens.
  • The model can be wrong with high confidence on cells that require knowledge outside the table; use the calibrated confidence and abstain below a threshold, as in the paper.
  • Generated code (coding specialist) should be executed in a sandbox.

Citation

@article{liu2026autofill,
  title={Auto-Fill: Learning to Predict Missing Values Accurately with Specialist Language Models},
  author={Liu, Yurong and He, Yeye and Dong, Haoyu and Xing, Junjie and Han, Shi and Zhang, Dongmei and Chaudhuri, Surajit},
  journal={Proceedings of the VLDB Endowment},
  volume={19},
  number={11},
  pages={3160--3173},
  year={2026}
}
Downloads last month
80
Safetensors
Model size
308k params
Tensor type
BF16
·
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Model tree for lyrain2001/Auto-Fill-Qwen3-8B-Knowledge

Finetuned
Qwen/Qwen3-8B
Finetuned
(2017)
this model

Dataset used to train lyrain2001/Auto-Fill-Qwen3-8B-Knowledge

Paper for lyrain2001/Auto-Fill-Qwen3-8B-Knowledge