论文标题

医学图像分类的新颖的对抗学习策略

A novel adversarial learning strategy for medical image classification

论文作者

Fan, Zong, Zhang, Xiaohui, Gasienica, Jacob A., Potts, Jennifer, Ruan, Su, Thorstad, Wade, Gay, Hiram, Song, Pengfei, Wang, Xiaowei, Li, Hua

论文摘要

深度学习(DL)技术已被广泛用于医学图像分类。大多数基于DL的分类网络通常是通过层次结构化的,并通过最小化网络末尾测量的单个损耗函数进行了优化。但是,这种单一的损失设计可能会导致优化一种特定的感兴趣价值,但无法利用中间层的信息特征,这些特征可能会受益于分类性能并降低过度拟合的风险。最近,辅助卷积神经网络(AUXCNNS)已在传统分类网络之上采用,以促进中间层的培训,以提高分类性能和鲁棒性。在这项研究中,我们提出了一种基于对抗性学习的AUXCNN,以支持对医学图像分类的深神经网络的培训。我们的AUXCNN分类框架采用了两项主要创新。首先,提出的AUXCNN体系结构包括图像发生器和图像歧视器,用于为医学图像分类提取更有信息的图像特征,这是由生成对抗网络(GAN)的概念及其在近似目标数据分布方面的令人印象深刻的能力。其次,混合损耗函数旨在通过合并分类网络和AUXCNN的不同目标来指导模型训练以减少过度拟合。全面的实验研究证明了所提出的模型的出色分类性能。研究了与网络相关因素对分类性能的影响。

Deep learning (DL) techniques have been extensively utilized for medical image classification. Most DL-based classification networks are generally structured hierarchically and optimized through the minimization of a single loss function measured at the end of the networks. However, such a single loss design could potentially lead to optimization of one specific value of interest but fail to leverage informative features from intermediate layers that might benefit classification performance and reduce the risk of overfitting. Recently, auxiliary convolutional neural networks (AuxCNNs) have been employed on top of traditional classification networks to facilitate the training of intermediate layers to improve classification performance and robustness. In this study, we proposed an adversarial learning-based AuxCNN to support the training of deep neural networks for medical image classification. Two main innovations were adopted in our AuxCNN classification framework. First, the proposed AuxCNN architecture includes an image generator and an image discriminator for extracting more informative image features for medical image classification, motivated by the concept of generative adversarial network (GAN) and its impressive ability in approximating target data distribution. Second, a hybrid loss function is designed to guide the model training by incorporating different objectives of the classification network and AuxCNN to reduce overfitting. Comprehensive experimental studies demonstrated the superior classification performance of the proposed model. The effect of the network-related factors on classification performance was investigated.

扫码加入交流群

加入微信交流群

微信交流群二维码

扫码加入学术交流群,获取更多资源