Workshop: Medical Imaging meets NeurIPS

PISTACHIO: Patch Importance Sampling To Accelerate CNNs via a Hash Index Optimizer

Zhaozhuo Xu · Anshumali Shrivastava


Convolution neural network(CNN) is the de-facto standard for medical imaging tasks. However, CNNs are known to be computation expensive. To take advantage of GPU's computing power, techniques such as Image Block to Column(im2col) are adopted at the cost of increasing memory usages. Such reasons prevent convolution-based architectures from training with large batch sizes and applying to high-definition and high-resolution input. In this work, we propose PISTACHIO: patch importance sampling to accelerate CNNs via a hash index optimizer. With efficient hashing-based sampling, we reduce the memory consumption of CNNs while preserving the final training accuracy.

Chat is not available.