Auto Dataset Builder: An Automated Annotation Framework with Vision-Language Verification and Neural Dataset Quality Scoring

The construction of high-quality training datasets remains a primary bottleneck in computer vision model development, with annotation reportedly consuming 60-80% of project time. We present Auto Dataset Builder (ADB), an automated framework that converts unlabeled images into YOLO-compatible annotated datasets through a three-stage pipeline: (1) YOLOv11 proposal generation, (2) SAM2 mask-guided bounding-box refinement, and (3) zero-shot semantic verification, for which the framework supports either a Vision-LLM backend (Qwen-VL-Chat / LLaVA) or a CPU-friendly CLIP ViT-B/32 zero-shot classifier. We additionally introduce the Neural Dataset Quality Score (Neural DQS), a lightweight regressor that estimates the expected [email protected] a YOLOv11 model will achieve after training on a given dataset, computed from six interpretable dataset-level features without requiring any model training. On 96 controlled degradation variants of COCO128, Neural DQS achieves a 5-fold cross-validated Pearson correlation of r=0.929 (R^2=0.854, p<0.001) with actual [email protected]; ablation and SHAP analyses identify CLIP-space semantic diversity as the dominant predictor, with its removal dropping r to 0.679. On a 597-image COCO2017 motorcycle subset, ADB's fully automated annotations reach 94.2% of the [email protected] obtained from human (Manual) annotations (0.441 vs. 0.468) with zero manual labeling effort. A single active-learning iteration (ADB+AL), which re-annotates the 100 most uncertain images from a held-out unlabeled pool, closes this gap entirely and matches the strongest baseline at [email protected]=0.476. Code, trained models, and benchmark dataset:- https://github.com/ericchen931209/auto-dataset-builder- https://huggingface.co/EricChenWei/neural-dqs- https://huggingface.co/datasets/EricChenWei/neural-dqs-benchmark

Authors

Institutions

Publication Details

Journal
Zenodo (CERN European Organization for Nuclear Research)
Published
2026-06-13
DOI
https://doi.org/10.5281/zenodo.20675895
Primary Topic
Multimodal Machine Learning Applications
Type
preprint
Controls
|||
ALL TIME
JAN
FEB
MAR
APR
MAY
JUN
JUL
AUG
SEP
OCT
preprint

Auto Dataset Builder: An Automated Annotation Framework with Vision-Language Verification and Neural Dataset Quality Scoring

Yuehua Chen
Zenodo (CERN European Organization for Nuclear Research)
Multimodal Machine Learning Applications
preprint

Auto Dataset Builder: An Automated Annotation Framework with Vision-Language Verification and Neural Dataset Quality Scoring

Yuehua Chen
preprint en

Abstract

The construction of high-quality training datasets remains a primary bottleneck in computer vision model development, with annotation reportedly consuming 60-80% of project time. We present Auto Dataset Builder (ADB), an automated framework that converts unlabeled images into YOLO-compatible annotated datasets through a three-stage pipeline: (1) YOLOv11 proposal generation, (2) SAM2 mask-guided bounding-box refinement, and (3) zero-shot semantic verification, for which the framework supports either a Vision-LLM backend (Qwen-VL-Chat / LLaVA) or a CPU-friendly CLIP ViT-B/32 zero-shot classifier. We additionally introduce the Neural Dataset Quality Score (Neural DQS), a lightweight regressor that estimates the expected [email protected] a YOLOv11 model will achieve after training on a given dataset, computed from six interpretable dataset-level features without requiring any model training. On 96 controlled degradation variants of COCO128, Neural DQS achieves a 5-fold cross-validated Pearson correlation of r=0.929 (R^2=0.854, p<0.001) with actual [email protected]; ablation and SHAP analyses identify CLIP-space semantic diversity as the dominant predictor, with its removal dropping r to 0.679. On a 597-image COCO2017 motorcycle subset, ADB's fully automated annotations reach 94.2% of the [email protected] obtained from human (Manual) annotations (0.441 vs. 0.468) with zero manual labeling effort. A single active-learning iteration (ADB+AL), which re-annotates the 100 most uncertain images from a held-out unlabeled pool, closes this gap entirely and matches the strongest baseline at [email protected]=0.476. Code, trained models, and benchmark dataset:- https://github.com/ericchen931209/auto-dataset-builder- https://huggingface.co/EricChenWei/neural-dqs- https://huggingface.co/datasets/EricChenWei/neural-dqs-benchmark

Zenodo (CERN European Organization for Nuclear Research)
National Yunlin University of Science and Technology (TW)
Multimodal Machine Learning Applications
AI Navigator

Ask Laika to Summarize, Analyze, and Connect papers live on the map.

Summarize Papers & Methodologies

Extract key findings, datasets, and comparative methods across publications.

Benchmark Rankings & Visual Analytics

Rank top research institutions, authors, funders, topics, and journals by Field-Weighted Citation Impact (FWCI) and paper volume with instant charts.

Connect Distant Disciplines

Bridge topological clusters on the map to find hidden collaborative intersections.