notesum.ai

Published at November 18

TP-UNet: Temporal Prompt Guided UNet for Medical Image Segmentation

cs.CV

cs.AI

Released Date: November 18, 2024

Authors: Ranmin Wang¹, Limin Zhuang¹, Hongkun Chen², Boyan Xu², Ruichu Cai²

Aff.: ¹School of Management, Guangdong University of Technology, Guangzhou, China; ²School of Computer Science, Guangdong University of Technology, Guangzhou, China

Arxiv: http://arxiv.org/abs/2411.11305v2

Model	Backbone	Large Intestine		Small Intestine		Stomach		Average
Model	Backbone	Dice $\uparrow$	Jacc $\uparrow$	Dice $\uparrow$	Jacc $\uparrow$	Dice $\uparrow$	Jacc $\uparrow$	Dice $\uparrow$	Jacc $\uparrow$
UNet [14]	VGG16 [15]	0.8709	0.8329	0.8546	0.8187	0.9210	0.9001	0.8822	0.8506
UNet [14]	Resnet50 [16]	0.8621	0.8211	0.8487	0.8122	0.9106	0.8880	0.8738	0.8405
UNet++ [17]	VGG16 [15]	0.8825	0.8463	0.8738	0.8394	0.9394	0.9201	0.8986	0.8686
UNet++ [17]	Resnet50 [16]	0.8613	0.8223	0.8537	0.8172	0.9178	0.8970	0.8776	0.8455
Attention UNet [18]	VGG16 [15]	0.8925	0.8580	0.8857	0.8519	0.9472	0.9292	0.9085	0.8797
scSE UNet [19]	VGG16 [15]	0.8909	0.8561	0.8750	0.8398	0.9491	0.9312	0.9050	0.8757
Trans UNet [20]	Vision Transformer [21]	0.8983	0.8597	0.8847	0.8543	0.9356	0.9185	0.9062	0.8775
Swin UNet [22]	Swin Transformer [23]	0.9034	0.8704	0.8908	0.8617	0.9467	0.9279	0.9136	0.8819
TP-UNet _Electra	VGG16 [15]	0.9170	0.8781	0.9078	0.8704	0.9551	0.9343	0.9266	0.8943
TP-UNet _CLIP	VGG16 [15]	$\boldsymbol{0.9190}$	$\boldsymbol{0.8810}$	$\boldsymbol{0.9102}$	$\boldsymbol{0.8727}$	$\boldsymbol{0.9566}$	$\boldsymbol{0.9358}$	$\boldsymbol{0.9286}$	$\boldsymbol{0.8965}$