An improved ACO-based decision tree algorithm for imbalanced datasets
by Muhamad Hasbullah Mohd Razali; Rizauddin Saian; Yap Bee Wah; Ku Ruhana Ku-Mahamud
International Journal of Mathematical Modelling and Numerical Optimisation (IJMMNO), Vol. 11, No. 4, 2021

Abstract: Prediction of the minority class is challenging for classification of datasets with skewed class distribution. Bio-inspired classifier such as ant colony optimisation (ACO) decision tree could not provide effective decision boundaries since its entropy-based heuristics is affected by the strong presence of the majority class. Consequently, the developed trees were dominated by the likelihood of the majority class while the rare class being inadequately represented. This study proposes an improved algorithm called Hellinger-ant-tree-miner (HATM) which is inspired by the ant colony optimisation (ACO) meta heuristic for imbalanced learning using decision tree classification algorithm. The proposed algorithm was compared to the existing algorithm with entropy-based heuristics, ant-tree-miner (ATM) using nine publicly available imbalanced datasets and a simulation study. Simulation procedure reveals the superiority of HATM under imbalanced class environment as the sample size increased. Experimental results show that the performance of the existing algorithm evaluated via minority class prediction (MCP) and F-measure has improved due to the class skew-insensitiveness of Hellinger distance. The statistical significance test shows that HATM has higher mean in both performance measures than ATM, indicating a potential improvement of ACO decision tree structure for imbalanced class domain.

Online publication date: Mon, 25-Oct-2021

The full text of this article is only available to individual subscribers or to users at subscribing institutions.

 
Existing subscribers:
Go to Inderscience Online Journals to access the Full Text of this article.

Pay per view:
If you are not a subscriber and you just want to read the full contents of this article, buy online access here.

Complimentary Subscribers, Editors or Members of the Editorial Board of the International Journal of Mathematical Modelling and Numerical Optimisation (IJMMNO):
Login with your Inderscience username and password:

Username:        Password:          Forgotten your password?

Want to subscribe?
A subscription gives you complete access to all articles in the current issue, as well as to all articles in the previous three years (where applicable). See our Orders page to subscribe.

If you still need assistance, please email subs@inderscience.com