arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2201.10675cs.CVcs.AIeess.IVeess.SP
收藏

用于半监督乳腺肿块分类的虚拟对抗训练

Virtual Adversarial Training for Semi-supervised Breast Mass Classification

University of Oklahoma(俄克拉荷马大学);ShanghaiTech University(上海科技大学);University of Oklahoma Health Sciences Center(俄克拉荷马大学健康科学中心)

Xuxin Chen, Ximin Wang, Ke Zhang, Kar-Ming Fung, Theresa C. Thai, Kathleen Moore, Robert S. Mannel, Hong Liu, Bin Zheng, Yuchen Qiu

摘要:

本研究的目的是开发一种新的计算机辅助诊断(CAD)计划乳腺摄影乳腺肿块分类使用半监督学习。尽管监督式深度学习在各种医学图像分析任务中取得了巨大的成功,但其成功依赖于大量高质量的注释,这在实践中可能具有挑战性。为了克服这一限制,我们提出采用半监督方法,即,虚拟对抗训练(VAT),利用和学习未标记数据中的有用信息,以更好地分类乳腺肿块。因此,我们基于VAT的模型有两种类型的损失,即监督损失和虚拟对抗损失。前者的损失作为监督分类,而后者的损失旨在提高模型对虚拟对抗扰动的鲁棒性,从而提高模型的泛化能力。为了评估我们基于VAT的CAD方案的性能,我们回顾性地收集了总共1024张乳腺肿块图像,其中良性肿块和恶性肿块的数量相等。在这项研究中使用了一个大的CNN和一个小的CNN,两者都在有和没有对抗性损失的情况下进行了训练。当标记率为40%和80%时,基于VAT的CNN分别提供了最高的分类准确率0.740和0.760。实验结果表明,基于增值税的CAD方案可以有效地利用有意义的知识,从未标记的数据,更好地分类乳腺摄影乳腺肿块图像。

英文摘要:

This study aims to develop a novel computer-aided diagnosis (CAD) scheme for mammographic breast mass classification using semi-supervised learning. Although supervised deep learning has achieved huge success across various medical image analysis tasks, its success relies on large amounts of high-quality annotations, which can be challenging to acquire in practice. To overcome this limitation, we propose employing a semi-supervised method, i.e., virtual adversarial training (VAT), to leverage and learn useful information underlying in unlabeled data for better classification of breast masses. Accordingly, our VAT-based models have two types of losses, namely supervised and virtual adversarial losses. The former loss acts as in supervised classification, while the latter loss aims at enhancing model robustness against virtual adversarial perturbation, thus improving model generalizability. To evaluate the performance of our VAT-based CAD scheme, we retrospectively assembled a total of 1024 breast mass images, with equal number of benign and malignant masses. A large CNN and a small CNN were used in this investigation, and both were trained with and without the adversarial loss. When the labeled ratios were 40% and 80%, VAT-based CNNs delivered the highest classification accuracy of 0.740 and 0.760, respectively. The experimental results suggest that the VAT-based CAD scheme can effectively utilize meaningful knowledge from unlabeled data to better classify mammographic breast mass images.

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

↑