Using the EM algorithm to train neural networks: misconceptions and a new algorithm for multiclass classification

Loading...
Thumbnail Image
File version
Author(s)
Ng, SK
McLachlan, GJ
Griffith University Author(s)
Primary Supervisor
Other Supervisors
Editor(s)
Date
2004
Size

350867 bytes

File type(s)

application/pdf

Location
License
Abstract

The expectation-maximization (EM) algorithm has been of considerable interest in recent years as the basis for various algorithms in application areas of neural networks such as pattern recognition. However, there exists some misconceptions concerning its application to neural networks. In this paper, we clarify these misconceptions and consider how the EM algorithm can be adopted to train multilayer perceptron (MLP) and mixture of experts (ME) networks in applications to multiclass classification. We identify some situations where the application of the EM algorithm to train MLP networks may be of limited value and discuss some ways of handling the difficulties. For ME networks, it is reported in the literature that networks trained by the EM algorithm using iteratively reweighted least squares (IRLS) algorithm in the inner loop of the M-step, often performed poorly in multiclass classification. However, we found that the convergence of the IRLS algorithm is stable and that the log likelihood is monotonic increasing when a learning rate smaller than one is adopted. Also, we propose the use of an expectation-conditional maximization (ECM) algorithm to train ME networks. Its performance is demonstrated to be superior to the IRLS algorithm on some simulated and real data sets.

Journal Title

IEEE Transactions on Neural Networks

Conference Title
Book Title
Edition
Volume

15

Issue

3

Thesis Type
Degree Program
School
Publisher link
Patent number
Funder(s)
Grant identifier(s)
Rights Statement
Rights Statement

© 2004 IEEE. Personal use of this material is permitted. However, permission to reprint/republish this material for advertising or promotional purposes or for creating new collective works for resale or redistribution to servers or lists, or to reuse any copyrighted component of this work in other works must be obtained from IEEE.

Item Access Status
Note
Access the data
Related item(s)
Subject

Artificial intelligence not elsewhere classified

Cognition

Persistent link to this record
Citation
Collections