A modified artificial neural network learning algorithm for imbalanced data set problem

Asrul Adam, Ibrahim Shapiai, Zuwairie Ibrahim, Marzuki Khalid, Lim Chun Chew, Lee Wen Jau, Junzo Watada

    研究成果: Conference contribution

    6 被引用数 (Scopus)

    抄録

    A modified learning algorithm of Artificial Neural Networks (ANN) is introduced in this paper to solve imbalanced data set problems. In solving imbalanced data set, it is critical to predict the minority class due to their imbalanced nature. In order to improve the standard ANN classifier prediction performance, this paper focuses on optimizing the decision boundary of the step function at the output layer of ANN using particle swarm optimization (PSO). A feedforward ANN is chosen in this study. Firstly, a conventional back propagation algorithm is employed to train the ANN. PSO is then applied to train the real predicted output of training data from this trained network. As the result, the optimum value of decision boundary is found and applied to the classifier. Prediction performance is assessed by G-mean, which is a measure to indicate the efficiency of classifiers for imbalanced data sets. Based on experimental results, the proposed model is able to solve imbalanced data sets problem with better performance compared to the standard ANN.

    本文言語English
    ホスト出版物のタイトルProceedings - 2nd International Conference on Computational Intelligence, Communication Systems and Networks, CICSyN 2010
    ページ44-48
    ページ数5
    DOI
    出版ステータスPublished - 2010
    イベント2nd International Conference on Computational Intelligence, Communication Systems and Networks, CICSyN 2010 - Liverpool
    継続期間: 2010 7 282010 7 30

    Other

    Other2nd International Conference on Computational Intelligence, Communication Systems and Networks, CICSyN 2010
    CityLiverpool
    Period10/7/2810/7/30

    ASJC Scopus subject areas

    • Hardware and Architecture
    • Computer Networks and Communications

    フィンガープリント 「A modified artificial neural network learning algorithm for imbalanced data set problem」の研究トピックを掘り下げます。これらがまとまってユニークなフィンガープリントを構成します。

    引用スタイル