arXiv:2503.06664cs.LGcs.AI2025-03被引 18

用大模型自动清洗表格数据,提升机器学习模型性能

Exploring LLM Agents for Cleaning Tabular Machine Learning Datasets

  • 让大模型结合上下文和迭代反馈识别错误数据
  • 可有效修正单行内逻辑错误和异常值,准确率超75%
  • 适合需要快速预处理的初学者或非专业数据清洗场景

高质量、无错误的数据集是构建可靠、准确且无偏机器学习模型的关键。然而,现实世界数据常因传感器故障、输入错误或跨源数据整合不当而存在缺陷,严重降低模型性能。检测与修复这些错误通常需定制化方案并依赖大量领域知识,导致自动化困难,过程繁琐耗时。本研究探索大型语言模型(LLMs)在减轻人工数据清洗负担方面的潜力。我们在多个经过人为注入错误的Kaggle数据集上进行实验,让一个配备Python的LLM在不修改训练流程或进行特征工程的前提下,负责清洗训练数据以提升学习算法表现。结果表明,该模型能利用同一行中其他特征的上下文信息,以及前序迭代的反馈,有效识别并修正诸如逻辑矛盾值和异常值等错误;但在识别需跨多行理解数据分布的复杂问题(如趋势、系统性偏差)方面仍存在局限。

原文摘要 · Abstract (English)

High-quality, error-free datasets are a key ingredient in building reliable, accurate, and unbiased machine learning (ML) models. However, real world datasets often suffer from errors due to sensor malfunctions, data entry mistakes, or improper data integration across multiple sources that can severely degrade model performance. Detecting and correcting these issues typically require tailor-made solutions and demand extensive domain expertise. Consequently, automation is challenging, rendering the process labor-intensive and tedious. In this study, we investigate whether Large Language Models (LLMs) can help alleviate the burden of manual data cleaning. We set up an experiment in which an LLM, paired with Python, is tasked with cleaning the training dataset to improve the performance of a learning algorithm without having the ability to modify the training pipeline or perform any feature engineering. We run this experiment on multiple Kaggle datasets that have been intentionally corrupted with errors. Our results show that LLMs can identify and correct erroneous entries, such as illogical values or outlier, by leveraging contextual information from other features within the same row, as well as feedback from previous iterations. However, they struggle to detect more complex errors that require understanding data distribution across multiple rows, such as trends and biases.

数据清洗大模型表格数据自动化

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。