Back to AI News
arxiv

CoGA2M: A Composite GAM That Beats LightGBM Under Distribution Shift

This paper introduces CoGA2M, a multi-resolution, higher-order Composite Generalised Additive Model, and benchmarks it on the TALENT suite. CoGA2M ranks in the top third across all baselines on regression and classification tasks, matching state-of-the-art tabular methods. Under new out-of-distribution splits, CoGA2M outperforms LightGBM, highlighting a structural advantage of additive models in distribution shift.

AI Brief

Provides concrete evidence that interpretable GAMs can compete with and even outperform gradient boosting under distribution shift, a key practical concern.

arXiv:2611.05012v1 Announce Type: cross Abstract: Generalised Additive Models (GAMs) are a cornerstone of interpretable machine learning, offering transparent predictions through structured additive decompositions. Despite advances from classical GAMs to modern neural and spline-based variants, two notable gaps remain. First, comprehensive benchmarking of spline-based GAMs on large-scale tabular learning tasks, the dominant modality in applied settings, is lacking. Second, their robustness under distribution shift, a setting where structured inductive biases should confer advantages, remains unexplored. We propose a multi-resolution, higher-order Composite Generalised Additive Model (CoGA2M) and evaluate it on the TALENT benchmark suite. CoGA2M ranks in the top third across all baselines (29 for regression, 32 for classification), establishing competitive performance with state-of-the-art tabular methods. We further extend TALENT with out-of-distribution (OOD) splits exhibiting train-test distributional shift and benchmark CoGA2M against LightGBM, a consistently top-performing model on TALENT. Under distribution shift, CoGA2M outperforms LightGBM, demonstrating a clear structural advantage. Our results position spline-based GAMs as strong contenders for tabular prediction and reveal a previously unrecognised benefit of additive structure in the OOD regime.
Be the first to rate this article.

Comments & Questions (0)

Captcha
Please be respectful — let's keep the conversation friendly.

No comments yet

Be the first to comment!