BMC Medical Informatics and Decision Making

Table 5 Confusion matrices of VGG, ResNet, and Efficient model

From: A reliable diabetic retinopathy grading via transfer learning and ensemble learning with quadratic weighted kappa metric

Model Name	VGG-16						VGG-19					ResNet-18					ResNet-34					ResNet-50
Real Result	Predicted Result
		0	1	2	3	4	0	1	2	3	4	0	1	2	3	4	0	1	2	3	4	0	1	2	3	4
	0	3638	0	0	0	0	3887	0	0	0	0	3825	0	0	0	0	3830	0	0	0	0	3864	0	0	0	0
	1	363	0	0	0	0	376	0	0	0	0	371	0	0	0	0	391	0	0	0	0	365	0	0	0	0
	2	763	0	0	0	0	793	0	0	0	0	806	0	0	0	0	804	0	0	0	0	806	0	0	0	0
	3	131	0	0	0	0	131	0	0	0	0	152	0	0	0	0	138	0	0	0	0	116	0	0	0	0
	4	113	0	0	0	0	113	0	0	0	0	114	0	0	0	0	105	0	0	0	0	117	0	0	0	0
Model Name	ResNet-101						ResNet-152					EfficientNet-B0					EfficientNet-B1					EfficientNet-B2
Real Result	Predicted Result
		0	1	2	3	4	0	1	2	3	4	0	1	2	3	4	0	1	2	3	4	0	1	2	3	4
	0	3848	0	0	0	0	3775	0	0	0	0	3787	0	105	2	5	3833	0	68	0	0	3761	0	120	0	0
	1	383	0	0	0	0	360	0	0	0	0	321	0	20	0	0	357	0	17	0	0	346	0	29	0	0
	2	783	0	0	0	0	814	0	0	0	0	532	0	228	25	1	501	0	273	0	2	487	0	307	0	0
	3	140	0	0	0	0	128	0	0	0	0	27	0	67	39	2	30	0	89	0	3	24	0	86	0	0
	4	103	0	0	0	0	90	0	0	0	0	28	0	47	17	15	34	0	41	0	18	17	0	91	0	0
Model Name	EfficientNet-B3_30						EfficientNet-B3_60					EfficientNet-B4					EfficientNet-B5					EfficientNet-B6
Real Result	Predicted Result
		0	1	2	3	4	0	1	2	3	4	0	1	2	3	4	0	1	2	3	4	0	1	2	3	4
	0	3868	3	20	0	0	3772	8	65	0	0	3851	0	38	0	2	3726	0	83	1	0	3809	0	59	0	2
	1	326	23	34	0	0	244	31	128	0	0	348	0	3	0	0	363	0	11	0	0	335	0	13	0	0
	2	247	34	501	0	0	114	9	624	5	0	605	0	194	5	8	587	0	234	21	0	515	0	269	1	10
	3	10	0	69	38	0	3	0	77	46	13	42	0	71	46	2	36	0	70	33	0	26	0	87	5	18
	4	16	0	29	15	37	9	0	14	6	98	44	0	58	6	16	19	0	65	21	0	33	0	66	1	21

Back to article page

ISSN: 1472-6947

Contact us

General enquiries: journalsubmissions@springernature.com