CoFEH: LLM-driven Feature Engineering Empowered by Collaborative Bayesian Hyperparameter Optimization

ACM SIGKDD Conference on Knowledge Discovery and Data Mining (KDD), 2026. Oral.

Beicheng Xu*, Keyao Ding*, Wei Liu, Yupeng Lu, Bin Cui

* Equal contribution.

Overview

CoFEH interleaves LLM-driven feature engineering and Bayesian hyperparameter optimization. A tree-of-thought feature engineering optimizer, a Bayesian optimization module, and a dynamic optimizer selector cooperate through mutual conditioning to improve end-to-end AutoML.

All publications