arrow
返回

A new large-scale learning algorithm for generalized additive models

delete2023-05-31
delete2
PRE
AI
顾彬 封面图
顾彬 (Bin Gu)
C
Chenkang Zhang
Z
Zhouyuan Huo
H
Heng Huang *
DOI:10.1007/s10994-023-06339-4delete
delete原文链接
delete原文求助
delete分享
delete收藏
摘要

摘要

En 中文
Additive model plays an important role in machine learning due to its flexibility and interpretability in the prediction function. However, solving large-scale additive models is a challenging task due to several difficulties. Until now, scaling up additive models is still an open problem. To address this challenging problem, in this paper, we propose a new doubly stochastic optimization algorithm for solving the generalized additive models (DSGAM). We first propose a generalized formulation of additive models without the orthogonal hypothesis on the basis function. After that, we propose a wrapper algorithm to optimize the generalized additive models. Importantly, we introduce a doubly stochastic gradient algorithm (DSG) to solve an inner subproblem in the wrapper algorithm, which can scale well in sample size and dimensionality simultaneously. Finally, we prove the fast convergence rate of our DSGAM algorithm. The experimental results on various large-scale benchmark datasets not only confirm the fast convergence of our DSGAM algorithm, but also show a huge reduction of computational time compared with existing algorithms, while retaining the similar generalization performance.
Keyword:
Additive model
Doubly stochastic gradient
Wrapper algorithm

期刊

Machine Learning 封面图
Machine Learning
IF:
2.9
论文数:
2.7K
被引数:
3.4W

机构

P
pennsylvania commonwealth system of higher education (pcshe)
学者数:
12.9W
论文数: 11.7W
被引数: 177
引用论文

引用论文

学者 查看更多内容