SymboLLM-FE: LLM-Accelerated Symbolic Regression for Automated Feature Engineering on Tabular Data
Analyze & explore related research in Conceptio
Connect this record directly into your research workflow or AI agent. Explore citations across open repositories, query full text, and export structured proof bundles.
SymboLLM-FE: LLM-Accelerated Symbolic Regression for Automated Feature Engineering on Tabular Data †
Zi-Jian Cheng1,2 , Zi-Yi Jia1,2 , Zhi Zhou1,3 , Yu-Feng Li1,3 , Lan-Zhe Guo1,2 1 National Key Laboratory for Novel Software Technology, Nanjing University 2 School of Intelligence Science and Technology, Nanjing University 3 School of Artificial Intelligence, Nanjing University jiazy@smail.nju.edu.cn {chengzj,zhouz,liyf,guolz}@lamda.nju.edu.cn
arXiv:2608.28408v1 [cs.LG] 28 Aug 2026
Abstract 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 (SymboLLMFE) 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.
1
Introduction
Tabular data (Altman and Krzywinski, 2017), a highly structured data format, organizes information in rows and columns, where each row represents an independent sample or instance, and each column corresponds to a specific feature or attribute (Sahakyan et al., 2021). Tabular data is extensively utilized in real-world applications. For †
Corresponding author.
instance, tabular data supports financial tasks such as credit scoring (West, 2000) and stock market prediction (Zhu et al., 2021). Tabular data also facilitates medical applications, including disease diagnosis (Yıldız and Kalayci, 2024) and drug development (Meijerink et al., 2020). However, in real-world applications, tabular data often suffers from insufficient feature informativeness and implicit high-order interactions. These inherent limitations create a gap that hinders models from effectively comprehending the underlying data structures, thereby leading to suboptimal predictive performance (Sayed et al., 2025; Cheng et al., 2025b; Gorishniy et al., 2021). Consequently, prior to predicting, feature engineering is typically applied to enhance the relationship between input features and the target through operations such as feature generation and selection (Ravishankar and Battineni, 2025). The processed tabular data is then fed into downstream models to improve prediction performance. Previously, feature engineering relied heavily on manual construction of high-quality features, a process requiring extensive iterative efforts that significantly increased labour costs (Susan and Tuteja, 2025). Hence, AutoFE has emerged as a critical research focus in tabular machine learning tasks. AutoFE aims to automatically and algorithmically discover and construct effective features to reduce manual effort and enhance machine learning model performance (Khurana et al., 2016). Traditional AutoFE include hybrid optimization with beam search (Horn et al., 2019), dynamic pruning (Zhang et al., 2023b) and feature selection (Li et al., 2017; Cheng et al., 2025a). They typically rely on predefined transformation rules, exhaustive search or optimization-based strategies to generate candidate features. While capable of improving model performance, traditional AutoFE still suffer from poor interpretability. The generated features are often com-
Traditional AutoFE
LLM-based AutoFE
SymboLLM-FE
Raw Columns:
Raw Columns & Task Background
Raw Columns & Task Background
LLM + Downstream Predictor
Symbolic Regression + LLM + Downstream Predictor
[Age,Income,Fever,...]
Generated Features Feature_new_1 = 푥�� + 2푥�푥� + 푥�� Feature_new_2 = 푚푖푛( 푥�, 푥� ) ln � Feature_new_3 = � �� ……
Generated Features Feature Name: Risk_Indicator IF (Age > 60 AND Fever=’Yes’) THEN ‘High’ ELSE ‘Low’
Generated Features Feature Name: Infection_Exposure_Risk (calculate a weighted symptom score) Infection_Exposure_Risk = (Fever=’Yes’) * 0.4 + (Fatigue=’Yes’) * 0.3 + Smooth * 0.2
Method Impact
Method Impact
Method Impact
Model Score Interpretability
Model Score Interpretability
Model Score Interpretability
Figure 1: Comparative analysis of feature generation mechanisms and interpretability paradigms across AutoFE. Traditional AutoFE rely on blind operator stacks, yielding high performance but poor interpretability. LLM-based AutoFE offer semantic clarity but may suffer from performance instability. Our proposed SymboLLM-FE synergizes symbolic regression with LLMs to achieve both high predictive accuracy and human-interpretable transparency.
plex mathematical compositions devoid of explicit semantic meaning, creating a disconnect from human-understandable domain-specific logic. This lack is particularly detrimental in scenarios like factor mining (Wang et al., 2025, 2026), where interpretability is not merely optional but essential for validating the economic rationale behind signals, ensuring regulatory compliance, and facilitating effective error diagnosis. Without clear reasoning logic, it becomes difficult to distinguish between robust predictive patterns and spurious correlations, thereby undermining trust in model decisions. In recent years, the advancement of LLMs has provided new possibilities for feature engineering through their capabilities in semantic understanding (Hollmann et al., 2023b), logical reasoning (Zou et al., 2026), and knowledge guidance (Nam et al., 2024). Hence, LLM-based AutoFE have garnered significant attention due to its ability to intelligently generate semantically meaningful features. Unlike traditional AutoFE that relies on predefined rules or exhaustive search, LLMbased AutoFE leverages its rich prior knowledge and contextual comprehension to automatically identify potential relationships among features in raw tabular data and construct more interpretable new features through mathematical operations and logical rules. However, LLM-based AutoFE still faces two critical challenges. First, generating performanceenhancing features often necessitates extensive iter-
ative interactions with downstream models for validation and refinement, resulting in prohibitive computational overhead and limited scalability (Abhyankar et al., 2025). Second, inherent susceptibility of LLMs to hallucinations and implicit biases can lead to the generation of semantically invalid or unfair features, which significantly compromise the reliability and predictive integrity of downstream models (Cheng et al., 2026; Han et al., 2024). It is evident that traditional AutoFE is limited by uninterpretable features, while LLM-based AutoFE struggles with the high iteration costs, hallucinations and implicit bias. Hence, it is urgent to research AutoFE for tabular machine learning tasks to develop high-efficiency feature engineering frameworks that yield interpretable features with significant performance gains to overcome the limitations of current paradigms. To this end, we propose SymboLLM-FE, a novel framework that combines symbolic regression and LLMs to generate mathematical and interpretable features which can enhance model performance. SymboLLM-FE operates through a two-stage collaborative pipeline. First, it employs a computationally efficient expanding-sliding window strategy guided by Spearman correlation analysis to reduce the search space from exponential O(2n ) to polynomial O(n2 ), enabling symbolic regression to generate explicit, mathematically grounded candidate formulas. Second, it leverages LLMs to inject domain-specific priors into these opaque mathematical formulas, transforming them into semantically meaningful, executable code via chain-of-thought reasoning. This iterative refinement loop not only validates feature utility through downstream predictor performance but also ensures alignment with business logic, thereby achieving a robust balance between high-performance feature discovery and human-interpretable transparency. Our contributions are summarized as follows: • We conduct empirical analysis and identify that traditional AutoFE suffer from insufficient interpretability, while LLM-based AutoFE face extensive iterative experimentation, hallucination and implicit bias. • We propose SymboLLM-FE to address current AutoFE challenges through leveraging symbolic regression to mine performanceenhanced features, which are then refined by LLMs to enhance interpretability.
AutoFE
Generated Features
Score
API Call
AutoFeat OpenFE CAAFE OcTree FEBP LLM-FE LLM-RANK SymboLLM-FE
2 1436 30 50 200 30 − 70
79.75 79.53 79.94 79.52 79.36 79.59 78.86 80.02
− − 10 50 29 35 7 4
Table 1: Comparative analysis of performance efficiency and computational cost across AutoFE. AutoFeat and OpenFE are traditional AutoFE which don’t rely on LLMs. LLM-RANK does not generate features because it’s a feature selector.
• Comprehensive evaluations on six real-world datasets and four Kaggle competitions demonstrate the superior performance and generalizability of SymboLLM-FE compared with current AutoFE methods. • SymboLLM-FE addresses the dual challenges of poor interpretability and iterative experimentation by employing a statistical priorgrounded LLM refinement mechanism and single-digit LLM calls.
2
Related Work
2.1
Tabular Machine Learning on LLMs.
The advent of LLMs possessing substantial domainspecific prior knowledge has prompted the exploration of their utility in addressing tabular machine learning tasks (Jia et al., 2026). For example, LIFT (Dinh et al., 2022) fine-tunes GPT3 (Floridi and Chiriatti, 2020) on training data, showing that general-purpose LLMs can improve performance though not surpassing tree-based models. TabLLM (Hegselmann et al., 2023) employs T0 (Sanh et al., 2021) for fine-tuning, effectively leveraging pre-trained knowledge with minimal labelled data. UniPredict (Wang et al., 2023) trains GPT-2 (Ethayarajh, 2019) across 169 tabular datasets, achieving competitive results without dataset-specific tuning. However, these LLMbased approaches demonstrate superior performance primarily in zero-shot or few-shot experimental settings (Hegselmann et al., 2023), while still lagging behind tree-based and deep learning models in terms of both computational efficiency and prediction accuracy in real-world scenarios (Cheng et al., 2025b). This suggests that directly employing LLMs as predictors is subopti-
mal, prompting researchers to explore leveraging LLMs as feature engineering methods to enhance data representation, thereby improving the performance of downstream predictors like tree-based or deep learning models (Hollmann et al., 2023b). 2.2
Traditional and LLM-based AutoFE.
Traditional AutoFE primarily rely on heuristic search or reinforcement learning, evolving from exhaustive frameworks (Katz et al., 2016) and DFS (Kanter and Veeramachaneni, 2015) to efficient, utility-driven approaches (Zhang et al., 2023b) and causally-guided models (Malarkkan et al., 2026). Despite these advancements, they often struggle with combinatorial explosion in complex search spaces and lack semantic alignment with domain-specific business logic. While recent advances in LLMs have driven progress in LLMbased AutoFE, which bifurcates into two key directions: feature generation and feature selection. For feature generation, methods include contextaware frameworks (Hollmann et al., 2023b), decision trees for rule discovery (Nam et al., 2024), differentiable feature mapping (Han et al., 2024), structured generation via reverse Polish notation (Zou et al., 2026), and evolutionary-knowledge distillation hybrids (Abhyankar et al., 2025). Feature selection techniques employ log-probability thresholds (Choi et al., 2022) and ranking paradigms (Jeong et al., 2024). However, these approaches face challenges including the uselessness of generated features, hallucinations and biases. 2.3
Automated Feature Engineering on Symbolic Regression.
Symbolic Regression (Yang et al., 2025) is a machine learning method that discovers optimal mathematical expressions to uncover latent data relationships. Its capacity to derive interpretable analytical forms has made it valuable for feature engineering (Shmuel et al., 2024), improving both model transparency and predictive accuracy while reducing manual effort. Recent advances include evolutionary computation integration for physical phenomenon extraction (Yu et al., 2026), modular multi-tree genetic programming for highdimensional feature reuse (San et al., 2021), and Spearman-based feature selection to enhance generalization (Kim et al., 2020). However, symbolic regression still struggles with expression complexity in high-dimensional spaces. While existing work has explored LLM-SR interactions for equation
generation (e.g., natural language processing (Shao et al., 2026) and code synthesis (Shojaee et al., 2025)), they have not extended its work to incorporate the equations generated by symbolic regression into LLMs for feature engineering.
3
Problem Formulation and Analysis
3.1
Problem Formulation
Formally, the goal of tabular prediction tasks is to train a machine learning model f : X → Y, where X is the input space and Y is the output space. We define the set of features in X as C. For the n-dimensional X , the ordinary feature set C = {c1 , c2 , ..., cn }. Feature engineering is aimed to expand the dimension of X through generating m new features, namely, the new feature set Cnew = {cn+1 , cn+2 , ..., cn+m } based on C. After feature engineering, the dimension of X increases from n to n + m and the performance P of f can be improved on the new feature set S. Our aim is to get the maximum of performance f , by changing Cnew generated by AutoFE g, i.e., max Pf (S) = max Pf (C ⊕ Cnew ) = max Pf (C ⊕ g(C)) 3.2
(1)
Analysis
The comparative analysis of various AutoFE in Figure 1 and Table 1 reveals a distinct dichotomy between traditional and LLM-based AutoFE, highlighting critical bottlenecks in both efficiency and feature quality. As illustrated in Figure 1, traditional AutoFE is strictly limited by uninterpretable features. Because it relies on a blind operator stack to mathematically combine raw columns without semantic understanding, it generates an excessively large feature set. These newly generated features lack semantic grounding and physical meaning, failing to provide actionable insights for future feature acquisition and offering poor generalization to downstream tasks due to a severe compromise in the interpretability of generated features. While LLM-based AutoFE ensures semantic interpretability by leveraging task background knowledge, it struggles with high iteration costs as shown in Table 1. Relying solely on LLMs for feature generation frequently yields features that, though semantically coherent and domain-relevant, lack
robust discriminative power. Hence, methods like OcTree require approximately 50 rounds to converge, indicating that features generated by LLMbased AutoFE necessitate multiple rounds of trialand-error validation and continuous refinement to enhance downstream model performance. This intrinsic uncertainty forces extensive iterative interactions with downstream predictors, leading to a prohibitive surge in both iteration rounds and computational time costs. Furthermore, these advanced LLM-based AutoFE remain inherently susceptible to severe hallucinations and implicit biases, which can ultimately cause a drop in model performance. These findings distinctly highlight the core limitations of existing traditional and LLM-based AutoFE, emphasizing the urgent need for a more optimized AutoFE framework that can bypass the high iteration overhead of LLMs and the uninterpretable nature of traditional mathematical combinations to generate features that are simultaneously interpretable and predictively effective.
4
Method
To address the limitations of the current AutoFE, we shift our focus to synergize symbolic regression and LLMs. Symbolic regression excels in efficiently discovering explicit mathematical formulas that are not only computationally lightweight but also address the need for performance-enhancing features. Complementarily, LLMs leverage their knowledge to refine and optimize these candidate symbolic formulas. By injecting domainspecific priors, LLMs transform opaque mathematical formulas, simultaneously enhancing their interpretability through natural language explanations and ensuring their alignment with business logic. This hybrid paradigm, namely SymboLLM-FE, effectively bridges the gap between efficacy and interpretability. Figure 2 demonstrates the detailed pipeline of SymboLLM-FE. Dataset Split. SymboLLM-FE strictly adheres to a fold-wise training protocol within a crossvalidation framework. The training set is exclusively utilized for Formula Construction by Symbolic Regression, supporting the entire pipeline from Spearman correlation-based feature reordering and expanding-sliding window subset generation to the fitting of symbolic models. The validation set serves as the feedback environment for Feature Generation via LLMs. It remains isolated from the former process but is actively used to
Predictor
Formulas
Task Description
Test Score
Validation Score
Features
Best Score
Symbolic Regression
S Re ymb gr oli ess c ion
201-dim
200-dim
Background ( Optional )
400-dim
Sliding 201-dim Sliding 200-dim 400-dim
200-dim Expanding
Feature Importance ( DESC )
Feature Generation via LLMs
lic bo on m ssi Sy gre Re
400-dim
Formula Construction by Symbolic Regression
Predict
Feature Merge
Figure 2: Overview of SymboLLM-FE, including Formula Construction by Symbolic Regression and Feature Generation via LLMs and downstream predictors.
compute validation scores that guide the LLM’s iterative code synthesis, rule mining, and feature optimization loop. Finally, the test set is reserved solely for the final downstream evaluation of the merged feature set to assess model performance. To strictly prevent data leakage, the validation set is employed exclusively for hyperparameter tuning. The downstream model architecture and parameters remained fixed and independent of the test evaluation process, ensuring that no information from the validation phase inadvertently influenced the final performance assessment. 4.1
Formula Construction by Symbolic Regression
We firstly construct numerous input feature subsets by using Spearman correlation (De Winter et al., 2016) and an expanding-sliding window approach. Subsequently, symbolic regression model is applied to each input feature subset separately to derive interpretable mathematical formulas that capture significant relationships with the target. Feature Set Sampling. Given C with n features, the total number of its non-empty feature subsets is 2n − 1. Applying symbolic regression to every subset would lead to exponential computational complexity (O(2n )). To optimize computational efficiency while preserving feature representational capacity, we propose a feature combination strategy based on Spearman correlation coefficient analysis and an expanding-sliding window approach. Specifically, we first evaluate the importance of
C using Spearman correlation coefficient analysis, followed by a monotonic ascending sorting based on the correlation coefficients. This process yields an ordered feature sequence C ′ = c′1 , c′2 , ..., c′n , where each feature satisfies the relation ϕc′1 ≤ ϕc′2 ≤ ... ≤ ϕc′n , with ϕc′i denoting the Spearman correlation coefficient between feature c′i and the target. We choose Spearman correlation as the criterion for feature importance quantification because, compared to other metrics such as Pearson correlation, Spearman correlation more robustly quantifies nonlinear dependencies between features and the target, thereby providing a more accurate measure of marginal feature importance (Cheng et al., 2025b). Appendix D empirically validates the statistical correlations between this metric and other feature importance criteria. To maintain formula quality while effectively controlling the quantity of generated features, we propose a dynamic expanding-sliding window approach. The approach initializes with window size k = ⌊n/2⌋ and operates through two core phases: • Expanding Phase: k incrementally increases from ⌊n/2⌋ to n with unit step size. • Sliding Phase: For each fixed k, the window slides rightward with single-feature steps to generate different feature subsets. For instance, when k = ⌊n/2⌋, the generated subsets are {{c′1 , ..., c′n/2 }, {c′2 , ..., c′n/2+1 }, ...}. The total number S of candidate input feature subsets generated by the expanding-sliding window
approach is upper bounded by an arithmetic series summation, i.e., n X
⌈n/2⌉+1
(n − k + 1) =
k=⌊n/2⌋
X
m
m=1
(n + 2)(n + 4) , 8 = (n + 3)(n + 5) , 8
if n is even if n is odd (2)
Equation 2 shows that this approach reduces the sum of feature subsets from exponential O(2n ) to polynomial O(n2 ). Formula Generating. For each Si where i ∈ [1, |S|] generated by the expanding-sliding window approach, we train one symbolic regression model and generate a formula like add(mul(c1,c2),c3) from Si to the target as the new feature. Appendix A shows details about symbolic regression. Through systematic training and construction, we ultimately establish a formula-based feature repository R containing |S| triples: |S|
R = {(Si , Pf (Si ), τi )}i=1
(3)
where each triple consists of: 1) the feature subset Si ; 2) symbolic regression performance P of trained symbolic model f on Si ; 3) the formula τi from Si to the target. 4.2
Feature Generation via LLMs
We then implement a multi-stage refinement pipeline via LLMs and downstream predictors based on R to generate features. LLM-guided Code Generation. The input to LLMs is constructed by converting every formula τ in R into its corresponding natural language description, concatenated with structured dataset background information and codegeneration prompt. Subsequently, LLMs generate executable feature engineering code in the format of Python pandas through chain-of-thought reasoning (Wei et al., 2022). Downstream Predictor Performance Validation. The generated code is executed to produce an augmented feature set Cnew and then enrich the original dataset C, i.e., Cf inal = C ⊕ Cnew . Sequentially, we train a downstream predictor and get the model performance result under Cf inal .
Iterative Feature Refinement. The predictor performance, along with the utilized features and their corresponding generation codes, is then fed back into LLMs for iterative optimization and feature refinement. This feedback loop continues until the predictor achieves a relatively optimal performance level on the augmented feature set. It should be emphasised that SymboLLM-FE addresses concerns regarding feature selection through rigorous feature control and de-redundancy mechanisms. For feature budget management, we dynamically prune features that degrade validation performance via downstream predictor feedback, yielding a compact set of refined features that effectively avoids the curse of dimensionality compared to baseline methods. Redundancy elimination is achieved through prompt-independent constraints. Specifically, our template enforces explicit feature dropping with justification, while LLM serves as a semantic filter to identify mathematical equivalence between symbolic regression formulas and original features, thereby removing multicollinearity. To suppress noise and overfitting, we impose a parsimony penalty at the symbolic regression stage following Occam’s razor and strictly confine LLM operations to the statistically validated subspace, which eliminates hallucination and ensures the robustness and compactness of the final feature sets.
5
Experiments
5.1
Experimental Setup
Datasets. We select a variety of open-source and reliable datasets from OpenML and Kaggle. These datasets encompass three primary tasks of binary classification, multi-class classification and regression, and span a range of fields such as finance and healthcare. The detailed information about these datasets is provided in Appendix C.1. Downstream Baselines. To validate the generalization and adaptability of SymboLLM-FE, we evaluate it on various tree-based models and deep learning models. For tree-based models, we choose CatBoost (Prokhorenkova et al., 2018) and XGBoost (Chen and Guestrin, 2016). For deep learning models, we choose MLP (Gorishniy et al., 2021) and TabPFN (Hollmann et al., 2025). Appendix C.2 shows these baselines’ information and full hyperparameter grids. Comparison AutoFE. To demonstrate the effectiveness of SymboLLM-FE, we compare it against two traditional AutoFE (AutoFeat (Horn et al.,
FE Method
Credit-g ↑
Spaceship ↑
Cmc ↑
Academic ↑
Ailerons ↓
Tesla ↓
Baseline AutoFeat OpenFE CAAFE OcTree FEBP LLM-FE LLM-RANK SymboLLM-FE
77.03±0.47 77.83±1.43 76.50±3.34 78.00±0.71 76.50±0.82 77.50±0.41 76.67±0.62 77.50±0.82 77.00±1.63
80.79±1.27 80.76±1.17 80.30±1.00 80.99±1.00 80.79±1.15 80.85±1.29 80.22±0.96 80.70±1.15 81.27±1.31
57.85±0.89 57.85±0.32 57.85±2.35 57.78±1.28 57.85±1.52 57.93±0.48 57.29±2.20 57.74±0.85 57.97±0.73
77.33±0.67 76.80±0.46 76.42±0.61 77.29±0.56 77.33±0.70 77.36±0.35 76.42±0.61 77.25±0.19 77.89±0.23
5.10±0.48 5.04±0.45 5.26±0.49 5.03±0.46 5.09±0.47 5.08±0.48 5.05±0.46 5.10±0.48 5.02±0.46
2.39±0.09 2.38±0.00 2.18±0.06 2.66±0.09 2.43±0.10 2.41±0.08 2.39±0.09 5.87±0.19 2.16±0.06