ProbML · 2026

CogFormer: Learn All Your Models Once

Jerry M. Huang, Lukas Schumacher, Niek Stevenson, Stefan T. Radev

We pilot CogFormer, a deep learning framework that uses masked flow matching to simultaneously perform Bayesian inference on many similar cognitive models. In practice, this means that cognitive scientists can fit an entire family of models to participant data at once, eliminating the bottleneck that makes model evaluation and iteration impractical.

amortized inference
simulation-based inference
Bayesian workflow
meta-amortized inference
cognitive modeling