CEL-Unet: a novel CNN architecture for 3D Segmentation of Knee Bones affected by Severe Osteoarthritis for PSI-Based...

Unet architectures are promising deep learning networks exploited to perform the automatic segmentation of bone CT images, in line with their ability to deal with pathological deformations and size-varying anatomies. However, bone degeneration, like the development of irregular osteophytes as well as mineral density alterations might interfere with this automated process and demand extensive manual refinement. The aim of this work is to implement an innovative Unet variant, the CEL-Unet, to improve the femur and tibia segmentation outcomes in osteoarthritic knee joints. In this network the decoding path is split into a region and contour-aware branch to increase the prediction reliability in such pathological conditions. The comparison between the segmentation results achieved with a standard Unet and its novel variant (CEL-Unet) was performed as follows: the Unet was trained with 5 different loss functions: Dice Loss, Focal Loss, Exponential Logarithmic Loss, Double Cross Entropy Loss and Distanced Cross Entropy loss. The CEL-Unet was instead trained with two loss functions, one for each of the network outputs, namely Mask and Edge, yielding the so-called Combined Edge Loss (CEL) function. A set of 259 knee CT scans was used to train the model and test segmentation performance. The CEL-Unet outperformed all other Unet-based models, reaching the highest Jaccard values of about 0.97 and 0.96 on femur and tibia, respectively. Clinical Relevance- With the increasing rate of Total Knee Arthoplasty deep learning-based methods can achieve fast accurate and automatic 3D segmentation of the knee joint bones to enhance new costumized pre-operative planning.