We gratefully acknowledge support from
the Simons Foundation and member institutions.
Full-text links:

Download:

Current browse context:

eess.IV

Change to browse by:

References & Citations

Bookmark

(what is this?)
CiteULike logo BibSonomy logo Mendeley logo del.icio.us logo Digg logo Reddit logo

Electrical Engineering and Systems Science > Image and Video Processing

Title: PocketNet: A Smaller Neural Network for Medical Image Analysis

Abstract: Medical imaging deep learning models are often large and complex, requiring specialized hardware to train and evaluate these models. To address such issues, we propose the PocketNet paradigm to reduce the size of deep learning models by throttling the growth of the number of channels in convolutional neural networks. We demonstrate that, for a range of segmentation and classification tasks, PocketNet architectures produce results comparable to that of conventional neural networks while reducing the number of parameters by multiple orders of magnitude, using up to 90% less GPU memory, and speeding up training times by up to 40%, thereby allowing such models to be trained and deployed in resource-constrained settings.
Subjects: Image and Video Processing (eess.IV); Computer Vision and Pattern Recognition (cs.CV); Machine Learning (cs.LG)
DOI: 10.1109/TMI.2022.3224873
Cite as: arXiv:2104.10745 [eess.IV]
  (or arXiv:2104.10745v4 [eess.IV] for this version)

Submission history

From: Adrian Celaya [view email]
[v1] Wed, 21 Apr 2021 20:10:30 GMT (1231kb,D)
[v2] Thu, 24 Jun 2021 03:29:46 GMT (3081kb,D)
[v3] Wed, 18 May 2022 14:34:55 GMT (4577kb,D)
[v4] Sun, 18 Sep 2022 17:56:01 GMT (4732kb,D)

Link back to: arXiv, form interface, contact.