arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2407.06566cs.CV

解决诊断成像中数据稀缺问题的鲁棒且可解释的框架

Robust and Explainable Framework to Address Data Scarcity in Diagnostic Imaging

Zehui Zhao, Laith Alzubaidi, Jinglan Zhang, Ye Duan, Usman Naseem, Yuantong Gu

AI总结:

提出ETSEF框架,结合迁移学习、自监督学习与集成学习,利用数据增强和可解释AI技术,在数据稀缺的医疗成像任务中显著提升诊断准确率与鲁棒性。

摘要:

深度学习显著推进了自动医疗诊断并释放了人力资源以减轻临床压力,然而该领域持续存在的数据稀缺挑战阻碍了其进一步的改进与应用。为了解决这一差距,我们引入了一种名为“基于高效迁移与自监督学习的集成框架”(ETSEF)的新型集成框架。ETSEF利用来自多个预训练深度学习模型的特征,从有限数量的数据样本中高效学习强大的表示。据我们所知,ETSEF是首个将两种预训练方法(迁移学习和自监督学习)与集成学习方法相结合的策略。还部署了各种数据增强技术,包括数据增强、特征融合、特征选择和决策融合,以最大化ETSEF模型的效率和鲁棒性。测试了五项独立的医学成像任务,包括内窥镜、乳腺癌、猴痘、脑肿瘤和青光眼检测,以证明ETSEF的有效性和鲁棒性。面对有限的样本数量和具有挑战性的医学任务,ETSEF证明了其有效性,与强大的集成基线模型相比,诊断准确率提高了10%至13.3%,与已发表的最先进方法相比提高了高达14.4%。此外,我们通过各种视觉可解释人工智能技术强调了ETSEF方法的鲁棒性和可信度,包括Grad-CAM、SHAP和t-SNE。与那些大规模深度学习模型相比,ETSEF可以灵活部署,并在具有挑战性的医学成像任务中保持卓越的性能,显示出应用于更多缺乏训练数据领域的潜力。

英文摘要:

Deep learning has significantly advanced automatic medical diagnostics and released the occupation of human resources to reduce clinical pressure, yet the persistent challenge of data scarcity in this area hampers its further improvements and applications. To address this gap, we introduce a novel ensemble framework called `Efficient Transfer and Self-supervised Learning based Ensemble Framework' (ETSEF). ETSEF leverages features from multiple pre-trained deep learning models to efficiently learn powerful representations from a limited number of data samples. To the best of our knowledge, ETSEF is the first strategy that combines two pre-training methodologies (Transfer Learning and Self-supervised Learning) with ensemble learning approaches. Various data enhancement techniques, including data augmentation, feature fusion, feature selection, and decision fusion, have also been deployed to maximise the efficiency and robustness of the ETSEF model. Five independent medical imaging tasks, including endoscopy, breast cancer, monkeypox, brain tumour, and glaucoma detection, were tested to demonstrate ETSEF's effectiveness and robustness. Facing limited sample numbers and challenging medical tasks, ETSEF has proved its effectiveness by improving diagnostics accuracies from 10\% to 13.3\% when compared to strong ensemble baseline models and up to 14.4\% improvements compared with published state-of-the-art methods. Moreover, we emphasise the robustness and trustworthiness of the ETSEF method through various vision-explainable artificial intelligence techniques, including Grad-CAM, SHAP, and t-SNE. Compared to those large-scale deep learning models, ETSEF can be deployed flexibly and maintain superior performance for challenging medical imaging tasks, showing the potential to be applied to more areas that lack training data

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

↑