CodeGenWrangler: Data Wrangling task automation using Code-Generating Models
Ashlesha Akella, Abhijit Manatkar, Krishnasuri Narayanam, Sameep Mehta
Abstract
Assuring the data quality of tabular datasets is essential for the efficiency of the diverse tabular downstream tasks (like summarization and fact-checking). Data-wrangling tasks effectively address the challenges associated with structured data processing to improve the quality of tabular data. Traditional statistical methods handle numeric data efficiently but often fail to understand the semantic context of the textual data in tables. Deep learning approaches are resource-intensive, requiring task and dataset-specific training. Addressing these shortcomings, we present an automated system that leverages LLMs to generate executable code for data-wrangling tasks like missing value imputation, error detection, and error correction. Our system aims to identify inherent patterns in the data while leveraging external knowledge, effectively addressing both memory-independent and memory-dependent tasks.
BibTeX
@inproceedings{akella-etal-2025-codegenwrangler,
title = "{C}ode{G}en{W}rangler: Data Wrangling task automation using Code-Generating Models",
author = "Akella, Ashlesha and
Manatkar, Abhijit and
Narayanam, Krishnasuri and
Mehta, Sameep",
editor = "Chen, Weizhu and
Yang, Yi and
Kachuee, Mohammad and
Fu, Xue-Yong",
booktitle = "Proceedings of the 2025 Conference of the Nations of the Americas Chapter of the Association for Computational Linguistics: Human Language Technologies (Volume 3: Industry Track)",
month = apr,
year = "2025",
address = "Albuquerque, New Mexico",
publisher = "Association for Computational Linguistics",
url = "https://aclanthology.org/2025.naacl-industry.70/",
pages = "949--960",
ISBN = "979-8-89176-194-0"
}