Journal Search Engine
Search Advanced Search Adode Reader(link)
Download PDF Export Citaion korean bibliography PMC previewer
ISSN : 1598-7248 (Print)
ISSN : 2234-6473 (Online)
Industrial Engineering & Management Systems Vol.10 No.4 pp.272-278
DOI :

Optimization of Decision Tree for Classification Using a Particle Swarm

Chi-Hyuck Jun, 1Yun-Ju Cho, 2Hyeseon Lee
Department of Industrial and Management Engineering Pohang University of Science and Technology, Pohang, 790-784, Korea
1Department of Industrial and Management Engineering Pohang University of Science and Technology, Pohang, 790-784, Korea, 2Department of Industrial and Management Engineering
Pohang University of Science and Technology, Pohang, 790-784, Korea
Received, September 4, 2011; Revised, October 20, 2011; Accepted, October 31, 2011

Abstract

Decision tree as a classification tool is being used successfully in many areas such as medical diagnosis, customer churn prediction, signal detection and so on. The main advantage of decision tree classifiers is their capability to break down a complex structure into a collection of simpler structures, thus providing a solution that is easy to interpret. Since decision tree is a top-down algorithm using a divide and conquer induction process, there is a risk of reaching a local optimal solution. This paper proposes a procedure ofoptimally determining thresholds of the chosen variables for a decision tree using an adaptive particle swarm optimization (APSO). The proposed algorithm consists of two phases. First, we construct a decision tree and choose the relevant variables. Second, we find the optimum thresholds simultaneously using an APSO for those selected variables. To validate the proposed algorithm, several artificial and real datasets are used. We compare our results with the original CART results and show that the proposed algorithm is promising for improving prediction accuracy.

10-4-05.pdf436.2KB

Reference