Track papers across labs, topics, and trends.
Search papers, labs, and topics across Lattice.
MatrixFormer is a novel matrix-native transformer architecture designed to address the limitations of entry-by-entry prediction in existing tabular foundation models by explicitly leveraging the two-dimensional structure of matrices. Trained on synthetic low-rank and latent-factor matrices with varied missingness, it predicts a full distribution for each missing entry in a single forward pass. Zero-shot application of the same pre-trained model yields competitive results across diverse tasks including causal inference, benchmark-score completion, tabular imputation, and recommendation systems.
A single pre-trained transformer now solves matrix completion across causal inference, recommendation, and tabular imputation tasks by treating matrices natively.
Matrix completion underlies problems from tabular imputation to causal inference, yet existing tabular foundation models treat it as entry-by-entry prediction, repeating context for every target and discarding the matrix's two-dimensional structure. We introduce MatrixFormer, a pre-trained matrix-native transformer that predicts a full distribution for every missing entry in a single forward pass. MatrixFormer is trained entirely on synthetic low-rank and latent-factor matrices under diverse missingness patterns. Applied zero-shot and with the same model weights, MatrixFormer achieves competitive performance on causal inference panel-data tasks, language-model benchmark-score completion, tabular imputation, and recommendation systems matrix completion. These results position MatrixFormer as a general-purpose foundation model for matrix completion.