Transfer Learning on Decision Tree with Class Imbalance

Published: 01 Jan 2019, Last Modified: 22 May 2024ICTAI 2019EveryoneRevisionsBibTeXCC BY-SA 4.0
Abstract: Transfer learning has attracted growing attention from the machine learning community. It addresses real-world issues that were not considered in the classical methods, especially with model-based methods when source data is not available. Recent work on transfer learning for Decision Trees has been proposed in the form of two algorithms named SER and STRUT. These methods tackle with success potential changes between source and target data, however several classification tasks may face imbalance data, hence degrading the performance of such transfer methods. In this paper, we study the impact of class imbalance on these algorithms and propose an adaptation for each of them, named SER* and STRUT*. These variants have been tested on real and synthetic data, these latter revealing to be of great use to control at the same time various transformations between source and target along with imbalance conditions. Results on both types of data show the benefits of our approach and suggest several perspectives for further research.
Loading