用稀疏交互与动态采样提升高维表格回归的精度与效率
TabNSM: Neural Sparse Mixer for Tabular Regression

- 通过稀疏注意力机制聚焦关键特征,降低计算开销
- 在9个真实数据集上超越基线模型,高维数据表现尤其突出
- 适合处理噪声多、维度高的工业级表格回归任务
大规模高维表格回归仍具挑战:树模型鲁棒但缺乏端到端表示学习能力,深度模型可灵活学习特征但常需昂贵的交互建模且对噪声和冗余特征敏感。我们提出TabNSM,一种可扩展的回归框架,基于前期稀疏注意力与混合架构改进。核心为自适应稀疏交互模块(ASIM),融合前景特征发现、稀疏局部交互编码与特征-令牌混合,固定稀疏配置下接近线性复杂度。针对回归任务,引入三项互补组件:多阶段回归头实现渐进式预测优化;GridLoss,一种考虑目标结构的序数感知软分箱损失函数;RISE(基于损失分位数的重加权实例采样),依据难例分布进行采样。在九个真实世界回归基准测试中,TabNSM展现强大预测性能与实际可扩展性,尤其在高维异构数据集上取得一致优势。结果表明,选择性交互建模、结构化监督与难易感知采样共同构成高效可扩展的深度表格回归方案。
原文摘要 · Abstract (English)
Large-scale, high-dimensional tabular regression remains challenging: tree-based models are robust but lack end-to-end representation learning, while deep models enable flexible feature learning but often incur costly interaction modeling and sensitivity to noisy or redundant features. We propose TabNSM, a scalable regression framework that extends our earlier sparse-attention and mixer architectures. At its core, the Adaptive Sparse Interaction Module (ASIM) integrates foreground feature discovery, sparse local interaction encoding, and Feature-Token Mixing, providing near-linear complexity under fixed sparse configurations. For regression, TabNSM introduces three complementary components: a Multi-Stage Regression Head for progressive prediction refinement; GridLoss, an ordinal-aware soft-binning objective that incorporates target structure into representation learning; and RISE (Reweighted Instance Sampling by Error), a difficulty-aware sampling strategy based on loss-quantile bins. Across nine real-world regression benchmarks, TabNSM delivers strong predictive performance and practical scalability, with particularly consistent gains on high-dimensional and heterogeneous datasets. These results demonstrate that selective interaction modeling, structured regression supervision, and difficulty-aware sampling provide an effective and scalable approach to deep tabular regression.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。