Skip to content
AI Atlas
PaperActive

Hierarchical and Permutation-Invariant Feature Transformation Learning via Policy-Guided Embedding Search

arxiv.org/abs/2609.10225

Updated 33 min ago · first seen 11 Sept 2026

paper_01M294GNA74T6M2DZVXCQ883D0

Published
11 Sept 2026
T1 · 33 min ago
arXiv
2609.10225
T1 · 33 min ago
Category
cs.LG
T1 · 33 min ago

Abstract

Feature transformation improves predictive performance on tabular data by constructing informative abstractions from raw features. Recent generative approaches encode transformation knowledge into continuous embedding spaces for efficient exploration of candidate strategies, but face three key limitations: (1) overlooking hierarchical relationships between low-level features, operations, and high-level abstractions; (2) enforcing order-sensitive embeddings on inherently permutation-invariant transformation sequences, thereby introducing systematic bias; and (3) relying on gradient-based search, which is ill-suited to non-convex transformation spaces. We propose a framework with two complementary components. First, a permutation-invariant hierarchical module captures interactions across features, operations, and abstraction levels, with a self-attention pooling mechanism that maps semantically equivalent structures to consistent embeddings aligned with downstream performance. Second, a policy-guided multi-objective reinforcement learning strategy initializes the search from empirically strong seeds and jointly optimizes predictive accuracy and transformation efficiency. Extensive experiments on diverse tabular benchmarks demonstrate the effectiveness and robustness of our framework against strong baselines. Our code and data are publicly available at: https://github.com/RayLiu1103/PHER.

Authors 8

Rui Liu, Tao Zhe, Yanyong Huang, Sankha Narayan Guria, Xiao Luo, Wei Fan, Yanjie Fu, Dongjie Wang

Specification

Official page

Source:arXiv (Atom API + RSS)T1observed 33 min agohigh

Arxiv announce type
cross

Source:arXiv (Atom API + RSS)T1observed 33 min agohigh

arXiv id
2609.10225

Source:arXiv (Atom API + RSS)T1observed 33 min agohigh

Categories
cs.LG, cs.AI

Source:arXiv (Atom API + RSS)T1observed 33 min agohigh

PDF

Source:arXiv (Atom API + RSS)T1observed 33 min agohigh

Primary category
cs.LG

Source:arXiv (Atom API + RSS)T1observed 33 min agohigh

Published
11 Sept 2026

Source:arXiv (Atom API + RSS)T1observed 33 min agohigh

Each value shows its source, tier and observation time. Conflicting claims are kept side by side and flagged — never averaged. How AI Atlas records facts →

Provenance

Attributed facts

9

Source tiers

T19

Freshest observation

33 min ago

Conflicts

None