arXiv:2411.01564cs.CV2024-11被引 3

提出可解释的医疗影像解析胶囊网络,提升诊断准确率与可信度。

ParseCaps: An Interpretable Parsing Capsule Network for Medical Image Diagnosis

  • 采用稀疏轴向注意力路由与解析卷积胶囊层构建树状结构
  • 在多个数据集上超越传统胶囊网络,分类精度更高且冗余更低
  • 无需概念标签也能生成可理解的解释,适合临床医生使用

深度学习在医学图像分类中表现优异,但因可解释性差限制了临床应用。胶囊网络虽能编码层级关系与空间特征,但传统模型结构浅,深层变体缺乏层级架构,削弱了可解释性。本文提出新型胶囊网络ParseCaps,通过稀疏轴向注意力路由和解析卷积胶囊层,构建类解析树结构,增强深度与可解释性。首先,稀疏轴向注意力路由优化子父胶囊间连接,并强化父胶囊实例化参数的权重分布;其次,解析卷积胶囊层生成符合解析树的胶囊预测;最后,基于不依赖概念真值的损失设计,使全局胶囊各维度对应可理解概念,提升医生对分类结果的信任。在CE-MRI、PH²、Derm7pt数据集上的实验表明,ParseCaps不仅在分类精度、冗余减少和鲁棒性方面优于其他胶囊网络变体,且无论是否存在概念标签,均能提供可解释的推理过程。

原文摘要 · Abstract (English)

Deep learning has excelled in medical image classification, but its clinical application is limited by poor interpretability. Capsule networks, known for encoding hierarchical relationships and spatial features, show potential in addressing this issue. Nevertheless, traditional capsule networks often underperform due to their shallow structures, and deeper variants lack hierarchical architectures, thereby compromising interpretability. This paper introduces a novel capsule network, ParseCaps, which utilizes the sparse axial attention routing and parse convolutional capsule layer to form a parse-tree-like structure, enhancing both depth and interpretability. Firstly, sparse axial attention routing optimizes connections between child and parent capsules, as well as emphasizes the weight distribution across instantiation parameters of parent capsules. Secondly, the parse convolutional capsule layer generates capsule predictions aligning with the parse tree. Finally, based on the loss design that is effective whether concept ground truth exists or not, ParseCaps advances interpretability by associating each dimension of the global capsule with a comprehensible concept, thereby facilitating clinician trust and understanding of the model's classification results. Experimental results on CE-MRI, PH$^2$, and Derm7pt datasets show that ParseCaps not only outperforms other capsule network variants in classification accuracy, redundancy reduction and robustness, but also provides interpretable explanations, regardless of the availability of concept labels.

医疗影像可解释性胶囊网络解析树

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。