PlexusNet: A neural network architectural concept for medical image classification. (March 2023)

• Record Type: Journal Article
• Authors: Eminaga, Okyaz; Abbas, Mahmoud; Shen, Jeanne; Laurie, Mark; Brooks, James D.; Liao, Joseph C.; Rubin, Daniel L.
• Is Part Of: Computers in biology and medicine. Volume 154(2023)
• Publication Date: 2023-03
• Subjects: Deep learning -- Computer vision -- Diagnostics -- Convolutional neural networks -- PlexusNet -- Compact models; Medicine -- Data processing -- Periodicals; Biology -- Data processing -- Periodicals; 610.285
• DOI: 10.1016/j.compbiomed.2023.106594
• Languages: English
• ISSNs: 0010-4825

Abstract

Abstract: State-of-the-art (SOTA) convolutional neural network models have been widely adapted in medical imaging and applied to address different clinical problems. However, the complexity and scale of such models may not be justified in medical imaging and subject to the available resource budget. Further increasing the number of representative feature maps for the classification task decreases the model explainability. The current data normalization practice is fixed prior to model development and discounting the specification of the data domain. Acknowledging these issues, the current work proposed a new scalable model family called PlexusNet; the block architecture and model scaling by the network's depth, width, and branch regulate PlexusNet's architecture. The efficient computation costs outlined the dimensions of PlexusNet scaling and design. PlexusNet includes a new learnable data normalization algorithm for better data generalization. We applied a simple yet effective neural architecture search to design PlexusNet tailored to five clinical classification problems that achieve a performance noninferior to the SOTA models ResNet-18 and EfficientNet B0/1. It also does so with lower parameter capacity and representative feature maps in ten-fold ranges than the smallest SOTA models with comparable performance. The visualization of representative features revealed distinguishable clusters associated with categories based on latent features generated by PlexusNet. The package and source code are at https://github.com/oeminaga/PlexusNet.git . Graphical abstract: Image 1 Highlights: Trade-off between model complexity and performance is poor in medical imaging. Covariate shift and overparameterization are commonly neglected in medical imaging. PlexusNet provides an optimized trade-off between model complexity and performance. PlexusNet successfully mitigates covariate shift and overparameterization issues. The image domain or modality can impact the optimal model architecture definition.