论文标题

在生物医学环境中使用紧凑网络的半监督学习用于图像分类

Semi-Supervised Learning for Image Classification using Compact Networks in the BioMedical Context

论文作者

Inés, Adrián, Díaz-Pinto, Andrés, Domínguez, César, Heras, Jónathan, Mata, Eloy, Pascual, Vico

论文摘要

移动和嵌入深层卷积神经模型的边缘应用的开发有可能改变生物医学。但是,大多数深度学习模型都需要智能手机或边缘设备中无法使用的计算资源;可以通过紧凑的模型面临的问题。这种模型的问题在于,它们通常至少不如更大的模型准确。在这项工作中,我们研究了如何通过半监督学习技术来解决这种限制。我们进行了几项统计分析,以比较使用半监督学习方法训练深层紧凑型体系结构的性能,以在生物医学背景下解决图像分类任务。特别是,我们探索了三个紧凑型网络的家庭,以及两个半监督的学习技术,用于10项生物医学任务。通过将半监督学习方法与紧凑的网络相结合,可以获得与标准尺寸网络相似的性能。通常,在将数据蒸馏与混合网和与Resnet-18结合使用时,获得了最佳结果。同样,通常,NAS网络比手动设计的网络和量化网络获得更好的结果。本文介绍的工作表明,将半监督方法应用于紧凑网络的好处;这使我们能够创建紧凑型型号,这些模型不仅与标准尺寸型号一样准确,而且更快,更轻。最后,我们开发了一个库,该库通过半监督的学习方法简化了紧凑型模型的构建。

The development of mobile and on the edge applications that embed deep convolutional neural models has the potential to revolutionise biomedicine. However, most deep learning models require computational resources that are not available in smartphones or edge devices; an issue that can be faced by means of compact models. The problem with such models is that they are, at least usually, less accurate than bigger models. In this work, we study how this limitation can be addressed with the application of semi-supervised learning techniques. We conduct several statistical analyses to compare performance of deep compact architectures when trained using semi-supervised learning methods for tackling image classification tasks in the biomedical context. In particular, we explore three families of compact networks, and two families of semi-supervised learning techniques for 10 biomedical tasks. By combining semi-supervised learning methods with compact networks, it is possible to obtain a similar performance to standard size networks. In general, the best results are obtained when combining data distillation with MixNet, and plain distillation with ResNet-18. Also, in general, NAS networks obtain better results than manually designed networks and quantized networks. The work presented in this paper shows the benefits of apply semi-supervised methods to compact networks; this allow us to create compact models that are not only as accurate as standard size models, but also faster and lighter. Finally, we have developed a library that simplifies the construction of compact models using semi-supervised learning methods.

扫码加入交流群

加入微信交流群

微信交流群二维码

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