Share Email Print

Proceedings Paper

Particle swarm optimization of neural network CAD systems with clinically relevant objectives
Author(s): Piotr A. Habas; Jacek M. Zurada; Adel S. Elmaghraby; Georgia D. Tourassi
Format Member Price Non-Member Price
PDF $14.40 $18.00
cover GOOD NEWS! Your organization subscribes to the SPIE Digital Library. You may be able to download this paper for free. Check Access

Paper Abstract

Neural networks (NN) are typically developed to minimize the squared difference between the network's output and the target value for a set of training patterns; namely the mean squared error (MSE). However, lower MSE does not necessarily translate into a clinically more useful decision model. The purpose of this study was to investigate the particle swarm optimization (PSO) algorithm as an alternative way of NN optimization with clinically relevant objective functions (e.g., ROC and partial ROC area indices). The PSO algorithm was evaluated with respect to a NN-based CAD system developed to discriminate mammographic regions of interest (ROIs) that contained masses from normal regions based on 8 computer-extracted morphology-oriented features. Neural networks were represented as points (particle locations) in a D-dimensional search/optimization space where each dimension corresponded to one adaptable NN parameter. The study database of 1,337 ROIs (681 with masses, 656 normal) was split into two subsets to implement two-fold cross-validation sampling scheme. Neural networks were optimized with the PSO algorithm and the following objective functions (1) MSE, (2) ROC area index AUC, and (3) partial ROC area indices TPFAUC with TPF=0.90 and TPF=0.98. For comparison, performance of neural networks of the same architecture trained with the traditional backpropagation algorithm was also evaluated. Overall, the study showed that when the PSO algorithm optimized network parameters using a particular training objective, the NN test performance was superior with respect to the corresponding performance index. This was particularly true for the partial ROC area indices where statistically significant improvements were observed.

Paper Details

Date Published: 29 March 2007
PDF: 8 pages
Proc. SPIE 6514, Medical Imaging 2007: Computer-Aided Diagnosis, 65140M (29 March 2007); doi: 10.1117/12.713829
Show Author Affiliations
Piotr A. Habas, Univ. of Louisville (United States)
Jacek M. Zurada, Univ. of Louisville (United States)
Adel S. Elmaghraby, Univ. of Louisville (United States)
Georgia D. Tourassi, Duke Univ. Medical Ctr. (United States)

Published in SPIE Proceedings Vol. 6514:
Medical Imaging 2007: Computer-Aided Diagnosis
Maryellen L. Giger; Nico Karssemeijer, Editor(s)

© SPIE. Terms of Use
Back to Top