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

一种用于脑肿瘤分割的弱监督且全局可解释的学习框架

A Weakly Supervised and Globally Explainable Learning Framework for Brain Tumor Segmentation

Shenzhen Institute of Advanced Technology, Chinese Academy of Sciences(中国科学院深圳先进技术研究院);University of Chinese Academy of Sciences(中国科学院大学)

Ruitao Xie, Limai Jiang, Xiaoxi He, Yi Pan, Yunpeng Cai

AI总结:

本文提出一种弱监督反事实生成框架,通过分离类别相关特征并在拓扑流形中生成正常对照样本,在无需像素级标注的情况下实现高性能且全局可解释的脑肿瘤分割。

摘要:

基于机器的脑肿瘤分割可以帮助医生做出更好的诊断。然而,脑肿瘤的复杂结构和昂贵的像素级标注给自动肿瘤分割带来了挑战。在本文中,我们提出了一种反事实生成框架,该框架不仅无需像素级标注即可实现出色的脑肿瘤分割性能,而且还提供可解释性。我们的框架有效地将样本的类别相关特征与类别无关特征分离开来,并通过嵌入不同的类别相关特征生成保留身份特征同时改变类别属性的新样本。我们对提取的类别相关特征进行拓扑数据分析,获得全局可解释的流形,对于每个待分割的异常样本,可以在流形内设计的基于规则的路径引导下有效生成有意义的正常样本,通过比较来识别肿瘤区域。我们在两个数据集上评估了我们提出的方法,证明了脑肿瘤分割的优越性能。代码可在 https://github.com/xrt11/tumor-segmentation 获取。

英文摘要:

Machine-based brain tumor segmentation can help doctors make better diagnoses. However, the complex structure of brain tumors and expensive pixel-level annotations present challenges for automatic tumor segmentation. In this paper, we propose a counterfactual generation framework that not only achieves exceptional brain tumor segmentation performance without the need for pixel-level annotations, but also provides explainability. Our framework effectively separates class-related features from class-unrelated features of the samples, and generate new samples that preserve identity features while altering class attributes by embedding different class-related features. We perform topological data analysis on the extracted class-related features and obtain a globally explainable manifold, and for each abnormal sample to be segmented, a meaningful normal sample could be effectively generated with the guidance of the rule-based paths designed within the manifold for comparison for identifying the tumor regions. We evaluate our proposed method on two datasets, which demonstrates superior performance of brain tumor segmentation. The code is available at https://github.com/xrt11/tumor-segmentation.

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

↑