Distributional Regression with Tabular Foundation Models: Evaluating Probabilistic Predictions via Proper Scoring Rules
summary
The gist
The gist The choice of scoring rule can induce different inductive biases during finite-sample optimization, leading to different model performances Evaluation Mismatch Modern tabular foundation
In short
This study investigates how choosing a specific scoring rule—a way to measure prediction quality—affects tabular foundation models like TabPFN and TabICL. Since standard benchmarks often ignore full distribution accuracy, the paper argues that using proper scoring rules for training and evaluation can induce different learning behaviors. This choice of score fundamentally alters the distribution a model learns.
Key concepts
- Proper Scoring Rules
- These are mathematical rules used to evaluate probabilistic forecasts, designed so that the true underlying distribution is always preferred over any incorrect one. While they are theoretically equivalent for the true population, different rules can guide a model's learning process differently on finite data.
- Distributional Predictions vs. Point Estimates
- Tabular foundation models naturally predict full probability distributions, but common benchmarks like those using RMSE only assess the single best guess (the conditional mean). This mismatch rewards models good at guessing the average value, not necessarily accurate in describing the entire range of possible outcomes.
- Inductive Bias
- The specific scoring rule used during training acts as an inductive bias, influencing how a machine learning model learns patterns from finite data. Different rules can lead to different learned behaviors and ultimately result in varied predictive performance when evaluated on downstream tasks.
Terminology used across episodes
This episode discusses
- Distributional Regression with Tabular Foundation Models: Evaluating Probabilistic Predictions via Proper Scoring Rules · Paper Radio
- TabPFN: A Transformer That Solves Small Tabular Classification Problems in a Second
- TabICL: A Tabular Foundation Model for In-Context Learning on Large Data
- TabICLv2: A better, faster, scalable, and open tabular foundation model · Paper Radio
- LimiX: Unleashing Structured-Data Modeling Capability for Generalist Intelligence
- Enforcing tail calibration when training probabilistic forecast models
- Asymmetric Penalties Underlie Proper Loss Functions in Probabilistic Forecasting
- ScoringBench: A Benchmark for Evaluating Tabular Foundation Models with Proper Scoring Rules · Paper Radio
- TabArena: A Living Benchmark for Machine Learning on Tabular Data
- TabReD: Analyzing Pitfalls and Filling the Gaps in Tabular Deep Learning Benchmarks
- TabTune: A Unified Library for Inference and Fine-Tuning Tabular Foundation Models
- Exploring Fine-Tuning for Tabular Foundation Models
- Benchmarking Tabular Foundation Models for Conditional Density Estimation in Regression
- From Aleatoric to Epistemic: Exploring Uncertainty Quantification Techniques in Artificial Intelligence
- Reassessing How to Compare and Improve the Calibration of Machine Learning Models
- Classifier Calibration at Scale: An Empirical Study of Model-Agnostic Post-Hoc Methods
- CalArena: A Large-Scale Post-Hoc Calibration Benchmark
- In-sample calibration yields conformal calibration guarantees
The paper
Distributional Regression with Tabular Foundation Models: Evaluating Probabilistic Predictions via Proper Scoring Rules · Read on arXiv
Jonas Landsgesell, Pascal Knoll, Tizian Wenzel
University of Stuttgart · Ludwig Maximilian University of Munich · Munich Center for Machine Learning
We study how the objective used to train or fine-tune a tabular foundation model affects its distributional predictions. Models such as TabPFN and TabICL provide forecasts of the full conditional distribution, but regression benchmarks including TabArena and TALENT assess them mainly through RMSE and R squared. Such assessments reward accurate conditional means while leaving the remaining distributional information unassessed. We therefore examine proper scoring rules as objectives for training and fine-tuning and as criteria for evaluating and ranking these models. Every strictly proper rule favors the true distribution at the population level, however, this shared optimum does not determine how learning proceeds with a finite sample. Our examples illustrate score-dependent preferences between forecasts, and our fine-tuning experiments show differences in predictive performance under different objectives. Experiments that fine-tune TabPFN and TabICL on several datasets reveal that changing the training objective can change performance across evaluation scores, with non-trivial interactions between the two. By connecting established methods from distributional regression, scoring-rule theory, and foundation-model adaptation, we show that selecting the score is part of specifying the model and can measurably alter the distribution it learns.
Transcript
Introduction to the show: ident: AI Radio. Generated commentary on the latest Artificial Intelligence papers.
Tom: Today's paper: "Distributional Regression with Tabular Foundation Models".
Jane: The gist The choice of scoring rule can induce different inductive biases during finite-sample optimization,
Tom: First, who's behind it and why it matters.
Title and authors: Tom: Let's look at the specifics of this paper, "Distributional Regression with Tabular Foundation Models: Evaluating Probabilistic Predictions via Proper Scoring Rules." It’s written by Jonas Landsgesell, Pascal Knoll, and Tizian Wenzel from institutions in Stuttgart and Munich. They are tackling this problem head-on.
Jane: They start by pointing out that traditional regression often uses point estimates like Mean Squared Error for the mean, or Pinball loss for quantiles. But the paper argues these are insufficient because they ignore aleatoric uncertainty, which is just the inherent variability from one data point to another
de Finetti, one thousand nine hundred seventy-five: <ref:2603.08206#pg1>.
Lu: They illustrate this with a toy dataset in Figure one <ref:2603.08206#pg1>. They show a problem where the target variable has multiple modes. The heat map on that figure compares what you get from training with the continuous ranked probability score, CRPS, against just the conditional mean estimate <ref:2603.08206#pg2>.
Meng: So they are showing that a model predicting the whole distribution with CRPS gets closer to the true target distribution than just predicting the average outcome. That's a visual comparison I can get behind from an engineering standpoint, but it highlights why relying only on point estimates is risky.
Lalam: It’s about capturing that inherent variability. The paper points out that point estimates don't account for how much noise there is from one observation to the next <ref:2603.08206#pg2>.
Tom: So they aren't just saying "distributions are better"; they are showing *why* and demonstrating that different scoring rules can change how the model optimizes itself during training, which is a crucial detail for anyone trying to get good results.
Jane: Right. The core of this paper is that the choice of scoring rule isn't just a footnote; it’s a major decision point when you are developing these models and figuring out how to evaluate them properly <ref:2603.08206#pg3>.
Lu: They state that during model development, the scoring rule for training and fine-tuning can be treated as a hyperparameter because different rules yield different performance outcomes, which isn't always self-aligned <ref:2603.08206#pg3>.
Meng: A hyperparameter that shifts the entire optimization path? That sounds like it adds complexity to the pipeline, but if it leads to better results on real data, I’m willing to look into that. But what about the cost of training with a CRPS loss instead of MSE?
Lalam: The paper suggests practitioners can narrow down which rule is appropriate using domain knowledge—like whether tail performance or center performance matters—though figuring out that specific choice is still a hard problem <ref:2603.08206#pg3>.
Tom: So they acknowledge the difficulty in choosing the right rule, but they’re proving that it has measurable consequences for model rankings, which is what we need to focus on when we are trying to compare different foundation models.
Jane: And that leads us into what the paper actually suggests we should be doing next, which is focusing on how these scoring rules impact performance in practice.
The paper's summary: Tom: Now let’s talk about the actual findings summarized in "Distributional Regression with Tabular Foundation Models: Evaluating Probabilistic Predictions via Proper Scoring Rules." Essentially, they are arguing that because models produce full distributions, we need scoring rules that measure those distributions correctly.
Jane: The summary boils down to this: point estimates are usually not enough to capture the important aspects of tabular data because they completely ignore aleatoric uncertainty <ref:2603.08206#pg2>.
Lu: They show that training a neural network with the continuous ranked probability score, CRPS, actually allows it to get closer to the true target distribution when compared to just looking at the conditional mean estimate <ref:2603.08206#pg2>.
Meng: So they are showing that distributional forecasts are better approximations of reality than just a single number. That’s good for understanding the underlying process, but how does that translate to what a user actually needs?
Lalam: It means we need metrics that reflect the actual predictive distribution rather than just minimizing the error on one single output <ref:2603.08206#pg1>. The paper is advocating for a shift in evaluation methodology.
Tom: They highlight that during model development, the choice of scoring rule for training and fine-tuning can be treated as a hyperparameter because different rules yield different performance outcomes, which isn't always self-aligned <ref:2603.08206#pg3>.
Jane: That’s important because it means that what looks like a good model trained with one loss function might actually rank poorly when evaluated by another scoring rule <ref:2603.08206#pg1>.
Lu: They show that different strictly proper scoring rules induce different inductive biases during finite-sample optimization, which is why we need to be careful about how we train and fine-tune <ref:2603.08206#pg1>.
Meng: So if I’m deploying a model, I need to know that the training objective I pick—whether it’s MSE or something else—isn't secretly biasing the model toward a specific kind of error, because that will show up in the final ranking.
The paper's improvements: Tom: The paper suggests several ways we can improve how we approach these models and evaluations. First, they emphasize incorporating a proper scoring rule as the primary loss function instead of just standard point-estimate losses like MSE <ref:2603.08206#pg1>.
Jane: They argue that this allows models to optimize for both calibration and sharpness simultaneously instead of just trying to minimize the conditional mean <ref:2603.08206#pg1>.
Lu: Because they show that good probabilistic forecasts need to be both sharp and calibrated, they highlight this trade-off—sharpness alone without calibration is misleading, which means we need both qualities <ref:2603.08206#pg5>.
Meng: So the improvement isn't just about getting a lower number on one metric; it’s about getting a forecast that is reliable in terms of how often it’s actually right and how narrow its spread is. That sounds like a much more robust deployment strategy.
Lalam: They also suggest that the AI system can adapt its evaluation objective based on the decision context, meaning we can tune models for task-specific scoring objectives <ref:2603.08206#pg3>.
Tom: That’s a huge practical implication, right? So if we’re in finance or weather forecasting, we aren't stuck using the same general rule for everything; we can use quantile-weighted CRPS for finance or something similar <ref:2603.08206#pg3>.
Jane: Exactly. It means the internal representation of the model can be tuned to prioritize specific risk structures relevant to that particular task, which moves us away from a one-size-fits-all evaluation approach.
Lu: Another improvement is making sure we optimize for distributional outputs themselves, like discretized densities or sets of quantiles <ref:2603.08206#pg21>. This is better than just predicting point estimates when the data generating process itself is complex, like bimodal <ref:2603.08206#pg1>.
Meng: So the architecture needs to be capable of outputting more than just a single number; it needs to handle those richer structures in the first place. That requires a different kind of model design.
Conclusion: Tom: So, wrapping up this discussion on "Distributional Regression with Tabular Foundation Models: Evaluating Probabilistic Predictions via Proper Scoring Rules," the main point is that we need to treat the choice of scoring rule as a first-class design decision when training and evaluating these models <ref:2603.08206#pg3>.
Jane: We’re moving away from just checking if a model gives us a good average prediction, toward asking which probabilistic objective best reflects the actual decision problem we are trying to solve <ref:2603.08206#pg1>.
Lu: The paper concludes that this shift means predictive distributions and the scoring rules used to assess them become central components of the entire machine learning pipeline <ref:2603.08206#pg1>.
Meng: From a deployment standpoint, it means we need to be more intentional about our objectives and how we measure success in production, not just chasing whatever benchmark gives us the highest R2 score <ref:2603.08206#pg1>.
Lalam: I think the big implication is that we should report proper scoring rules in benchmarks for tabular foundation model regression so others can see how these different methods perform <ref:2603.08206#pg1>.
Tom: It’s a shift toward developing probabilistic models where the scoring rules used to assess them become central to the whole process, which is what this paper argues we need to do with tabular foundation models.
Jane: That’s it for today on this paper. We've looked at how scoring rules change training and evaluation for these distribution-heavy foundation models, and now we’re ready for the next piece of research on arXiv.
More episodes
- 2510.03419-Multi-task Neural Diffusion Processes
- 2509.25146-Fast Feature Field (F cubed): A Predictive Representation of Events
- 2507.22612-Adaptive Duration Model for Text Speech Alignment
- 2510.26309-GraphCompliance: Structure-Aware Regulatory Compliance through Policy-Context Graph Alignment
- 2511.09907-Learning to Pose Problems: Reasoning-Driven and Solver-Adaptive Data Synthesis
- 2511.04689-Adaptive Testing for LLM Evaluation: A Psychometric Alternative to Static Benchmarks
- 2502.09854-Scaling Multimodal Search and Recommendation with Small Language Models via Upside-Down Reinforcement Learning
- 2512.20159-Benchmarking Code LLM-as-a-Judge via Constrained Perturbation and Calibration
- 2602.00122-VDE Bench: Evaluating The Capability of Image Editing Models to Modify Visual Documents
- 2602.18891-Orchestrating LLM Agents for Scientific Research: A Pilot Study of Multiple Choice Question (MCQ) Generation and Evaluation