Recent research has leveraged Large Language Models (LLMs) to enhance Automated Feature Engineering (AutoFE) through semantic descriptions and trajectory-based prompting. However, there exist two challenges that limit their applicability and scalability in long-horizon optimization: (1) semantic metadata is unavailable in many practical settings, and (2) trajectory accumulation increases the risk of exceeding the context window, while without it, the generation process can become unstable, leading to becoming stuck in the local optima and a high duplicate rate of generated features. To this end, we propose a SHAP-enhanced Implicit-trajectory Generation for Metadata-free AutoFE (SIGMA), a scalable constant-context optimization framework. SIGMA leverages SHAP values to provide task-aware signals for guiding group feature generation instead of semantic information. In addition, we adopt an EXposed-feature Implicit Trajectory (EXIT) approach, where the exposed features in the prompt implicitly represent the trajectory. Empirical results demonstrate that SIGMA achieves performance comparable to the state-of-the-art (SOTA) LLM baselines with a nearly constant prompt length. Notably, EXIT significantly reduces the duplicate ratio of generated features from 37.2% to 6.8%. At the same time, SIGMA matches traditional SOTA performance with only 5.4 features on average, demonstrating substantial efficiency gains in feature utilization.
Figures & tables
Figure 1: The overview of SIGMA.
Dataset
Baseline (w.o. AutoFE)
CAAFE
OCTree
SIGMA (ours)
eucalyptus
64.94 ± 1.70
65.27 ± 2.04
65.12 ± 1.24
66.39 ± 2.29
diabetes
73.55 ± 5.40
73.58 ± 3.62
73.62 ± 4.37
74.82 ± 3.16
credit-g
74.28 ± 3.11
73.46 ± 2.90
74.56 ± 1.73
74.84 ± 2.19
pc1
92.71 ± 1.21
92.64 ± 0.95
92.83 ± 1.42
92.22 ± 1.06
cmc
51.22 ± 2.60
51.76 ± 2.01
49.97 ± 1.95
51.46 ± 2.50
wine
80.53 ± 0.39
80.07 ± 1.66
80.15 ± 1.36
79.13 ± 0.89
Table 1: F1-score comparison of LLM-based AutoFE using Qwen3-4B-Instruct-2507. The best results are highlighted in bold , and the second-best results are underlined .
Figure 2: Efficiency analysis of methods: (a) the average prompt token trends of LLM-based AutoFE methods. Since SIGMA generates two features at each iteration, the final step is 25. (b) the feature-efficiency comparison with traditional AutoFE.
Dataset
Baseline (w.o. AutoFE)
AutoFeat
DFS
OpenFE
SIGMA (Ours)
eucalyptus
64.94 ± 1.70
65.72 ± 2.93
66.71 ± 1.71
65.67 ± 0.41
66.39 ± 2.29
diabetes
73.55 ± 5.40
74.87 ± 2.34
74.32 ± 2.95
75.93 ± 2.82
74.82 ± 3.16
credit-g
74.28 ± 3.11
73.97 ± 2.33
74.18 ± 3.05
73.79 ± 2.24
74.84 ± 2.19
pc1
92.71 ± 1.21
92.94 ± 0.66
92.65 ± 1.29
92.39 ± 0.38
92.22 ± 1.06
cmc
51.22 ± 2.60
50.51 ± 2.75
52.32 ± 2.96
52.60 ± 2.30
51.46 ± 2.50
wine
80.53 ± 0.39
78.42 ± 1.46
79.87 ± 1.52
78.83 ± 0.32
79.13 ± 0.89
Table 2: F1-score results of traditional AutoFE methods under the same feature budget. Bold and underlined indicate the best and second-best results, respectively.
Figure 3: The topology of generated features on the jungle_chess and compass datasets: (a) and (b) are the jungle_chess and compass datasets, respectively.
Figure 4: Ablation results of SIGMA. “ w/o predefine operations ” represents that EXIT is still used, but operations are not provided. “ w/o track operations ” represents that EXIT is still used and operations are provided, but LLMs are not forbidden from using the top-2 frequent operations. “ w/o EXIT ” represents that operation predefinition and tracking are kept the same as SIGMA, but without the EXIT. (a) depicts the acceptance rate of generated features over all experiments. (b) and (c) show the code error rate and feature duplication rate, respectively.
Dataset Size
Mean F1-Score
Mean Rank
vs Qwen3-4B
Qwen3 4B
Qwen3 Coder
Llama 70B
Qwen3 4B
Qwen3 Coder
Llama 70B
Qwen3 Coder
Llama 70B
Small ( ≤ 2000)
71.95%
71.34%
71.27%
1.60
2.2
2.2
-0.61%
-0.67%
Large ( > 2000)
83.37%
83.51%
83.30%
2.36
1.73
1.91
+0.14%
-0.07%
All
79.80%
79.71%
79.54%
2.12
1.88
2.00
-0.10%
-0.26%
Table 3: SIGMA performance with different LLMs by dataset size. LLama-70B represents Llama-3.1-70B. The best results are highlighted in bold .
Appendix figures & tables3 assets
Supplementary material from the paper’s appendix.
Appendix
Metric
Baseline (w.o. AutoFE)
CAAFE
OCTree
SIGMA
Average ACC
79.22
79.91 ± 0.18
79.10 ± 0.14
79.98 ± 0.24
Avg ACC Rank
2.50
2.38
3.12
2.00
Average AUC
87.01
87.43 ± 0.12
86.77 ± 0.10
87.41 ± 0.04
Avg AUC Rank
2.44
2.12
3.38
2.00
Appendix
Table 4: Overall accuracy (ACC) and AUC-ROC comparison of LLM-based AutoFE.
Dataset
Baseline (w.o. AutoFE)
AutoFeat
DFS
OpenFE
SIGMA
Average ACC
79.22
79.36
79.39
80.18
79.98 ± 0.24
Avg ACC Rank
3.38
3.06
3.44
2.38
2.69
Average AUC
87.01
87.30
87.09
87.60
87.41 ± 0.04
Avg AUC Rank
3.38
2.81
3.44
2.50
2.81
Appendix
Table 5: Overall accuracy (ACC) and AUC-ROC comparison of traditional methods under a feature budget of 20.
Dataset
C
F
N
Llama3.1-70B
Qwen3-4B
Qwen3-Coder-Next
eucalyptus
5
19
736
65.15 ± 2.35
66.39 ± 2.29
65.43 ± 2.45
diabetes
2
8
768
73.50 ± 3.72
74.82 ± 3.16
73.97 ± 3.92
credit-g
2
20
1,000
73.33 ± 2.70
74.84 ± 2.19
73.84 ± 2.00
pc1
2
21
1,109
92.70 ± 1.04
92.22 ± 1.06
92.32 ± 1.06
cmc
3
9
1,473
51.66 ± 2.65
51.46 ± 2.50
51.14 ± 3.16
wine
2
11
2,554
79.76 ± 1.36
79.13 ± 0.89
79.31 ± 1.24
Appendix
Table 6: Impact of LLMs on each dataset of F1-score. C denotes the number of classes, F denotes the number of features, and N denotes the number of samples.
Automatic feature engineering (AutoFE) for tabular data requires discovering informative transformations from a large program space. Existing approaches suffer from three limitations: classical methods rely on fixed operator libraries with limited expressivity, LLM-based methods generate proposals from static prompts without retaining search experience, and evolutionary methods use fixed migration policies that ignore task-specific cross-family transfer utility. We introduce TOPOFE, a framework that formulates AutoFE as graph-structured multi-island evolutionary program search. The transformation space is partitioned into semantically coherent families, each explored by an island through LLM-guided mutation and crossover. Each island maintains a Prompt Adaptation Memory that accumulates accept/reject feedback to steer proposals toward productive regions without parameter updates. To coordinate global exploration, TOPOFE dynamically learns a directed topology graph whose edge weights encode transfer utility between transformation families. Cross-island transfer is triggered by adaptive saturation detection and performed through LLM-mediated hybrid synthesis, enabling discovery of compositional feature programs that cannot emerge from isolated local search. Experiments on 29 tabular datasets show that TOPOFE consistently outperforms most state-of-the-art AutoFE methods on classification and regression tasks. Beyond predictive performance, TOPOFE produces feature sets with lower redundancy and higher representational coverage, while the learned topology graph acquires meaningful task-specific transfer structure correlated with downstream gains. The discovered feature programs transfer reliably across diverse predictors and LLM backbones, demonstrating that improvements arise from TOPOFE's structured search and adaptive coordination rather than backbone-specific generation capability.
Sha Li, Naren Ramakrishnan
Department of Computer Science Virginia Tech Blacksburg, VA 24060 · Department of Computer Science Virginia Tech Alexandria, VA 22305
Tabular data, as a core data format in machine learning, often lacks the discriminative power needed for high-performance modeling due to insufficient feature informativeness. Automated Feature Engineering (AutoFE) overcomes this by automating feature generation and selection, ensuring both model performance and operational efficiency. However, traditional AutoFE often yield features with poor interpretability because they rely on blind mathematical transformations, while large language models (LLM)-based AutoFE faces challenges in requiring costly multi-round iterations to generate high-utility features to effectively enhance model performance, compounded by inherent risks of bias and hallucination. In this paper, we combine symbolic regression with LLMs for feature engineering (SymboLLM-FE) to solve these challenges. We extract mathematically expressive formulas strongly correlated with the target via symbolic regression, which can enhance model performance, then refine them by LLMs with rich prior knowledge to ensure interpretability. Empirical results on six real-world datasets and four Kaggle competitions demonstrate that SymboLLM-FE outperforms existing AutoFE. SymboLLM-FE also addresses the dual challenges of poor interpretability and numerous iterations by employing a statistical prior-grounded LLM refinement mechanism and single-digit LLM calls.
Zi-Jian Cheng, Zi-Yi Jia, Zhi Zhou +2
National Key Laboratory for Novel Software Technology, Nanjing University · School of Intelligence Science and Technology, Nanjing University · School of Artificial Intelligence, Nanjing University
Automated feature generation extracts informative features from raw tabular data without manual intervention and is crucial for accurate, generalizable machine learning. Traditional methods rely on predefined operator libraries and cannot leverage task semantics, limiting their ability to produce diverse, high-value features for complex tasks. Recent Large Language Model (LLM)-based approaches introduce richer semantic signals, but still suffer from a restricted feature space due to fixed generation patterns and from the absence of feedback from the learning objective. To address these challenges, we propose a Memory-Augmented LLM-based Multi-Agent System (MALMAS) for automated feature generation. MALMAS decomposes the generation process into agents with distinct responsibilities, and a Router Agent activates an appropriate subset of agents per iteration, further broadening exploration of the feature space. We further integrate a memory module comprising procedural memory, feedback memory, and conceptual memory, enabling iterative refinement that adaptively guides subsequent feature generation and improves feature quality and diversity. Extensive experiments on multiple public datasets against state-of-the-art baselines demonstrate the effectiveness of our approach. The code is available at https://github.com/fxdong24/MALMAS
Fengxian Dong, Zhi Zheng, Xiao Han +5
University of Science and Technology of China · Zhejiang University of Technology · Meituan