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

• URL: http://arxiv.org/abs/2408.01191v1
• Date: Fri, 2 Aug 2024 11:18:32 GMT
• Title: A Weakly Supervised and Globally Explainable Learning Framework for Brain Tumor Segmentation
• Authors: Ruitao Xie, Limai Jiang, Xiaoxi He, Yi Pan, Yunpeng Cai,
• Abstract summary: Machine-based brain tumor segmentation can help doctors make better diagnoses. The complex structure of brain tumors and expensive pixel-level annotations present challenges for automatic tumor segmentation. We propose a counterfactual generation framework that achieves exceptional brain tumor segmentation performance without the need for pixel-level annotations.
• Score: 8.253446049933483
• License: http://arxiv.org/licenses/nonexclusive-distrib/1.0/
• Abstract: 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.

Related papers

• AEPL: Automated and Editable Prompt Learning for Brain Tumor Segmentation [12.347340694969212]
  Brain tumor segmentation is crucial for accurate diagnosis and treatment planning. Existing methods of-ten fail to effectively incorporate medical domain knowledgesuch as tumor grade. We propose an Automated and Editable Prompt Learning framework that integrates tumor grade into the seg-mentation process.
  arXiv  Detail & Related papers  (2024-10-21T20:29:44Z)
• UniCell: Universal Cell Nucleus Classification via Prompt Learning [76.11864242047074]
  We propose a universal cell nucleus classification framework (UniCell) It employs a novel prompt learning mechanism to uniformly predict the corresponding categories of pathological images from different dataset domains. In particular, our framework adopts an end-to-end architecture for nuclei detection and classification, and utilizes flexible prediction heads for adapting various datasets.
  arXiv  Detail & Related papers  (2024-02-20T11:50:27Z)
• ZePT: Zero-Shot Pan-Tumor Segmentation via Query-Disentangling and Self-Prompting [11.087654014615955]
  We propose a new zero-shot pan-tumor segmentation framework (ZePT) based on query-disentangling and self-prompting to segment unseen tumor categories. ZePT disentangles the object queries into two subsets and trains them in two stages. Experiments on various tumor segmentation tasks demonstrate the performance superiority of ZePT.
  arXiv  Detail & Related papers  (2023-12-07T12:09:56Z)
• Prototype-Driven and Multi-Expert Integrated Multi-Modal MR Brain Tumor Image Segmentation [23.127213250062425]
  The impact of information aliasing caused by the mutual inclusion of tumor sub-regions is often ignored. Existing methods usually do not take tailored efforts to highlight the single tumor sub-region features. We propose a multi-modal MR brain tumor segmentation method with tumor prototype-driven and multi-expert integration.
  arXiv  Detail & Related papers  (2023-07-22T22:41:11Z)
• Optimized Global Perturbation Attacks For Brain Tumour ROI Extraction From Binary Classification Models [0.304585143845864]
  We propose a weakly supervised approach to obtain regions of interest using binary class labels. We also propose a novel objective function to train the generator model based on a pretrained binary classification model.
  arXiv  Detail & Related papers  (2022-11-09T14:52:36Z)
• Triplet Contrastive Learning for Brain Tumor Classification [99.07846518148494]
  We present a novel approach of directly learning deep embeddings for brain tumor types, which can be used for downstream tasks such as classification. We evaluate our method on an extensive brain tumor dataset which consists of 27 different tumor classes, out of which 13 are defined as rare.
  arXiv  Detail & Related papers  (2021-08-08T11:26:34Z)
• Mitigating Generation Shifts for Generalized Zero-Shot Learning [52.98182124310114]
  Generalized Zero-Shot Learning (GZSL) is the task of leveraging semantic information (e.g., attributes) to recognize the seen and unseen samples, where unseen classes are not observable during training. We propose a novel Generation Shifts Mitigating Flow framework for learning unseen data synthesis efficiently and effectively. Experimental results demonstrate that GSMFlow achieves state-of-the-art recognition performance in both conventional and generalized zero-shot settings.
  arXiv  Detail & Related papers  (2021-07-07T11:43:59Z)
• Generalized Organ Segmentation by Imitating One-shot Reasoning using Anatomical Correlation [55.1248480381153]
  We propose OrganNet which learns a generalized organ concept from a set of annotated organ classes and then transfer this concept to unseen classes. We show that OrganNet can effectively resist the wide variations in organ morphology and produce state-of-the-art results in one-shot segmentation task.
  arXiv  Detail & Related papers  (2021-03-30T13:41:12Z)
• G-MIND: An End-to-End Multimodal Imaging-Genetics Framework for Biomarker Identification and Disease Classification [49.53651166356737]
  We propose a novel deep neural network architecture to integrate imaging and genetics data, as guided by diagnosis, that provides interpretable biomarkers. We have evaluated our model on a population study of schizophrenia that includes two functional MRI (fMRI) paradigms and Single Nucleotide Polymorphism (SNP) data.
  arXiv  Detail & Related papers  (2021-01-27T19:28:04Z)
• Unsupervised Feature Selection for Tumor Profiles using Autoencoders and Kernel Methods [1.9078991171384014]
  This work aims to learn meaningful and low dimensional representations of tumor samples and find tumor subtype clusters. The proposed method named Latent Kernel Feature Selection (LKFS) is an unsupervised approach for gene selection in tumor gene expression profiles.
  arXiv  Detail & Related papers  (2020-07-12T21:59:05Z)
• Weakly supervised multiple instance learning histopathological tumor segmentation [51.085268272912415]
  We propose a weakly supervised framework for whole slide imaging segmentation. We exploit a multiple instance learning scheme for training models. The proposed framework has been evaluated on multi-locations and multi-centric public data from The Cancer Genome Atlas and the PatchCamelyon dataset.
  arXiv  Detail & Related papers  (2020-04-10T13:12:47Z)

This list is automatically generated from the titles and abstracts of the papers in this site.

This site does not guarantee the quality of this site (including all information) and is not responsible for any consequences.