Table 3.
Dice | Balanced accuracy | |
---|---|---|
CNN | 0.864 (0.052) | 0.942 (0.028) |
DNN | 0.831 (0.122) | 0.922 (0.018) |
SVM | 0.818 (0.115) | 0.895 (0.071) |
Mean (SD) Dice index and balanced accuracy of the three machine learning models for DWMA detection. The proposed CNN model outperformed compared DNN and SVM in the 10-fold cross-validation with 50 subjects. CNN, convolutional neural networks; DNN, deep neural networks; SVM, support vector machine. The boldface denotes the best performance for individual metrics in the validation experiment.