CNN‐based fully automatic wrist cartilage volume quantification in MR images: A comparative analysis between different CNN architectures

Abstract

PurposeAutomatic measurement of wrist cartilage volume in MR images.MethodsWe assessed the performance of four manually optimized variants of the U‐Net architecture, nnU‐Net and Mask R‐CNN frameworks for the segmentation of wrist cartilage. The results were compared to those from a patch‐based convolutional neural network (CNN) we previously designed. The segmentation quality was assessed on the basis of a comparative analysis with manual segmentation. The best networks were compared using a cross‐validation approach on a dataset of 33 3D VIBE images of mostly healthy volunteers. Influence of some image parameters on the segmentation reproducibility was assessed.ResultsThe U‐Net‐based networks outperformed the patch‐based CNN in terms of segmentation homogeneity and quality, while Mask R‐CNN did not show an acceptable performance. The median 3D DSC value computed with the U‐Net_AL (0.817) was significantly larger than DSC values computed with the other networks. In addition, the U‐Net_AL provided the lowest mean volume error (17%) and the highest Pearson correlation coefficient (0.765) with respect to the ground truth values. Of interest, the reproducibility computed using U‐Net_AL was larger than the reproducibility of the manual segmentation. Moreover, the results indicate that the MRI‐based wrist cartilage volume is strongly affected by the image resolution.ConclusionsU‐Net CNN with attention layers provided the best wrist cartilage segmentation performance. In order to be used in clinical conditions, the trained network can be fine‐tuned on a dataset representing a group of specific patients. The error of cartilage volume measurement should be assessed independently using a non‐MRI method.