FlowPipe: LLM-Enhanced Conditional Generative Flow Networks for Data Preparation Pipeline Construction
Not on the map. Why?
Abstract (arXiv)
Data preparation pipelines improve data quality in machine learning by transforming raw tables into learning-ready data through sequential cleaning and feature transformation operators. However, automatically constructing such pipelines is computationally difficult because operator sequences are combinatorial and end-to-end evaluation is expensive. Existing state-of-the-art (SOTA) Multi-DQN methods still face three key limitations: decoupled value estimators weaken long-horizon credit assignment, dataset context is only weakly injected into the policy, and exploration is inefficient in a sparse search space with many invalid states. To address these issues, we propose FlowPipe, a unified framework that formulates pipeline synthesis as conditional probabilistic flow generation over a directed acyclic graph. FlowPipe uses Conditional Generative Flow Networks (C-GFlowNets) with a Trajectory Balance objective to connect terminal validation rewards with early pipeline decisions. It further introduces Deep Semantic Modulation through Feature-wise Linear Modulation (FiLM), allowing LLM-derived logical priors to condition the policy's internal activations according to dataset semantics. In addition, FlowPipe incorporates failure awareness into the flow objective to avoid invalid states and concentrate search on high-potential regions. Experiments on two benchmark suites with 74 real-world datasets show that FlowPipe outperforms SOTA baselines, improving accuracy by 11.96% on average and achieving 12.5x faster training convergence. Source code is available at https://github.com/KunyuNi/FlowPipe.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 26 of its 59 references that are in the map. First posted on arXiv 23 Jun 2026.
Builds on
26 of 59 references.
References · 26
- 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,399
- 2017Proximal Policy Optimization Algorithmscited by 15,215
- 2024The Llama 3 Herd of Modelscited by 14,715
- 2017FiLM: Visual Reasoning with a General Conditioning Layercited by 2,333
- 2018DARTS: Differentiable Architecture Searchcited by 1,798
- 2015Deep Reinforcement Learning with Double Q-learningcited by 1,745
- 2018Efficient Neural Architecture Search via Parameter Sharingcited by 837
- 2023Towards General Text Embeddings with Multi-stage Contrastive Learningcited by 723
- 2016Hyperband: A Novel Bandit-Based Approach to Hyperparameter Optimizationcited by 699
- 2014OpenML: networked science in machine learningcited by 526
- 2020TaBERT: Pretraining for Joint Understanding of Textual and Tabular Datacited by 398
- 2021GFlowNet Foundationscited by 290
14 more
- 2022Can Foundation Models Wrangle Your Data?cited by 175
- 2023Tree-Structured Parzen Estimator: Understanding Its Algorithm Components and Their Roles for Better Empirical Performancecited by 147
- 2017Feature Engineering for Predictive Modeling using Reinforcement Learningcited by 78
- 2023Large Language Models as Data Preprocessorscited by 34
- 2018OBOE: Collaborative Filtering for AutoML Model Selectioncited by 31
- 2020AutoFS: Automated Feature Selection via Diversity-aware Interactive Reinforcement Learningcited by 24
- 2021Production Machine Learning Pipelines: Empirical Analysis and Optimization Opportunitiescited by 23
- 2019DeepLine: AutoML Tool for Pipelines Generation using Deep Reinforcement Learning and Hierarchical Actions Filteringcited by 21
- 2019SystemDS: A Declarative Machine Learning System for the End-to-End Data Science Lifecyclecited by 20
- 2020MLCask: Efficient Management of Component Evolution in Collaborative Data Analytics Pipelinescited by 15
- 2023DiffPrep: Differentiable Data Preprocessing Pipeline Search for Learning over Tabular Datacited by 9
- 2021Auto-Pipeline: Synthesizing Complex Data Pipelines By-Target Using Reinforcement Learning and Searchcited by 8
- 2022DiffML: End-to-end Differentiable ML Pipelinescited by 7
- 2023Evaluation of Dataframe Libraries for Data Preparation on a Single Machinecited by 3
Built on by
Not cited yet.