Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Automatic Dataset Construction (ADC): Sample Collection, Data Curation, and Beyond

About

Large-scale data collection is essential for developing personalized training data, mitigating the shortage of training data, and fine-tuning specialized models. However, creating high-quality datasets quickly and accurately remains a challenge due to annotation errors, the substantial time and costs associated with human labor. To address these issues, we propose Automatic Dataset Construction (ADC), an innovative methodology that automates dataset creation with negligible cost and high efficiency. Taking the image classification task as a starting point, ADC leverages LLMs for the detailed class design and code generation to collect relevant samples via search engines, significantly reducing the need for manual annotation and speeding up the data generation process. To demonstrate ADC at scale, we construct Clothing-ADC: a dataset of over 1 million images spanning 12 main classes and 12,000 fine-grained subclasses. Our automated curation achieves 79\% agreement with human annotators and reduces label noise from 22.2\% to 10.7\%. Despite these advantages, ADC also encounters real-world challenges such as label errors (label noise) and imbalanced data distributions (label bias). We provide open-source software that incorporates existing methods for label error detection, robust learning under noisy and biased data, ensuring a higher-quality training data and more robust model training procedure. Furthermore, we design three benchmark datasets focused on label noise detection, label noise learning, and class-imbalanced learning. These datasets are vital because there are few existing datasets specifically for label noise detection, despite its importance. Finally, we evaluate the performance of existing popular methods on these datasets, thereby facilitating further research in the field.

Minghao Liu, Zonglin Di, Jiaheng Wei, Zhongruo Wang, Hengxiang Zhang, Ruixuan Xiao, Haoyu Wang, Jinlong Pang, Hao Chen, Ankit Shah, Hongxin Wei, Xinlei He, Zhaowei Zhao, Haobo Wang, Lei Feng, Jindong Wang, James Davis, Yang Liu• 2024

Related benchmarks

TaskDatasetResultRank
Label Noise LearningClothing-ADC clean full (test)--
12
Label Noise LearningClothing-ADC held-out clean tiny (test)--
12
Image ClassificationVarious Image Classification Datasets (iNaturalist, WebVision, ANIMAL-10N, CIFAR-10N, CIFAR-100N, Food-101N, Clothing1M, Clothing-ADC)
Noise Rate22.2
8
Image ClassificationClothing-ADC CLT rho=10 (test)--
7
Image ClassificationClothing-ADC CLT rho=50 (test)--
7
Image ClassificationClothing-ADC CLT rho=100 (test)--
7
Label Noise DetectionClothing-ADC 20,000 samples subset--
4
Showing 7 of 7 rows

Other info

Follow for update