arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2005.11217cs.LGcs.CVeess.IVstat.ML
收藏

基于全局潜在混合的半监督医学图像分类

Semi-supervised Medical Image Classification with Global Latent Mixing

Rochester Institute of Technology(罗切斯特理工学院)

Prashnna Kumar Gyawali, Sandesh Ghimire, Pradeep Bajracharya, Zhiyuan Li, Linwei Wang

摘要:

通过深度学习进行的计算机辅助诊断依赖于大规模的注释数据集,当涉及专家知识时,这可能是昂贵的。半监督学习(SSL)通过利用未标记的数据来缓解这一挑战。一种有效的SSL方法是通过单个数据点周围的扰动来正则化神经函数的局部光滑性。在这项工作中,我们认为,通过填充数据点之间的空隙来正则化神经函数的全局光滑性可以进一步提高SSL。我们提出了一种新的SSL方法,训练神经网络的线性混合的标记和未标记的数据,在输入和潜在的空间,以正则化网络的不同部分。我们评估了两个不同的医学图像数据集的胸部疾病和皮肤病变的半监督分类模型,证明其改进的性能SSL与局部扰动和SSL与全球混合,但在输入空间。我们的代码可在https://github.com/Prasanna1991/LatentMixing上获得。

英文摘要:

Computer-aided diagnosis via deep learning relies on large-scale annotated data sets, which can be costly when involving expert knowledge. Semi-supervised learning (SSL) mitigates this challenge by leveraging unlabeled data. One effective SSL approach is to regularize the local smoothness of neural functions via perturbations around single data points. In this work, we argue that regularizing the global smoothness of neural functions by filling the void in between data points can further improve SSL. We present a novel SSL approach that trains the neural network on linear mixing of labeled and unlabeled data, at both the input and latent space in order to regularize different portions of the network. We evaluated the presented model on two distinct medical image data sets for semi-supervised classification of thoracic disease and skin lesion, demonstrating its improved performance over SSL with local perturbations and SSL with global mixing but at the input space only. Our code is available at https://github.com/Prasanna1991/LatentMixing.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑