Skip to main content

Table 1 A comparison between the proposed WDRIV-Net with other models

From: WDRIV-Net: a weighted ensemble transfer learning to improve automatic type stratification of lumbar intervertebral disc bulge, prolapse, and herniation

No

Model

Classifier

Accuracy (%)

Type

Precision

Recall

F1

AUC (%)

1

NASNetMobile [52]

Sigmoid

60.00

Single

0.60

0.60

0.60

60.0

Sigmoid

60.00

Comorbidity

0.60

0.60

0.60

60.0

2

Pre-trained ResNet152V2

Sigmoid

71.25

Single

1.00

0.42

0.60

71.25

Sigmoid

71.25

Comorbidity

0.63

1.00

0.78

71.25

3

DenseNet201 [42]

Sigmoid

73.75

Single

1.00

0.47

0.64

73.75

Sigmoid

73.75

Comorbidity

0.66

1.00

0.79

73.75

4

Pre-trained VGG16

Sigmoid

73.75

Single

0.77

0.68

0.72

73.75

Sigmoid

73.75

Comorbidity

0.71

0.80

0.75

73.75

5

MobileNetV2 [54]

Sigmoid

76.25

Single

0.92

0.57

0.71

76.25

Sigmoid

76.25

Comorbidity

0.69

0.95

0.80

76.25

6

ResNet50 [39]

Sigmoid

81.25

Single

0.78

0.88

0.82

81.25

Sigmoid

81.25

Comorbidity

0.86

0.75

0.80

81.25

7

Pre-trained

DenseNet169

Sigmoid

82.50

Single

0.82

0.82

0.82

82.50

Sigmoid

82.50

Comorbidity

0.82

0.82

0.82

82.50

8

NasNetLarge [52]

Sigmoid

82.50

Single

0.93

0.70

0.80

82.50

Sigmoid

82.50

Comorbidity

0.76

0.95

0.84

82.50

9

InceptionResNetV2 [23]

Sigmoid

85.00

Single

0.83

0.88

0.85

85.00

Sigmoid

85.00

Comorbidity

0.87

0.82

0.85

85.00

10

Pre-trained

ResNet101

Sigmoid

87.5

Single

0.86

0.90

0.88

87.5

Sigmoid

87.5

Comorbidity

0.89

0.85

0.87

87.5

11

VGG19 [55]

Sigmoid

88.75

Single

0.94

0.82

0.88

88.75

Sigmoid

88.75

Comorbidity

0.84

0.95

0.89

88.75

12

ResNet152 [39]

Sigmoid

90.00

Single

0.97

0.82

0.89

90.00

Sigmoid

90.00

Comorbidity

0.85

0.97

0.91

90.00

13

DenseNet121 [42]

Sigmoid

90.00

Single

0.97

0.82

0.89

90.00

Sigmoid

90.00

Comorbidity

0.85

0.97

0.91

90.00

14

Pre-trained

InceptionV3

Sigmoid

93.75

Single

0.97

0.90

0.94

93.75

Sigmoid

93.75

Comorbidity

0.91

0.97

0.94

93.75

15

Tavana et al. [16]

Sigmoid

88.75

Single

0.97

0.80

0.88

88.75

Sigmoid

88.75

Comorbidity

0.83

0.97

0.90

88.75

16

WDRIV-Net (proposed)

Sigmoid

96.25

Single

1.00

0.93

0.96

96.25

Sigmoid

96.25

Comorbidity

0.93

1.00

0.96

96.25