cs.LGMay 26, 2026

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

Authors: Yingshuo WangXian SunYanhang LiZhichao FanZexin Zhuang

Abstract

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.

Explore similar work

CardsList
  1. Do Tabular Foundation Models Agree with Themselves?

    Aug 6, 2026Christian Klötergens, Vijaya Krishna Yalavarthi, Lars Schmidt-Thieme +1Tabular Foundation ModelsPosterior