Statistical Learning
Contrastive learning of global and local features for medical image segmentation with limited annotations
For Prostate, although we have 48 volumes, labels were provided only for a subset of them, so the number of volumes for each set were adjusted accordingly. We used the ACDC dataset for these experiments. Here, we studied the effect of the batch size used during the encoder pre-training. The results are presented in Table 2. We observed that for medical images, higher batch sizes do not improve the results any further, rather the performance deteriorated for the batch size of 450.