mlsanityData to 14 Oct 2026

Unsupervised Feature Transformation via In-context Generation, Generator-critic LLM Agents, and Duet-play Teaming

arXiv 2504.21304 ↗First posted 30 Apr 2025cs.LGcited by 11

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

  1. 2023RLHF-V: Towards Trustworthy MLLMs via Behavior Alignment from Fine-grained Correctional Human Feedbackcited by 412
  2. 2023Large Language Models in Finance: A Surveycited by 226
  3. 2023Large Language Models for Automated Data Science: Introducing CAAFE for Context-Aware Automated Feature Engineeringcited by 127
  4. 2023Is RLHF More Difficult than Standard RL?cited by 96
  5. 2017Feature Engineering for Predictive Modeling using Reinforcement Learningcited by 78
  6. 2019The autofeat Python Library for Automated Feature Engineering and Selectioncited by 73
  7. 2025MixLLM: Dynamic Routing in Mixed Large Language Modelscited by 59
  8. 2022OpenFE: Automated Feature Generation with Expert-level Performancecited by 52
  9. 2022Augmenting Interpretable Models with LLMs during Trainingcited by 38
  10. 2022Group-wise Reinforcement Feature Generation for Optimal and Explainable Representation Space Reconstructioncited by 35
  11. 2023Self-optimizing Feature Generation via Categorical Hashing Representation and Hierarchical Reinforcement Crossingcited by 30
  12. 2024Evolutionary Large Language Model for Automated Feature Transformationcited by 29
9 more
  1. 2024Dynamic and Adaptive Feature Generation with LLMcited by 27
  2. 2024LLM-Enhanced User-Item Interactions: Leveraging Edge Information for Optimized Recommendationscited by 24
  3. 2023Reinforcement-Enhanced Autoregressive Feature Transformation: Gradient-steered Search in Continuous Space for Postfix Expressionscited by 23
  4. 2024Unsupervised Generative Feature Transformation via Graph Contrastive Pre-training and Multi-objective Fine-tuningcited by 21
  5. 2024Knockoff-Guided Feature Selection via A Single Pre-trained Reinforced Agentcited by 15
  6. 2024Neuro-Symbolic Embedding for Short and Effective Feature Selection via Autoregressive Generationcited by 14
  7. 2025A Survey on Data-Centric AI: Tabular Learning from Reinforcement Learning and Generative AI Perspectivecited by 14
  8. 2024Reinforcement Feature Transformation for Polymer Property Performance Predictioncited by 13
  9. 2024Topology-aware Reinforcement Feature Space Reconstruction for Graph Datacited by 11

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2025: 9; 2026: 2.

11 papers.

Similar, not cited

Close in wording, no citation either way

  1. 2026Hierarchical and Permutation-Invariant Feature Transformation Learning via Policy-Guided Embedding Search
  2. 2025Heterogeneous Multi-Agent Reinforcement Learning with Attention for Cooperative and Scalable Feature Transformation
  3. 2025Two-Stage Feature Generation with Transformer and Reinforcement Learning
  4. 2025Agentic Feature Augmentation: Unifying Selection and Generation with Teaming, Planning, and Memories
  5. 2026Memory-Augmented LLM-based Multi-Agent System for Automated Feature Generation on Tabular Data
  6. 2025Dual-Agent Reinforcement Learning for Automated Feature Generation
  7. 2026Towards Reliable ML Feature Engineering via Planning in Constrained-Topology of LLM Agents
  8. 2025Collaborative Multi-Agent Reinforcement Learning for Automated Feature Transformation with Graph-Driven Path Optimization