mlsanityData to 7 Oct 2026

Auditing and Fixing Economic Validity in Tabular Foundation Models for Discrete Choice

arXiv 2605.26559 ↗First posted 26 May 2026cs.LG cs.AI econ.EMcited by 6

Not on the map. Why?

Abstract (arXiv)

Tabular foundation models achieve strong accuracy on choice prediction tasks, but their predictions often violate the economic logic those tasks require: raising a price sometimes increases predicted demand, and implied willingness-to-pay estimates are frequently negative or implausible. We propose a two-stage adapter that embeds foundation model predictions within a utility-maximization framework. In the first stage, we estimate a standard choice model whose parameters are constrained to obey economic theory. In the second stage, we freeze those parameters and train a correction term that incorporates the foundation model's predictions as additional information. The result is a model that inherits the foundation model's accuracy gains while guaranteeing monotonic price-demand relationships under policy perturbation and producing analytically computable trade-off measures. On two transportation datasets, the adapter recovers up to 13 percentage points of accuracy over a standard logit model while maintaining perfect economic consistency, something neither the raw foundation models nor conventional distillation achieve.

Key facts. Cited by 6 papers in the map (data through 7 Oct 2026). Builds on 4 of its 13 references that are in the map. First posted on arXiv 26 May 2026.

Builds on

4 of 13 references.

References · 4

  1. 2015Distilling the Knowledge in a Neural Networkcited by 10,076
  2. 2025Mitra: Mixed Synthetic Priors for Enhancing Tabular Foundation Modelscited by 67
  3. 2021Choice modelling in the age of machine learning -- discussion papercited by 26
  4. 2020A Neural-embedded Choice Model: TasteNet-MNL Modeling Taste Heterogeneity with Flexibility and Interpretabilitycited by 20

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2026: 6.

6 papers.