. Author manuscript; available in PMC: 2023 Nov 1.

Published in final edited form as: Med Image Anal. 2022 Sep 14;82:102615. doi: 10.1016/j.media.2022.102615

Table 2:

The architecture hyperparameters of the TransMorph models used in the ablation study. “Embed. Dimension” denotes the embedding dimension, C, in the very first stage (described in section 3.2); “Swin-T.” denotes Swin Transformer.

Model	Embed. Dimension	Swin-T. block numbers	Head numbers	Parameters (M)
`TransMorph`	96	{2, 2, 4, 2}	{4, 4, 8, 8}	46.77
`TransMorph`-tiny	6	{2, 2, 4, 2}	{4, 4, 8, 8}	0.24
`TransMorph`-small	48	{2, 2, 4, 2}	{4, 4, 4, 4}	11.76
`TransMorph`-large	128	{2, 2, 12, 2}	{4, 4, 8, 16}	108.34
`VoxelMorph`-huge	-	-	-	63.25