notesum.ai

Published at November 21

SegBook: A Simple Baseline and Cookbook for Volumetric Medical Image Segmentation

eess.IV

cs.CV

Released Date: November 21, 2024

Authors: Jin Ye¹, Ying Chen², Yanjun Li³, Haoyu Wang¹, Zhongying Deng⁴, Ziyan Huang¹, Yanzhou Su¹, Chenglong Ma¹, Yuanfeng Ji⁵, Junjun He¹

Aff.: ¹Shanghai AI Laboratory; ²Shanghai AI Laboratory, Xiamen University; ³Shanghai AI Laboratory, East China Normal University; ⁴University of Cambridge; ⁵Stanford University

Arxiv: http://arxiv.org/abs/2411.14525v1

Method	PT	Params	TFLOPs	Dataset Scale			Average
Method	PT	Params	TFLOPs	S	M	L	Average
nnU-Net		$\sim$ 31M	$\sim$ 0.54	74.83	74.47	68.73	73.62
STU-Net-base		58M	0.51	73.96	74.84	70.05	73.66
STU-Net-base	✔	58M	0.51	76.17	76.59	72.81	75.77
$\Delta(base)$				2.21	1.75	2.76	2.11
STU-Net-large		440M	3.81	74.14	75.71	70.48	74.18
STU-Net-large	✔	440M	3.81	77.05	77.23	73.84	76.57
$\Delta(large)$				2.91	1.52	3.36	2.39
STU-Net-huge		1.4B	12.60	73.55	75.2	70.55	73.73
STU-Net-huge	✔	1.4B	12.60	76.87	77.14	74.21	76.53
$\Delta(huge)$				3.32	1.94	3.66	2.80