A Weakly Supervised and Globally Explainable Learning Framework for Brain 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.
Code (1)
Tasks
Brain Tumor SegmentationcounterfactualSegmentationTopological Data AnalysisTumor SegmentationSimilar Papers 제목 키워드 기반
Weakly-supervised segmentation using inherently-explainable classification models and their application to brain tumour classification
Deep learning models have shown their potential for several applications. However, most of the models are opaque and difficult to trust due to their complex reasoning - commonly known as the black-box problem. Some field…
ClassificationDecision MakingSegmentationTumour Classification+1Deep Superpixel Generation and Clustering for Weakly Supervised Segmentation of Brain Tumors in MR Images
Training machine learning models to segment tumors and other anomalies in medical images is an important step for developing diagnostic tools but generally requires manually annotated ground truth segmentations, which ne…
Binary ClassificationBrain Tumor SegmentationClusteringDiagnostic+4Weakly Supervised Learning of Cortical Surface Reconstruction from Segmentations
Existing learning-based cortical surface reconstruction approaches heavily rely on the supervision of pseudo ground truth (pGT) cortical surfaces for training. Such pGT surfaces are generated by traditional neuroimage pr…
Surface ReconstructionWeakly-supervised LearningTumorXAI: Self-Supervised Deep Learning Framework for Explainable Brain MRI Tumor Classification
Classifying brain tumors using magnetic resonance imaging (MRI) is crucial for early diagnosis and treatment; however, tumor heterogeneity and a dearth of annotated datasets restrict the use of supervised deep learning a…
Brain Tumor ClassificationSelf-Supervised LearningWeakly Supervised Explainable Phrasal Reasoning with Neural Fuzzy Logic
Natural language inference (NLI) aims to determine the logical relationship between two sentences, such as Entailment, Contradiction, and Neutral. In recent years, deep learning models have become a prevailing approach t…
Explanation GenerationLogical ReasoningNatural Language InferenceSentence