piranha icon indicating copy to clipboard operation
piranha copied to clipboard

Questions on the communication size of AlexNet (on CIFAR 10) and LeNet (on MNIST) in Table 2 of the paper

Open SkyTu opened this issue 2 years ago • 1 comments

We noticed that the shape of data samples in CIFAR 10 is 32x32x3, while for MNIST is 28x28x1. Moreover, the number of parameters in AlexNet is much bigger than LeNet. Combining these two observations, we arrived at why, with the same protocol (e.g., Falcon), the communication size of LeNet training MNIST is 485.90 GB while the communication size of AlexNet training CIFAR 10 is 382.18GB. Is there any optimization used in AlexNet?

SkyTu avatar Nov 17 '22 01:11 SkyTu

Both AlexNet and VGG16 use avgpool instead of Maxpool, I guess that's the reason.

ElleryQu avatar Jun 03 '23 09:06 ElleryQu