解决表格数据上下文学习中的多尺度依赖与计算效率问题。
Orion-MSP: Multi-Scale Sparse Attention for Tabular In-Context Learning
- 采用多尺度处理捕捉特征层级交互
- 块稀疏注意力实现高效长程连接,支持高维表格
- 记忆模块支持组件间双向通信,提升表征能力
表格数据仍是真实应用场景中的主流格式。然而,由于特征类型异质性和多尺度复杂交互,构建有效的神经模型仍具挑战。近期表格上下文学习(ICL)方法如TabPFN和TabICL已在无需任务微调的情况下达到与梯度提升树(GBTs)相当的性能。但现有架构存在三大局限:(1) 单一尺度特征处理,忽略层次依赖;(2) 密集注意力导致表宽呈二次方增长;(3) 严格顺序处理,无法实现迭代优化与跨组件通信。为此,我们提出Orion-MSP,包含三项核心创新:(1) 多尺度处理以捕获层次化特征交互;(2) 块稀疏注意力结合窗口、全局与随机模式,兼顾可扩展性与长程连通性;(3) 类Perceiver记忆机制,支持安全双向信息流动。在多个基准测试中,Orion-MSP表现达或超越当前最优,且能有效扩展至高维表格,确立了高效表格上下文学习的新标准。模型已公开于https://github.com/Lexsi-Labs/Orion-MSP。
原文摘要 · Abstract (English)
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 .
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。