arXiv:2507.13710cs.DBcs.LG2025-07

用软引导强化学习自动优化数据预处理流程

SoftPipe: A Soft-Guided Reinforcement Learning Framework for Automated Data Preparation

  • 用大模型生成策略先验,结合质量评分与长期价值做决策
  • 在18个数据集上提升管道质量13.9%,收敛速度加快2.8倍
  • 适合想自动化数据清洗与特征工程的研究者和工程师

数据准备是机器学习生命周期中的基础环节,但因其组合爆炸的搜索空间而极具挑战。现有强化学习方法依赖刚性约束来压缩搜索空间,常导致次优解。本文提出SoftPipe,一种基于软引导的强化学习框架,将动作选择建模为贝叶斯推断问题。高阶战略先验由大语言模型生成,与两个来源的估计值协同:监督学习排序模型提供的细粒度质量评分,以及智能体Q函数的长期价值估计。在18个多样化数据集上的实验表明,SoftPipe相比现有方法,管道质量最高提升13.9%,收敛速度加快2.8倍。

原文摘要 · Abstract (English)

Data preparation is a foundational yet notoriously challenging component of the machine learning lifecycle, characterized by a vast combinatorial search space. While reinforcement learning (RL) offers a promising direction, state-of-the-art methods suffer from a critical limitation: to manage the search space, they rely on rigid ``hard constraints'' that prematurely prune the search space and often preclude optimal solutions. To address this, we introduce SoftPipe, a novel RL framework that replaces these constraints with a flexible ``soft guidance'' paradigm. SoftPipe formulates action selection as a Bayesian inference problem. A high-level strategic prior, generated by a Large Language Model (LLM), probabilistically guides exploration. This prior is combined with empirical estimators from two sources through a collaborative process: a fine-grained quality score from a supervised Learning-to-Rank (LTR) model and a long-term value estimate from the agent's Q-function. Through extensive experiments on 18 diverse datasets, we demonstrate that SoftPipe achieves up to a 13.9\% improvement in pipeline quality and 2.8$\times$ faster convergence compared to existing methods.

自动化数据预处理强化学习大模型应用

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