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

    Research output: Chapter in Book/Report/Conference proceedingConference contribution

    6 Citations (Scopus)

    Abstract

    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.

    Original languageEnglish
    Title of host publicationProceedings - 2nd International Conference on Computational Intelligence, Communication Systems and Networks, CICSyN 2010
    Pages44-48
    Number of pages5
    DOIs
    Publication statusPublished - 2010
    Event2nd International Conference on Computational Intelligence, Communication Systems and Networks, CICSyN 2010 - Liverpool
    Duration: 2010 Jul 282010 Jul 30

    Other

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

    Keywords

    • Artificial neural network
    • Imbalanced data set problems
    • Particle swarm optimization

    ASJC Scopus subject areas

    • Hardware and Architecture
    • Computer Networks and Communications

    Fingerprint Dive into the research topics of 'A modified artificial neural network learning algorithm for imbalanced data set problem'. Together they form a unique fingerprint.

  • Cite this

    Adam, A., Shapiai, I., Ibrahim, Z., Khalid, M., Chew, L. C., Jau, L. W., & Watada, J. (2010). A modified artificial neural network learning algorithm for imbalanced data set problem. In Proceedings - 2nd International Conference on Computational Intelligence, Communication Systems and Networks, CICSyN 2010 (pp. 44-48). [5614727] https://doi.org/10.1109/CICSyN.2010.9