Meeting Banner
Abstract #0807

The Effect of Activation Functions and Loss Functions on Deep Learning Based Fully Automated Knee Joint Segmentation

Sibaji Gaj1, Dennis Chan1, and Xiaojuan Li1
1Department of Biomedical Engineering, Cleveland Clinic, Cleveland, OH, United States

Studies on systematic evaluations of effects of activation functions and loss functions on deep learning-based automated knee compartments segmentation models are limited. In this work, we present a 2D-UNet model for simultaneous automated bone and cartilage segmentation, and analyze the effect of different activation functions (rectified linear unit[relu], sigmoid and softmax) at all or last layer, and different loss functions (categorical cross-entropy, multiclass dice coefficient loss) with and without surface distance weights, on model performance. The results showed significant performance differences in average surface distance (ASD) between different activation functions. Adding surface distance to loss functions improved segmentation performances.

How to access this content:

For one year after publication, abstracts and videos are only open to registrants of this annual meeting. Registrants should use their existing login information. Non-registrant access can be purchased via the ISMRM E-Library.

After one year, current ISMRM & ISMRT members get free access to both the abstracts and videos. Non-members and non-registrants must purchase access via the ISMRM E-Library.

After two years, the meeting proceedings (abstracts) are opened to the public and require no login information. Videos remain behind password for access by members, registrants and E-Library customers.

Click here for more information on becoming a member.

Keywords