Achieving More with Less: A Tensor-Optimization-Powered Ensemble Method

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Yuan, Jinghui, Jiang, Weijin, Cao, Zhe, Xie, Fangyuan, Wang, Rong, Nie, Feiping, Yuan, Yuan
Natura: Preprint
Pubblicazione: 2024
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866913464534958080
author Yuan, Jinghui
Jiang, Weijin
Cao, Zhe
Xie, Fangyuan
Wang, Rong
Nie, Feiping
Yuan, Yuan
author_facet Yuan, Jinghui
Jiang, Weijin
Cao, Zhe
Xie, Fangyuan
Wang, Rong
Nie, Feiping
Yuan, Yuan
contents Ensemble learning is a method that leverages weak learners to produce a strong learner. However, obtaining a large number of base learners requires substantial time and computational resources. Therefore, it is meaningful to study how to achieve the performance typically obtained with many base learners using only a few. We argue that to achieve this, it is essential to enhance both classification performance and generalization ability during the ensemble process. To increase model accuracy, each weak base learner needs to be more efficiently integrated. It is observed that different base learners exhibit varying levels of accuracy in predicting different classes. To capitalize on this, we introduce confidence tensors $\tilde{\mathbfΘ}$ and $\tilde{\mathbfΘ}_{rst}$ signifies the degree of confidence that the $t$-th base classifier assigns the sample to class $r$ while it actually belongs to class $s$. To the best of our knowledge, this is the first time an evaluation of the performance of base classifiers across different classes has been proposed. The proposed confidence tensor compensates for the strengths and weaknesses of each base classifier in different classes, enabling the method to achieve superior results with a smaller number of base learners. To enhance generalization performance, we design a smooth and convex objective function that leverages the concept of margin, making the strong learner more discriminative. Furthermore, it is proved that in gradient matrix of the loss function, the sum of each column's elements is zero, allowing us to solve a constrained optimization problem using gradient-based methods. We then compare our algorithm with random forests of ten times the size and other classical methods across numerous datasets, demonstrating the superiority of our approach.
format Preprint
id arxiv_https___arxiv_org_abs_2408_02936
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Achieving More with Less: A Tensor-Optimization-Powered Ensemble Method
Yuan, Jinghui
Jiang, Weijin
Cao, Zhe
Xie, Fangyuan
Wang, Rong
Nie, Feiping
Yuan, Yuan
Machine Learning
Ensemble learning is a method that leverages weak learners to produce a strong learner. However, obtaining a large number of base learners requires substantial time and computational resources. Therefore, it is meaningful to study how to achieve the performance typically obtained with many base learners using only a few. We argue that to achieve this, it is essential to enhance both classification performance and generalization ability during the ensemble process. To increase model accuracy, each weak base learner needs to be more efficiently integrated. It is observed that different base learners exhibit varying levels of accuracy in predicting different classes. To capitalize on this, we introduce confidence tensors $\tilde{\mathbfΘ}$ and $\tilde{\mathbfΘ}_{rst}$ signifies the degree of confidence that the $t$-th base classifier assigns the sample to class $r$ while it actually belongs to class $s$. To the best of our knowledge, this is the first time an evaluation of the performance of base classifiers across different classes has been proposed. The proposed confidence tensor compensates for the strengths and weaknesses of each base classifier in different classes, enabling the method to achieve superior results with a smaller number of base learners. To enhance generalization performance, we design a smooth and convex objective function that leverages the concept of margin, making the strong learner more discriminative. Furthermore, it is proved that in gradient matrix of the loss function, the sum of each column's elements is zero, allowing us to solve a constrained optimization problem using gradient-based methods. We then compare our algorithm with random forests of ten times the size and other classical methods across numerous datasets, demonstrating the superiority of our approach.
title Achieving More with Less: A Tensor-Optimization-Powered Ensemble Method
topic Machine Learning
url https://arxiv.org/abs/2408.02936