Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Orion-MSP: Multi-Scale Sparse Attention for Tabular In-Context Learning

About

Tabular data remain the predominant format for real-world applications. Yet, developing effective neural models for tabular data remains challenging due to heterogeneous feature types and complex interactions occurring at multiple scales. Recent advances in tabular in-context learning (ICL), such as TabPFN and TabICL, have achieved state-of-the-art performance comparable to gradient-boosted trees (GBTs) without task-specific fine-tuning. However, current architectures exhibit key limitations: (1) single-scale feature processing that overlooks hierarchical dependencies, (2) dense attention with quadratic scaling in table width, and (3) strictly sequential component processing that prevents iterative representation refinement and cross-component communication. To address these challenges, we introduce Orion-MSP, a tabular ICL architecture featuring three key innovations: (1) multi-scale processing to capture hierarchical feature interactions; (2) block-sparse attention combining windowed, global, and random patterns for scalable efficiency and long-range connectivity; and (3) a Perceiver-style memory enabling safe bidirectional information flow across components. Across diverse benchmarks, Orion-MSP matches or surpasses state-of-the-art performance while scaling effectively to high-dimensional tables, establishing a new standard for efficient tabular in-context learning. The model is publicly available at https://github.com/Lexsi-Labs/Orion-MSP .

Mohamed Bouadi, Pratinav Seth, Aditya Tanna, Vinay Kumar Sankarapu• 2025

Related benchmarks

TaskDatasetResultRank
Multiclass ClassificationMulticlass panel 3 healthcare datasets v1.0 (test)
Macro AUC77.3
31
Binary Classificationbinary health datasets avg (test)
AUC0.864
29
Group Fairness8 health datasets aggregated across 4 attributes
DP Difference0.153
24
Tabular Classification153 datasets shared-coverage v1 (single seed)
AUC87.8
21
Tabular Data InferenceStructured Health Data Average (various)
Latency (ms)1.91e+3
13
Tabular ClassificationBenchmark runs Macro-mean
AUC87.8
10
Showing 6 of 6 rows

Other info

Follow for update