Saved in:
Bibliographic Details
Main Authors: Gao, Wanfu, He, Zebin, Gao, Jun
Format: Preprint
Published: 2025
Subjects:
Online Access:https://arxiv.org/abs/2512.15082
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866915681226719232
author Gao, Wanfu
He, Zebin
Gao, Jun
author_facet Gao, Wanfu
He, Zebin
Gao, Jun
contents Existing feature engineering methods based on large language models (LLMs) have not yet been applied to multi-label learning tasks. They lack the ability to model complex label dependencies and are not specifically adapted to the characteristics of multi-label tasks. To address the above issues, we propose Feature Engineering Automation for Multi-Label Learning (FEAML), an automated feature engineering method for multi-label classification which leverages the code generation capabilities of LLMs. By utilizing metadata and label co-occurrence matrices, LLMs are guided to understand the relationships between data features and task objectives, based on which high-quality features are generated. The newly generated features are evaluated in terms of model accuracy to assess their effectiveness, while Pearson correlation coefficients are used to detect redundancy. FEAML further incorporates the evaluation results as feedback to drive LLMs to continuously optimize code generation in subsequent iterations. By integrating LLMs with a feedback mechanism, FEAML realizes an efficient, interpretable and self-improving feature engineering paradigm. Empirical results on various multi-label datasets demonstrate that our FEAML outperforms other feature engineering methods.
format Preprint
id arxiv_https___arxiv_org_abs_2512_15082
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle The Semantic Architect: How FEAML Bridges Structured Data and LLMs for Multi-Label Tasks
Gao, Wanfu
He, Zebin
Gao, Jun
Machine Learning
Existing feature engineering methods based on large language models (LLMs) have not yet been applied to multi-label learning tasks. They lack the ability to model complex label dependencies and are not specifically adapted to the characteristics of multi-label tasks. To address the above issues, we propose Feature Engineering Automation for Multi-Label Learning (FEAML), an automated feature engineering method for multi-label classification which leverages the code generation capabilities of LLMs. By utilizing metadata and label co-occurrence matrices, LLMs are guided to understand the relationships between data features and task objectives, based on which high-quality features are generated. The newly generated features are evaluated in terms of model accuracy to assess their effectiveness, while Pearson correlation coefficients are used to detect redundancy. FEAML further incorporates the evaluation results as feedback to drive LLMs to continuously optimize code generation in subsequent iterations. By integrating LLMs with a feedback mechanism, FEAML realizes an efficient, interpretable and self-improving feature engineering paradigm. Empirical results on various multi-label datasets demonstrate that our FEAML outperforms other feature engineering methods.
title The Semantic Architect: How FEAML Bridges Structured Data and LLMs for Multi-Label Tasks
topic Machine Learning
url https://arxiv.org/abs/2512.15082