用软引导强化学习自动优化数据预处理流程
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 官方产品;中文卡片由大模型生成,请以原文为准。