Unsupervised Feature Transformation via In-context Generation, Generator-critic LLM Agents, and Duet-play Teaming
On the map: LLM agents and tool use → Web, search, and research agents
Abstract (arXiv)
Feature transformation involves generating a new set of features from the original dataset to enhance the data's utility. In certain domains like material performance screening, dimensionality is large and collecting labels is expensive and lengthy. It highly necessitates transforming feature spaces efficiently and without supervision to enhance data readiness and AI utility. However, existing methods fall short in efficient navigation of a vast space of feature combinations, and are mostly designed for supervised settings. To fill this gap, our unique perspective is to leverage a generator-critic duet-play teaming framework using LLM agents and in-context learning to derive pseudo-supervision from unsupervised data. The framework consists of three interconnected steps: (1) Critic agent diagnoses data to generate actionable advice, (2) Generator agent produces tokenized feature transformations guided by the critic's advice, and (3) Iterative refinement ensures continuous improvement through feedback between agents. The generator-critic framework can be generalized to human-agent collaborative generation, by replacing the critic agent with human experts. Extensive experiments demonstrate that the proposed framework outperforms even supervised baselines in feature transformation efficiency, robustness, and practical applicability across diverse datasets.
Key facts. Cited by 11 papers in the map (data through 14 Oct 2026). Placed in Web, search, and research agents, one of 16 directions in LLM agents and tool use. Builds on 21 of its 28 references that are in the map. First posted on arXiv 30 Apr 2025.
Builds on
21 of 28 references.
References · 21
- 2023RLHF-V: Towards Trustworthy MLLMs via Behavior Alignment from Fine-grained Correctional Human Feedbackcited by 412
- 2023Large Language Models in Finance: A Surveycited by 226
- 2023Large Language Models for Automated Data Science: Introducing CAAFE for Context-Aware Automated Feature Engineeringcited by 127
- 2023Is RLHF More Difficult than Standard RL?cited by 96
- 2017Feature Engineering for Predictive Modeling using Reinforcement Learningcited by 78
- 2019The autofeat Python Library for Automated Feature Engineering and Selectioncited by 73
- 2025MixLLM: Dynamic Routing in Mixed Large Language Modelscited by 59
- 2022OpenFE: Automated Feature Generation with Expert-level Performancecited by 52
- 2022Augmenting Interpretable Models with LLMs during Trainingcited by 38
- 2022Group-wise Reinforcement Feature Generation for Optimal and Explainable Representation Space Reconstructioncited by 35
- 2023Self-optimizing Feature Generation via Categorical Hashing Representation and Hierarchical Reinforcement Crossingcited by 30
- 2024Evolutionary Large Language Model for Automated Feature Transformationcited by 29
9 more
- 2024Dynamic and Adaptive Feature Generation with LLMcited by 27
- 2024LLM-Enhanced User-Item Interactions: Leveraging Edge Information for Optimized Recommendationscited by 24
- 2023Reinforcement-Enhanced Autoregressive Feature Transformation: Gradient-steered Search in Continuous Space for Postfix Expressionscited by 23
- 2024Unsupervised Generative Feature Transformation via Graph Contrastive Pre-training and Multi-objective Fine-tuningcited by 21
- 2024Knockoff-Guided Feature Selection via A Single Pre-trained Reinforced Agentcited by 15
- 2024Neuro-Symbolic Embedding for Short and Effective Feature Selection via Autoregressive Generationcited by 14
- 2025A Survey on Data-Centric AI: Tabular Learning from Reinforcement Learning and Generative AI Perspectivecited by 14
- 2024Reinforcement Feature Transformation for Polymer Property Performance Predictioncited by 13
- 2024Topology-aware Reinforcement Feature Space Reconstruction for Graph Datacited by 11
Built on by
Citing papers per year: 2025: 9; 2026: 2.
11 papers.
- 2025The FM Agentcited by 21
- 2025Efficient Post-Training Refinement of Latent Reasoning in Large Language Modelscited by 7
- 2025Teaching Time Series to See and Speak: Forecasting with Aligned Visual and Textual Perspectivescited by 7
- 2025Autonomous Data Agents: A New Opportunity for Smart Datacited by 6
- 2025LLM-ML Teaming: Integrated Symbolic Decoding and Gradient Search for Valid and Stable Generative Feature Transformationcited by 5
- 2025Bridging the Domain Gap in Equation Distillation with Reinforcement Feedbackcited by 4
- 2025Distribution Shift Aware Neural Tabular Learningcited by 4
- 2025StorageXTuner: An LLM Agent-Driven Automatic Tuning Framework for Heterogeneous Storage Systemscited by 1
- 2025Dataforge: Agentic Platform for Autonomous Data Engineeringcited by 1
- 2026Evolving Demonstration Optimization for Chain-of-Thought Feature Transformationcited by 0
- 2026LATTEArena: An Evaluation Framework for LLM-powered Tabular Feature Engineering (Extended Version)cited by 0
- 2026LATTEArena: An Evaluation Framework for LLM-powered Tabular Feature Engineering (Extended Version)8 Jun 2026
- 2026Evolving Demonstration Optimization for Chain-of-Thought Feature Transformation13 Feb 2026
- 2025Dataforge: Agentic Platform for Autonomous Data Engineering9 Nov 2025
- 2025The FM Agent30 Oct 2025
- 2025StorageXTuner: An LLM Agent-Driven Automatic Tuning Framework for Heterogeneous Storage Systems28 Oct 2025
- 2025Autonomous Data Agents: A New Opportunity for Smart Data23 Sep 2025
- 2025Distribution Shift Aware Neural Tabular Learning27 Aug 2025
- 2025Teaching Time Series to See and Speak: Forecasting with Aligned Visual and Textual Perspectives30 Jun 2025
- 2025LLM-ML Teaming: Integrated Symbolic Decoding and Gradient Search for Valid and Stable Generative Feature Transformation10 Jun 2025
- 2025Efficient Post-Training Refinement of Latent Reasoning in Large Language Models10 Jun 2025
- 2025Bridging the Domain Gap in Equation Distillation with Reinforcement Feedback21 May 2025
Similar, not cited
Close in wording, no citation either way
- 2026Hierarchical and Permutation-Invariant Feature Transformation Learning via Policy-Guided Embedding Search
- 2025Heterogeneous Multi-Agent Reinforcement Learning with Attention for Cooperative and Scalable Feature Transformation
- 2025Two-Stage Feature Generation with Transformer and Reinforcement Learning
- 2025Agentic Feature Augmentation: Unifying Selection and Generation with Teaming, Planning, and Memories
- 2026Memory-Augmented LLM-based Multi-Agent System for Automated Feature Generation on Tabular Data
- 2025Dual-Agent Reinforcement Learning for Automated Feature Generation
- 2026Towards Reliable ML Feature Engineering via Planning in Constrained-Topology of LLM Agents
- 2025Collaborative Multi-Agent Reinforcement Learning for Automated Feature Transformation with Graph-Driven Path Optimization