Title
An Analysis Of Robust Cost Functions For Cnn In Computer-Aided Diagnosis
Abstract
Deep convolutional neural networks (CNNs) have proven to be powerful and flexible tools that advance the state-of-the-art in many fields, e.g. speech recognition, computer vision and medical imaging. Usually deep CNN models employ the logistic (soft-max) loss function in the training process of classification tasks. Recent evidence on a computer vision benchmark data-set indicates that the hinge (SVM) loss might give smaller misclassification errors on the test set compared to the logistic loss (i.e. offer better generality). In this paper, we study and compare four different loss functions for deep CNNs in the context of computer-aided abdominal and mediastinal lymph node detection and diagnosis (CAD) using CT images. Besides the logistic loss, we compare three other CNN losses that have not been previously studied for CAD problems. The experiments confirm that the logistic loss performs the worst among the four losses, and an additional 3% increase in detection rate at 3 false positives/volume can be obtained by just replacing it with Lorenz loss. The free-receiver operating characteristic curves of two of the three loss functions consistently outperform the logistic loss in testing.
Year
DOI
Venue
2018
10.1080/21681163.2016.1138240
COMPUTER METHODS IN BIOMECHANICS AND BIOMEDICAL ENGINEERING-IMAGING AND VISUALIZATION
Keywords
Field
DocType
Lymph node detection, convolutional neural networks, computer aided diagnosis
CAD,Medical imaging,Convolutional neural network,Computer science,Support vector machine,Computer-aided diagnosis,Artificial intelligence,Machine learning,Test set,False positive paradox
Journal
Volume
Issue
ISSN
6
3
2168-1163
Citations 
PageRank 
References 
1
0.35
10
Authors
5
Name
Order
Citations
PageRank
Adrian Barbu176858.59
Le Lu2129786.78
Holger Roth373745.70
Ari Seff450824.31
Ronald M. Summers589386.16