Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

pTNAS: Progressive Neural Architecture Search for Tabular Data

About

Recent advances have shifted the paradigm of tabular learning toward tabular foundation models, yet their accuracy relies on a heavy inference cost that scales poorly with context size. Deep neural networks remain a highly competitive and more efficient modeling paradigm when equipped with well-designed architectures; however, identifying such architectures in a data-adaptive and budget-aware manner remains challenging. We propose pTNAS, the first progressive neural architecture search (NAS) approach tailored for tabular data, which enables fast identification of a viable architecture and continuously improves its search performance as more budget becomes available. pTNAS adopts a filter-and-refine optimization strategy that combines efficient training-free and effective training-based architecture evaluation. In the filtering phase, we introduce pTProxy, a novel zero-cost proxy specifically designed for tabular networks that jointly captures architectural trainability and expressivity, enabling fast filtering of large architecture search spaces. In the refinement phase, pTNAS employs a fixed-budget scheduling algorithm to accurately identify the best-performing architecture from a small set of promising candidates. We further propose a budget-aware coordinator to optimize budget allocation holistically. Experiments show that pTNAS reduces the time to reach the globally best architecture by up to 82.75 X compared with other NAS approaches, achieves the best average predictive rank, and improves end-to-end efficiency by up to 4.78 X compared with TabPFN.

Naili Xing, Shaofeng Cai, Lingze Zeng, Jiaqi Zhu, Peng Lu, Jian Pei, Beng Chin Ooi• 2024

Related benchmarks

TaskDatasetResultRank
Classification4 Tabular Classification Datasets (Event user-repeat, Beer user-active, Trial study-out, Avito user-click)
Average Rank1
16
Tabular Data Analysis8 Tabular Datasets Overall
Average Rank1.38
16
Regression4 Tabular Regression Datasets (Event user-attend, Beer beer-pos, Trial site-success, HM item-sales)
Avg Rank1.75
15
Neural Architecture Search Proxy Evaluation8 Tabular Datasets Aggregated
Mean SRCC0.82
10
Regression8 multi-table datasets aggregated (test)
Fitting Time (s)33.12
7
Classification8 multi-table datasets aggregated (test)
Fitting Time (s)18.81
7
Showing 6 of 6 rows

Other info

Follow for update