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.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Classification | 4 Tabular Classification Datasets (Event user-repeat, Beer user-active, Trial study-out, Avito user-click) | Average Rank1 | 16 | |
| Tabular Data Analysis | 8 Tabular Datasets Overall | Average Rank1.38 | 16 | |
| Regression | 4 Tabular Regression Datasets (Event user-attend, Beer beer-pos, Trial site-success, HM item-sales) | Avg Rank1.75 | 15 | |
| Neural Architecture Search Proxy Evaluation | 8 Tabular Datasets Aggregated | Mean SRCC0.82 | 10 | |
| Regression | 8 multi-table datasets aggregated (test) | Fitting Time (s)33.12 | 7 | |
| Classification | 8 multi-table datasets aggregated (test) | Fitting Time (s)18.81 | 7 |