A Non-Convex Proximal Approach for Centroid-Based Classification - CentraleSupélec
Communication Dans Un Congrès Année : 2022

A Non-Convex Proximal Approach for Centroid-Based Classification

Résumé

In this paper, we propose a novel variational approach for supervised classification based on transform learning. Our approach consists of formulating an optimization problem on both the transform matrix and the centroids of the classes in a low-dimensional transformed space. The loss function is based on the distance to the centroids, which can be chosen in a flexible manner. To avoid trivial solutions or highly correlated clusters, our model incorporates a penalty term on the centroids, which encourages them to be separated. The resulting non-convex and non-smooth minimization problem is then solved by a primal-dual alternating minimization strategy. We assess the performance of our method on a bunch of supervised classification problems and compare it to state-of-the-art methods.

Dates et versions

hal-03688210 , version 1 (03-06-2022)

Identifiants

Citer

Mewe-Hezoudah Kahanam, Laurent Le Brusquet, Segolene Martin, Jean-Christophe Pesquet. A Non-Convex Proximal Approach for Centroid-Based Classification. 2022 IEEE International Conference on Acoustics, Speech and Signal Processing (ICASSP 2022), May 2022, Singapore, Singapore. pp.5702-5706, ⟨10.1109/ICASSP43922.2022.9747071⟩. ⟨hal-03688210⟩
72 Consultations
0 Téléchargements

Altmetric

Partager

More