arXiv Statistics ML

LLM knowledge for improved Out-of-Distribution Tabular Prediction - A Cross-Validated Regularisation Approach

The study investigates whether coefficient priors derived from large language models (LLMs) can enhance out‑of‑distribution (OOD) tabular prediction by serving as a shrinkage target in regularised regression. Across five datasets, LLM‑elicited priors improved OOD mean squared error on four datasets by up to 7.9% and yielded better probabilistic accuracy in classification tasks, though the gains varied and were sometimes comparable to in‑domain improvements. Cross‑validation was used to select priors, but the variability among candidates meant that the method mainly avoided the worst priors rather than consistently finding the best. "whyItMatters":"The findings suggest that LLM‑derived priors can provide a scalable alternative to expert elicitation for improving model robustness to distribution shifts, though their effectiveness depends on dataset size and prior variability."

arXiv Machine Learning
Oct 16

INCONVENIENT: A Benchmark to Assess LLM Generalization Capabilities from Convenience Samples

INCONVENIENT is a new benchmark that evaluates how machine‑learning models, especially large language models, generalize when trained on convenience samples that bias feature–target relationships. Unlike prior distribution‑shift tests that rely on natural domain splits, INCONVENIENT explicitly perturbs these relationships while keeping the deployment population unchanged, allowing controlled study of shift severity and direction. The benchmark covers 200 tabular classification and regression datasets and produces about 15,500 shifted training distributions, revealing significant performance degradation.

By Sara Rezaeimanesh, Chad Klochko, Kundan Thind, Mohammad Ghassemi
arXiv AI
Sep 18

Is It Still Worth Training a Classical Model in the Era of LLMs? A Crossover Benchmark on Tabular Data

The paper investigates whether training classical machine learning models remains worthwhile when large language models (LLMs) can label tabular data without training. By defining a labeled‑data crossover point (N*) where a trained classical model surpasses a frozen LLM’s flat error, the authors analyze 126 student evaluations of GPT models across 18 datasets and compare them to power‑law learning curves of six classical model families. Results show that in 86% of cases a classical model outperforms the LLM with no more labeled data than already available, and the crossover occurs at a median of about 6% of the training set, suggesting that collecting a few hundred labels and training a gradient‑boosted model is typically advantageous.

By Kaihua Ding
arXiv AI
5d ago

Dr. Post-Training: A Data Regularization Perspective on LLM Post-Training

The paper introduces Dr. Post-Training, a framework that treats general training data as a data‑induced regularizer to prevent overfitting to scarce target objectives, rather than merely a pool for selection. It constructs a feasible set of model update directions from general data and projects target‑driven updates onto this set, unifying standard training and existing data‑selection methods as special cases along a bias‑variance spectrum. Experiments on SFT, RLHF, and RLVR demonstrate that these methods outperform both standard training and state‑of‑the‑art data‑selection baselines, especially when general and target data are highly misaligned, while system optimizations keep overhead minimal.

By Pingbang Hu, Xueshen Liu, Z. Morley Mao, Jiaqi W. Ma
arXiv Machine Learning
Sep 11

Data Scarcity and Model Sparsity: Mixtures-of-Experts Overfit More to Repeated Data

The paper investigates how data repetition affects Mixture-of-Experts (MoE) language models compared to dense Transformers. Across models from 80 M to 1 B active parameters, MoEs degrade more quickly as data is repeated, with performance dropping significantly beyond 4× repetition and overtaking dense models only when strong regularization is applied. The study also identifies routing stabilization and expert specialization as key factors in MoE overfitting, and explores regularization techniques that can partially mitigate this issue.

By Atindra Jha, Margaret Li, Jure Leskovec, Percy Liang, Luke Zettlemoyer
arXiv AI
Sep 4

Xiaomi-TabLDM: A Tabular Foundation Model Technical Report

Xiaomi-TabLDM is a tabular foundation model that performs classification and regression via in-context learning without task‑specific fine‑tuning. It is pretrained solely on synthetic data from structural causal models, achieving top‑ranked regression results on multiple benchmarks while reducing training and prediction time compared to leading models. The architecture incorporates a three‑stage training strategy, dual‑stream feature grouping, lightweight attention residuals, and sparse mixture‑of‑experts, and it can further improve accuracy through test‑time compute scaling.

By TabLDM Team, Penghui Wang, Wei Liu, Hong Wang, Chengyue Huang, Yuxi Sun, Zirui Wang, Hongming Huang, Quan Wang, Chunxiao Liu, Erli Meng, Bin Wang
arXiv Machine Learning
Jul 24

DataPrep-Bench: Benchmarking LLMs as Training Data Preparators

arXiv:2607. 20465v1 Announce Type: new Abstract: The quality of training data fundamentally determines the capabilities of large language models (LLMs), yet no unified benchmark exists to measure how well LLMs, agents, and data-centric workflows actually prepare training data end to end.

By Hao Liang, Qifeng Cai, Yibo Lin, Jianzhuo Du, Qifeng Xia, Sizhe Qiu, Linzhuang Sun, Meiyi Qiang, Zhaoyang Han, Xiaochen Ma, Bohan Zeng, Ruichuan An, Conghui He, Wentao Zhang