J4 ›› 2014, Vol. 41 ›› Issue (6): 83-88.doi: 10.3969/j.issn.1001-2400.2014.06.014

• Original Articles • Previous Articles     Next Articles

SVM decision-tree multi-classification strategy via electromagnetism-like mechanism

JIANG Jianguo;ZHAO Yuan;MENG Hongwei;LI Bo   

  1.  (School of Computer Science and Technology, Xidian Univ., Xi'an  710071, China)
  • Received:2013-03-22 Online:2014-12-20 Published:2015-01-19
  • Contact: JIANG Jianguo E-mail:jgjiang@mail.xidian.edu.cn

Abstract:

Based on the characteristics of the classification problem, a modified electromagnetism-like mechanism (EM) algorithm is designed, which is suitable for classification. Then a modified EM-based optimal decision-tree algorithm is proposed to deal with the SVM multi-class classification problem. First, EM is used to create an optimal or near-optimal decision tree automatically, which makes the margin between two classes maximal at every decision node. Then at every decision node, standard SVM is used to make binary classification. Finally, the SVM decision tree achieves multi-classification. Experimental results show that the proposed method is better than the traditional methods such as “1-a-1”, “1-a-r”, “DAG-SVM”,“DT-SVM” and “GADT-SVM”.

Key words: electromagnetism-like mechanism algorithm, support vector machine, multi-classification, maximal margin, decision tree

CLC Number: 

  • TP391