mlsanityData to 14 Oct 2026

Boosting gets full Attention for Relational Learning

arXiv 2402.14926 ↗First posted 22 Feb 2024cs.LGcited by 0

Not on the map. Why?

Abstract (arXiv)

More often than not in benchmark supervised ML, tabular data is flat, i.e. consists of a single m × d (rows, columns) file, but cases abound in the real world where observations are described by a set of tables with structural relationships. Neural nets-based deep models are a classical fit to incorporate general topological dependence among description features (pixels, words, etc.), but their suboptimality to tree-based models on tabular data is still well documented. In this paper, we introduce an attention mechanism for structured data that blends well with tree-based models in the training context of (gradient) boosting. Each aggregated model is a tree whose training involves two steps: first, simple tabular models are learned descending tables in a top-down fashion with boosting's class residuals on tables' features. Second, what has been learned progresses back bottom-up via attention and aggregation mechanisms, progressively crafting new features that complete at the end the set of observation features over which a single tree is learned, boosting's iteration clock is incremented and new class residuals are computed. Experiments on simulated and real-world domains display the competitiveness of our method against a state of the art containing both tree-based and neural nets-based models.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 12 of its 36 references that are in the map. First posted on arXiv 22 Feb 2024.

Builds on

12 of 36 references.

References · 12

  1. 2017Attention Is All You Needcited by 60,361
  2. 2017Graph Attention Networkscited by 8,923
  3. 2018Graph Neural Networks: A Review of Methods and Applicationscited by 1,576
  4. 2022TabPFN: A Transformer That Solves Small Tabular Classification Problems in a Secondcited by 674
  5. 2021Deep Neural Networks and Tabular Data: A Surveycited by 491
  6. 2017Representation Learning on Graphs: Methods and Applicationscited by 457
  7. 2021A Survey of Transformerscited by 401
  8. 2023Multimodal Deep Learningcited by 258
  9. 2022Why do tree-based models still outperform deep learning on tabular data?cited by 199
  10. 2022Transfer Learning with Deep Tabular Modelscited by 52
  11. 2022Yggdrasil Decision Forests: A Fast and Extensible Decision Forests Librarycited by 13
  12. 2023R-Cut: Enhancing Explainability in Vision Transformers with Relationship Weighted Out and Cutcited by 3

Built on by

Not cited yet.