CRUMB: Efficient Prior Fitted Network Inference via Distributionally Matched Context Batching

Jamie Heredge, Mattia J. Villani, Pranav Deshpande +2

Prior-fitted networks (PFNs) are a promising class of tabular foundation models that perform in-context learning, whereby the entire labelled training set is supplied as context, a…