mlsanityData to 13 Oct 2026

PACE: Plug-and-Play Contextual Embedding for Feature Screening with Pretrained Tabular Foundation Models

arXiv 2609.23574 ↗First posted 20 Sep 2026stat.ML cs.LG stat.APcited by 0

Not on the map. Why?

Abstract (arXiv)

In high-dimensional tabular learning, feature screening provides a lightweight, model-agnostic way to remove irrelevant features before model fitting. However, scoring raw values directly can miss nonlinear or distributional structure. We introduce PACE (Plug-and-Play Contextual Embedding), which inserts a frozen tabular foundation model (TFM) column encoder before an existing feature-scoring rule, expanding each feature into a higher-dimensional contextual representation. Across controlled studies, PACE improves raw-space screening of complex nonlinear dependence with only modest additional encoding cost. These gains translate to downstream prediction on TALENT datasets: PACE-DC improves binary AUC by 0.077 and multiclass macro-AUC by 0.064, with a median normalized RMSE improvement of 0.063 across ten learners. Matched random-weight and random-feature controls show that PACE gains from pretrained structure beyond generic dimensional expansion. PACE further achieves favorable performance--time trade-offs against task-fitted selectors and attribution-based methods, positioning pretrained column geometry as a reusable upstream primitive for tabular learning.

Key facts. No citing papers in the map yet (data through 13 Oct 2026). Builds on 30 of its 60 references that are in the map. First posted on arXiv 20 Sep 2026.

Builds on

30 of 60 references.

References · 30

  1. 2017A Unified Approach to Interpreting Model Predictionscited by 6,515
  2. 2016XGBoost: A Scalable Tree Boosting Systemcited by 5,143
  3. 2016Distribution-Free Predictive Inference For Regressioncited by 660
  4. 2025TabICL: A Tabular Foundation Model for In-Context Learning on Large Datacited by 272
  5. 2026TabICLv2: A better, faster, scalable, and open tabular foundation modelcited by 124
  6. 2025TabPFN-2.5: Advancing the State of the Art in Tabular Foundation Modelscited by 121
  7. 2024TabM: Advancing Tabular Deep Learning with Parameter-Efficient Ensemblingcited by 109
  8. 2024TabDPT: Scaling Tabular Foundation Models on Real Datacited by 101
  9. 2024Better by Default: Strong Pre-Tuned MLPs and Boosted Trees on Tabular Datacited by 88
  10. 2019LassoNet: A Neural Network with Feature Sparsitycited by 81
  11. 2024TALENT: A Tabular Analytics and Learning Toolboxcited by 47
  12. 2025A Closer Look at TabPFN v2: Understanding Its Strengths and Extending Its Capabilitiescited by 39
18 more
  1. 2023A Performance-Driven Benchmark for Feature Selection in Tabular Deep Learningcited by 24
  2. 2023Scaling TabPFN: Sketching and Feature Selection for Tabular Prior-Data Fitted Networkscited by 18
  3. 2025xRFM: Accurate, scalable, and interpretable feature learning models for tabular datacited by 15
  4. 2026Beyond IID: How General Are Tabular Foundation Models, Really?cited by 15
  5. 2025TabPFN-Wide: Continued Pre-Training for Extreme Feature Countscited by 14
  6. 2019Model-free Feature Screening and FDR Control with Knockoff Featurescited by 9
  7. 2014The fused Kolmogorov filter: A nonparametric model-free screening methodcited by 7
  8. 2022Deep Feature Screening: Feature Selection for Ultra High-Dimensional Data via Deep Neural Networkscited by 6
  9. 2026Towards Universal Tabular Embeddings: A Benchmark Across Data Taskscited by 4
  10. 2024Air-HOLP: Adaptive Regularized Feature Screening for High Dimensional Correlated Datacited by 3
  11. 2025Comparing Task-Agnostic Embedding Models for Tabular Datacited by 3
  12. 2026ExplainerPFN: Towards tabular foundation models for model-free zero-shot feature importance estimationscited by 3
  13. 2026Mitra-v2 Technical Reportcited by 3
  14. 2025Beyond Black-Box Predictions: Identifying Marginal Feature Effects in Tabular Transformer Networkscited by 2
  15. 2026GOTabPFN: From Feature Ordering to Compact Tokenization for Tabular Foundation Models on High-Dimensional Datacited by 2
  16. 2025SAND: One-Shot Feature Selection with Additive Noise Distortioncited by 1
  17. 2025Label-shift robust federated feature screening for high-dimensional classificationcited by 1
  18. 2025Knockoffs Inference under Privacy Constraintscited by 1

Built on by

Not cited yet.