arXiv Machine Learning

TabClustPFN: A Prior-data Fitted Network for Tabular Data Clustering

TabClustPFN is a Prior-data Fitted Network designed for clustering tabular data. It jointly addresses the lack of supervision, unknown cluster cardinality, and label switching by integrating a hybrid pretraining prior, a decoupled partition‑inference and cardinality‑inference architecture, and a permutation‑invariant SoftARI training objective. On a curated benchmark of 44 real‑world tabular datasets, it outperforms classical, deep, and amortized baselines while maintaining efficient runtime.

arXiv AI
Jun 11

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

arXiv:2606. 11473v1 Announce Type: cross Abstract: 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, and predictions for test queries are produced in a single forward pass.

By Jamie Heredge, Mattia J. Villani, Pranav Deshpande, Akshay Seshadri, Niraj Kumar
arXiv Machine Learning
Oct 12

TabMF: Learning Dataset Representations with Tabular Foundation Models

TabMF is a framework that repurposes pretrained tabular foundation models (TFMs) as encoders for supervised datasets, adding a dataset-level readout and fine‑tuning via contrastive post‑training on synthetic views. The authors instantiate TabMF on TabICLv2 and TabPFN‑2.5 and evaluate the resulting representations on six tasks involving unseen real datasets. The approach achieves the best aggregate performance among both learned and classical dataset representations, excelling in dataset retrieval, compact benchmark construction, and distribution‑shift severity assessment.

By Rafiq Kamel, Nicolas Atienza, Aziz Fouch\'e, Nicola Cancedda, Josif Grabocka
arXiv Machine Learning
Aug 28

ClusterAttention: A training-free speedup of bidirectional attention

ClusterAttention is a training‑free technique that speeds up bidirectional attention by recursively clustering keys and queries into fixed‑size, power‑of‑two blocks, enabling block‑sparse attention to match dense attention latency on GPUs. The method derives error bounds for sparse attention, showing tighter clusters can reduce error when compensated via centroids, and demonstrates significant speedups—up to six‑fold on large tabular data and 1.8× on video generation—while preserving over 99% of dense accuracy.

By Kasper Nordenram, Amelie Dittmann
Hugging Face Trending Papers
Oct 5

Adapting prior-data fitted networks for tabular anomaly detection

The paper explores how prior-data fitted networks (PFNs) can be adapted for anomaly detection in tabular data. Using frozen TabPFN features, the authors show that scoring samples by nearest-neighbor distance in feature space yields strong results, and they identify optimal layers and extraction procedures. They further improve performance by fine‑tuning the model on a reference set, achieving higher AUROC on the ADBench benchmark with both a fine‑tuning‑free (ZEN) and a fine‑tuned (FOCUS) approach.

arXiv AI
Oct 7

Towards One-for-All Foundation Model for Attributed Graph Clustering

The paper introduces OFAG, a foundation model designed for attributed graph clustering that can be trained once and applied to diverse graphs without graph‑specific tuning. It learns a reusable clustering strategy from synthetic graphs and uses a dimension‑agnostic encoder to handle varying feature spaces, producing clustering‑friendly node representations in a single forward pass. Across ten datasets, a frozen OFAG model outperforms baselines in both speed and clustering quality, and the authors provide code and pretrained checkpoints for easy adoption.

By Yunhui Liu, Xudong Jin, Kang Zhang, Danshuo An, Yu Xing, Te Song, Jia Liu, Tieke He
arXiv Machine Learning
Sep 17

TabICLv2: A better, faster, scalable, and open tabular foundation model

TabICLv2 is a new state‑of‑the‑art tabular foundation model that outperforms existing methods on regression and classification tasks. It relies on a synthetic data generation engine for diverse pretraining, architectural innovations such as a scalable softmax attention, and optimized training protocols that replace AdamW with the Muon optimizer. On the TabArena and TALENT benchmarks, TabICLv2 surpasses the current best model, RealTabPFN‑2.5, without any tuning, while also being faster and capable of handling million‑scale datasets with limited GPU memory.

By Jingang Qu, David Holzm\"uller, Ga\"el Varoquaux, Marine Le Morvan
arXiv AI
Sep 15

Generalization Can Emerge in Tabular Foundation Models From a Single Table

The paper demonstrates that a tabular foundation model can achieve strong generalization using only a single real table for self‑supervised pre‑training, challenging the belief that large synthetic or real datasets are necessary. By systematically pre‑training and evaluating across diverse benchmarks, the authors show that the number and quality of tasks that can be derived from a dataset are critical for downstream performance. This finding suggests that carefully constructed task sets from limited data can enable effective transfer learning in tabular models.

By Junwei Ma, Nour Shaheen, Alex Labach, Amine Mhedhbi, Frank Hutter, Anthony L. Caterini, Valentin Thomas
arXiv Machine Learning
Oct 15

OrionMoE-v1.0: A Mixture-of-Experts Foundation Model for Tabular In-Context Learning

OrionMoE-v1.0 is a compact tabular foundation model that uses a dual‑axis Transformer and a sparse mixture‑of‑experts architecture to perform in‑context learning on labeled support tables. It selectively activates experts along the feature axis, guided by both individual feature representations and overall dataset context, enabling adaptive computation across diverse tabular problems. Pretrained on synthetic classification tasks, it achieves competitive performance on multiple benchmarks, especially on small‑context, high‑dimensional, and class‑imbalanced datasets.

By Mohamed Bouadi, Nassim Bouarour, Aditya Tanna, Vinay Kumar Sankarapu
arXiv AI
Sep 21

Federated Deep Clustering Networks for High-Dimensional and Heterogeneous Data

The paper introduces FedDCN, a federated deep clustering network that jointly optimizes reconstruction and clustering losses for high‑dimensional, heterogeneous data. It addresses challenges of non‑IID client data by generating synthetic augmentations and applying geometric regularization to align latent spaces. Experiments show the method’s effectiveness under both IID and non‑IID settings, and the authors outline future research directions.

By Morris Stallmann, Charalampos S. Kouzinopoulos, Marcin Pietrasik, Anna Wilbik