arrow
Return

Multi-class feature selection via Sparse Softmax with a discriminative regularization

delete2024-05-05
delete1
PRE
AI
Z
Zhenzhen Sun *
Z
Zexiang Chen
J
Jinghua Liu
于元隆 (Yuanlong Yu) *
DOI:10.1007/s13042-024-02185-5delete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
Feature selection plays a critical role in many machine learning applications as it effectively addresses the challenges posed by the curse of dimensionality and enhances the generalization capability of trained models. However, existing approaches for multi-class feature selection (MFS) often combine sparse regularization with a simple classification model, such as least squares regression, which can result in suboptimal performance. To address this limitation, this paper introduces a novel MFS method called Sparse Softmax Feature Selection ((SFS)-F-2). (SFS)-F-2 combines a l(2,0)-norm regularization with the Softmax model to perform feature selection. By utilizing the l(2,0)-norm, (SFS)-F-2 produces a more precise sparsity solution for the feature selection matrix. Additionally, the Softmax model improves the interpretability of the model's outputs, thereby enhancing the classification performance. To further enhance discriminative feature selection, a discriminative regularization, derived based on linear discriminate analysis (LDA), is incorporated into the learning model. Furthermore, an efficient optimization algorithm, based on the alternating direction method of multipliers (ADMM), is designed to solve the objective function of (SFS)-F-2. Extensive experiments conducted on various datasets demonstrate that (SFS)-F-2 achieves higher accuracy in classification tasks compared to several contemporary MFS methods.
Keywords:
Multi-class feature selection
L-2,L-0-norm regularization
Discriminative regularization
Alternating direction method of multipliers

Journal

International Journal of Machine Learning and Cybernetics cover
International Journal of Machine Learning and Cybernetics
IF:
2.7
Papers:
3.1K
Citations:
5.6K

Organization

H
huaqiao university
Scholars:
1.0W
Papers: 7.0K
Citations: 131
F
fuzhou university
Scholars:
3.2W
Papers: 2.1W
Citations: 31