Skip to main content

Table 2 Top 10 accuracy parameters, models and results optimized by NNI

From: Deep transfer learning to quantify pleural effusion severity in chest X-rays

 

Data Sample

Loss

LR

Optimizer

Model

Accuracy (%)

1

None

Focal

0.005

Adagrad

DenseNet121

73.07

2

None

CrossEntropy

0.005

Adagrad

DenseNet121

72.56

3

None

CrossEntropy

0.001

Adamax

resnet18

72.19

4

Over sample

Focal

0.005

Adagrad

DenseNet121

72.04

5

None

CrossEntropy

0.005

Adagrad

resnet18

71.98

6

None

CrossEntropy

0.005

SGD

DenseNet121

71.94

7

None

Focal

0.005

Adagrad

resnet18

71.83

8

None

CrossEntropy

0.001

Adagrad

resnet18

71.51

9

None

CrossEntropy

0.005

Adagrad

resnet50

71.34

10

Over sample

CrossEntropy

0.001

Adagrad

resnet50

71.30