PocketNet: A Smaller Neural Network For Medical Image Analysis

Date
2023-04-24
Journal Title
Journal ISSN
Volume Title
Publisher
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.

Description
Degree
Master of Arts
Type
Thesis
Keywords
Neural network, segmentation, pattern recognition and classification
Citation

Celaya, Adrian. "PocketNet: A Smaller Neural Network For Medical Image Analysis." (2023) Master’s Thesis, Rice University. https://hdl.handle.net/1911/114903.

Has part(s)
Forms part of
Published Version
Rights
Copyright is held by the author, unless otherwise indicated. Permission to reuse, publish, or reproduce the work beyond the bounds of fair use or other exemptions to copyright law must be obtained from the copyright holder.
Link to license
Citable link to this page