PlexusNet: A neural network architectural concept for medical image classification

Okyaz Eminaga, Mahmoud Abbas, Jeanne Shen, Mark Laurie, James D. Brooks, Joseph C. Liao, Daniel L. Rubin

Research output: Contribution to journalArticlepeer-review

5 Scopus citations

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.

Original languageEnglish (US)
Article number106594
JournalComputers in Biology and Medicine
Volume154
DOIs
StatePublished - Mar 2023
Externally publishedYes

Keywords

  • Compact models
  • Computer vision
  • Convolutional neural networks
  • Deep learning
  • Diagnostics
  • PlexusNet

ASJC Scopus subject areas

  • Health Informatics
  • Computer Science Applications

Fingerprint

Dive into the research topics of 'PlexusNet: A neural network architectural concept for medical image classification'. Together they form a unique fingerprint.

Cite this