.png){kind=icon}
.png){kind=icon}
Return
Generalized sparse radial basis function networks for multi-classification problems
.png){kind=icon}
PRE
AI
DOI：10.1016/j.asoc.2024.111361.png)
.png){kind=icon}
Original request for help.png){kind=icon}
Share.png){kind=icon}
Save
Abstract
En 中文
Over the past decades, the radial basis function network (RBFN) has attracted extensive attention due to its simple network structure and powerful learning ability. Meanwhile, regularization methods have been intensively applied in RBFNs to enhance the performance of networks. A common regularization method is the l(2) regularization, which improves the stability and generalization ability but leads to dense networks. Another common regularization method is to employ the l(1) regularization that can successfully improve the sparsity of RBFN. The better strategy is to use the elastic-net regularization that combines both l(2) and l(1) regularization to improve stability and sparsity simultaneously. However, in multi-classification tasks, even the elastic-net regularization can only prune the redundant weights of nodes and cannot ensure the sparsity at the node level. In this paper, we propose a generalized sparse RBFN (GS-RBFN) based on the extended elastic-net regularization to handle multi-classification problems. By using the extended elastic-net regularization that integrates the Frobenius norm and L-2,L-1 norm, we accomplish the stability and sparsity of RBFN for multi-classification problems, of which the binary classification problem is only a special case. In order to improve the training efficiency under large-scale tasks, we further propose the parallel GS-RBFN (PGS-RBFN) with the matrix inversion lemma to accelerate the intensive computation. The alternating direction method of multipliers (ADMM) and its consensus variant are applied to train our proposed models, and we demonstrate their convergence in solving corresponding optimization problems. Experimental results on multi-classification datasets illustrate the effectiveness and advantages of our algorithms in accuracy, sparsity and convergence.
Keywords:
Radial basis function network
Sparsity
Multi-classification
Alternating direction method of multipliers
Parallel computing