Center for Artificial Intelligence in Medicine & Imaging and Department of Urology, Stanford School of Medicine, Stanford, CA, 94305, USA; Department of Urology, Stanford School of Medicine, Stanford, CA, 94305, USA.
Department of Pathology, University of Muenster, Muenster, Germany.
Comput Biol Med. 2023 Mar;154:106594. doi: 10.1016/j.compbiomed.2023.106594. Epub 2023 Jan 26.
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.
最先进的 (SOTA) 卷积神经网络模型已广泛应用于医学成像,并应用于解决不同的临床问题。然而,此类模型的复杂性和规模在医学成像中可能并不合理,并且受到可用资源预算的限制。进一步增加分类任务的代表性特征图数量会降低模型的可解释性。目前的数据归一化实践是在模型开发之前固定的,并且不考虑数据域的规范。鉴于这些问题,目前的工作提出了一种新的可扩展模型系列,称为 PlexusNet;该模型的块架构和模型缩放由网络的深度、宽度和分支来调节。高效的计算成本概述了 PlexusNet 缩放和设计的维度。PlexusNet 包括一种新的可学习的数据归一化算法,以实现更好的数据泛化。我们应用了一种简单而有效的神经架构搜索来设计适用于五个临床分类问题的 PlexusNet,其性能不逊于 SOTA 模型 ResNet-18 和 EfficientNet B0/1。它还具有比性能相当的最小 SOTA 模型低十倍的参数容量和代表性特征图。代表性特征的可视化揭示了基于 PlexusNet 生成的潜在特征与类别相关的可区分聚类。该软件包和源代码位于 https://github.com/oeminaga/PlexusNet.git。