Data Wrangling Task Automation Using Code-Generating Language Models

Ensuring data quality in large tabular datasets is a critical challenge, typically addressed through data wrangling tasks. Traditional statistical methods, though efficient, cannot often understand the semantic context and deep learning approaches are resource-intensive, requiring task and dataset-specific training. To overcome these shortcomings, we present an automated system that utilizes large language models to generate executable code for 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-dependent and memory-independent tasks.

@article{akella2025_2502.15732,
  title={ Data Wrangling Task Automation Using Code-Generating Language Models },
  author={ Ashlesha Akella and Krishnasuri Narayanam },
  journal={arXiv preprint arXiv:2502.15732},
  year={ 2025 }
}