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

用于多标签疾病诊断与解释的跨图像与图像内原型学习

Cross- and Intra-image Prototypical Learning for Multi-label Disease Diagnosis and Interpretation

Australian Institute for Machine Learning(澳大利亚机器学习研究所);University of Adelaide(阿德莱德大学);Cornell University(康奈尔大学);St Vincent’s Hospital Melbourne(墨尔本圣文森特医院);Centre for Vision, Speech and Signal Processing(视觉、语音与信号处理中心);University of Surrey(萨里大学)

Chong Wang, Fengbei Liu, Yuanhong Chen, Helen Frazer, Gustavo Carneiro

AI总结:

本文提出CIPL框架,通过跨图像语义解耦多疾病并利用双层对齐正则化增强图像内信息,实现了准确的多标签疾病诊断与解释,在多个基准上达到SOTA并提升了弱监督定位性能。

摘要:

原型学习的最新进展展现出显著潜力,能通过将激活图和预测结果与特定类别的训练原型关联起来,提供有用的决策解释。这种原型学习已在多种单标签疾病中得到充分研究,但对于密切相关且更具挑战性的多标签诊断(即图像内常并发多种疾病),由于多种疾病的纠缠,现有的原型学习模型难以获得有意义的激活图和有效的类别原型。本文提出了一种新颖的跨图像与图像内原型学习(CIPL)框架,用于从医学图像中进行准确的多标签疾病诊断与解释。CIPL利用常见的跨图像语义在学习原型时解耦多种疾病,从而全面理解复杂的病理病变。此外,我们提出了一种新的基于双层对齐的正则化策略,该策略有效利用一致的图像内信息来增强解释鲁棒性和预测性能。大量实验表明,我们的CIPL在两个公开的多标签疾病诊断基准(胸部X光和眼底图像)中获得了最先进的分类准确率。定量可解释性结果表明,在弱监督胸部疾病定位方面,CIPL也优于其他领先的基于显著性图和原型的解释方法。

英文摘要:

Recent advances in prototypical learning have shown remarkable potential to provide useful decision interpretations associating activation maps and predictions with class-specific training prototypes. Such prototypical learning has been well-studied for various single-label diseases, but for quite relevant and more challenging multi-label diagnosis, where multiple diseases are often concurrent within an image, existing prototypical learning models struggle to obtain meaningful activation maps and effective class prototypes due to the entanglement of the multiple diseases. In this paper, we present a novel Cross- and Intra-image Prototypical Learning (CIPL) framework, for accurate multi-label disease diagnosis and interpretation from medical images. CIPL takes advantage of common cross-image semantics to disentangle the multiple diseases when learning the prototypes, allowing a comprehensive understanding of complicated pathological lesions. Furthermore, we propose a new two-level alignment-based regularisation strategy that effectively leverages consistent intra-image information to enhance interpretation robustness and predictive performance. Extensive experiments show that our CIPL attains the state-of-the-art (SOTA) classification accuracy in two public multi-label benchmarks of disease diagnosis: thoracic radiography and fundus images. Quantitative interpretability results show that CIPL also has superiority in weakly-supervised thoracic disease localisation over other leading saliency- and prototype-based explanation methods.

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

↑