arXiv:2410.24210cs.LG2024-10ICLR被引 166

用参数高效集成提升表格数据深度学习表现,性能超越现有模型。

TabM: Advancing Tabular Deep Learning with Parameter-Efficient Ensembling

  • 通过共享参数的隐式多模型集成,实现高效预测
  • 在多个公开数据集上表现最优,优于注意力与检索类模型
  • 适合追求高性能与低资源消耗的表格学习研究者

面向表格数据的深度学习架构从简单的多层感知机(MLP)到复杂的Transformer和检索增强方法不一而足。本研究揭示了一个此前被忽视的重要机会:基于MLP的表格架构可通过参数高效集成实现显著提升。所提出的TabM模型通过高效集成机制,使单个模型能同时生成多个预测结果,其底层隐式MLP共享大部分参数并同步训练,相比传统深度集成,在性能和效率上均有显著提升。以TabM为新基准,我们在多个公共数据集上对表格深度学习模型进行了大规模评估,涵盖任务表现与运行效率,重新审视了该领域的技术格局。结果显示,包括TabM在内的MLP类模型整体表现更优且更具实用性,优于基于注意力或检索的架构。具体而言,TabM在各类表格深度学习模型中表现最佳。进一步实证分析表明,TabM的多个预测结果个体较弱,但集体作用强大。本工作为表格深度学习带来了一项重要技术突破,显著优化了性能与效率的权衡,为研究者和实践者提供了一个简单而强大的新基准。

原文摘要 · Abstract (English)

Deep learning architectures for supervised learning on tabular data range from simple multilayer perceptrons (MLP) to sophisticated Transformers and retrieval-augmented methods. This study highlights a major, yet so far overlooked opportunity for designing substantially better MLP-based tabular architectures. Namely, our new model TabM relies on efficient ensembling, where one TabM efficiently imitates an ensemble of MLPs and produces multiple predictions per object. Compared to a traditional deep ensemble, in TabM, the underlying implicit MLPs are trained simultaneously, and (by default) share most of their parameters, which results in significantly better performance and efficiency. Using TabM as a new baseline, we perform a large-scale evaluation of tabular DL architectures on public benchmarks in terms of both task performance and efficiency, which renders the landscape of tabular DL in a new light. Generally, we show that MLPs, including TabM, form a line of stronger and more practical models compared to attention- and retrieval-based architectures. In particular, we find that TabM demonstrates the best performance among tabular DL models. Then, we conduct an empirical analysis on the ensemble-like nature of TabM. We observe that the multiple predictions of TabM are weak individually, but powerful collectively. Overall, our work brings an impactful technique to tabular DL and advances the performance-efficiency trade-off with TabM -- a simple and powerful baseline for researchers and practitioners.

表格学习高效集成MLP性能优化

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。