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

FEAT: A Linear-Complexity Foundation Model for Extremely Large Structured Data

About

Structured data is foundational to healthcare, finance, e-commerce, and scientific data management. Large structured-data models (LDMs) extend the foundation model paradigm to unify heterogeneous datasets for tasks such as classification, regression, and decision support. However, existing LDMs face major limitations. First, most rely on sample-wise self-attention, whose O(N^2) complexity limits the sample count. Second, linear sequence models often degrade representations due to hidden-state compression and artificial causal bias. Third, synthetic-only pre-training often fails to match real-world distributions. We propose FEAT, a linear-complexity foundation model for extremely large structured data. FEAT introduces a multi-layer dual-axis architecture that replaces quadratic attention with hybrid linear encoding. The architecture combines adaptive-fusion bi-Mamba-2 (AFBM) for local sample dependencies and convolutional gated linear attention (Conv-GLA) for global memory. This design enables linear-complexity cross-sample modeling while preserving expressive representations. To improve robustness, FEAT adopts a hybrid structural causal model pipeline and a stable reconstruction objective. Experiments on 11 real-world datasets show that FEAT consistently outperforms baselines in zero-shot performance, while scaling linearly and achieving up to 40x faster inference.

Zhenghang Song, Tang Qian, Lu Chen, Yushuai Li, Zhengke Hu, Bingbing Fang, Yumeng Song, Junbo Zhao, Sheng Zhang, Tianyi Li• 2026

Related benchmarks

TaskDatasetResultRank
InferenceScalability and Efficiency Evaluation D=20 (test)
Inference Latency (ms)149.2
26
RegressionGI-REG
RMSE0.4703
10
RegressionBCCO-REG
RMSE0.406
10
ClassificationGI-CLS
AUC0.8991
9
ClassificationTabarena CLS
AUC0.8638
9
ClassificationTabzilla CLS
AUC92.51
9
RegressionCTR23-REG
RMSE0.4053
9
RegressionPFN REG
RMSE0.5257
9
ClassificationBCCO-CLS
AUC85.79
9
RegressionTalent-REG
RMSE0.4708
9
Showing 10 of 12 rows

Other info

Follow for update