A Two-Stage Multiple Instance Learning Framework for the Detection of
Breast Cancer in Mammograms
- URL: http://arxiv.org/abs/2004.11726v1
- Date: Fri, 24 Apr 2020 13:06:47 GMT
- Title: A Two-Stage Multiple Instance Learning Framework for the Detection of
Breast Cancer in Mammograms
- Authors: Sarath Chandra K, Arunava Chakravarty, Nirmalya Ghosh, Tandra Sarkar,
Ramanathan Sethuraman, Debdoot Sheet
- Abstract summary: Mammograms are commonly employed in the large scale screening of breast cancer.
We propose a two-stage Multiple Instance Learning framework for image-level detection of malignancy.
A global image-level feature is computed as a weighted average of patch-level features learned using a CNN.
Our method performed well on the task of localization of masses with an average Precision/Recall of 0.76/0.80 and acheived an average AUC of 0.91 on the imagelevel classification task.
- Score: 13.842620686759616
- License: http://arxiv.org/licenses/nonexclusive-distrib/1.0/
- Abstract: Mammograms are commonly employed in the large scale screening of breast
cancer which is primarily characterized by the presence of malignant masses.
However, automated image-level detection of malignancy is a challenging task
given the small size of the mass regions and difficulty in discriminating
between malignant, benign mass and healthy dense fibro-glandular tissue. To
address these issues, we explore a two-stage Multiple Instance Learning (MIL)
framework. A Convolutional Neural Network (CNN) is trained in the first stage
to extract local candidate patches in the mammograms that may contain either a
benign or malignant mass. The second stage employs a MIL strategy for an image
level benign vs. malignant classification. A global image-level feature is
computed as a weighted average of patch-level features learned using a CNN. Our
method performed well on the task of localization of masses with an average
Precision/Recall of 0.76/0.80 and acheived an average AUC of 0.91 on the
imagelevel classification task using a five-fold cross-validation on the
INbreast dataset. Restricting the MIL only to the candidate patches extracted
in Stage 1 led to a significant improvement in classification performance in
comparison to a dense extraction of patches from the entire mammogram.
Related papers
- Active Learning Enhances Classification of Histopathology Whole Slide
Images with Attention-based Multiple Instance Learning [48.02011627390706]
We train an attention-based MIL and calculate a confidence metric for every image in the dataset to select the most uncertain WSIs for expert annotation.
With a novel attention guiding loss, this leads to an accuracy boost of the trained models with few regions annotated for each class.
It may in the future serve as an important contribution to train MIL models in the clinically relevant context of cancer classification in histopathology.
arXiv Detail & Related papers (2023-03-02T15:18:58Z) - Stain-invariant self supervised learning for histopathology image
analysis [74.98663573628743]
We present a self-supervised algorithm for several classification tasks within hematoxylin and eosin stained images of breast cancer.
Our method achieves the state-of-the-art performance on several publicly available breast cancer datasets.
arXiv Detail & Related papers (2022-11-14T18:16:36Z) - Deep Superpixel Generation and Clustering for Weakly Supervised
Segmentation of Brain Tumors in MR Images [0.0]
This work proposes the use of a superpixel generation model and a superpixel clustering model to enable weakly supervised brain tumor segmentations.
We used 2D slices of magnetic resonance brain scans from the Multimodal Brain Tumor Challenge 2020 dataset and labels indicating the presence of tumors to train the pipeline.
Our method achieved a mean Dice coefficient of 0.691 and a mean 95% Hausdorff distance of 18.1, outperforming existing superpixel-based weakly supervised segmentation methods.
arXiv Detail & Related papers (2022-09-20T18:08:34Z) - EMT-NET: Efficient multitask network for computer-aided diagnosis of
breast cancer [58.720142291102135]
We propose an efficient and light-weighted learning architecture to classify and segment breast tumors simultaneously.
We incorporate a segmentation task into a tumor classification network, which makes the backbone network learn representations focused on tumor regions.
The accuracy, sensitivity, and specificity of tumor classification is 88.6%, 94.1%, and 85.3%, respectively.
arXiv Detail & Related papers (2022-01-13T05:24:40Z) - Multi-Scale Input Strategies for Medulloblastoma Tumor Classification
using Deep Transfer Learning [59.30734371401316]
Medulloblastoma is the most common malignant brain cancer among children.
CNN has shown promising results for MB subtype classification.
We study the impact of tile size and input strategy.
arXiv Detail & Related papers (2021-09-14T09:42:37Z) - Robust Deep AUC Maximization: A New Surrogate Loss and Empirical Studies
on Medical Image Classification [63.44396343014749]
We propose a new margin-based surrogate loss function for the AUC score.
It is more robust than the commonly used.
square loss while enjoying the same advantage in terms of large-scale optimization.
To the best of our knowledge, this is the first work that makes DAM succeed on large-scale medical image datasets.
arXiv Detail & Related papers (2020-12-06T03:41:51Z) - Leveraging SLIC Superpixel Segmentation and Cascaded Ensemble SVM for
Fully Automated Mass Detection In Mammograms [1.7205106391379026]
This paper proposes a rigorous segmentation method, supported by morphological enhancement using grayscale linear filters.
A novel cascaded ensemble of support vector machines (SVM) is used to effectively tackle the class imbalance and provide significant predictions.
arXiv Detail & Related papers (2020-10-20T15:02:25Z) - Synthesizing lesions using contextual GANs improves breast cancer
classification on mammograms [0.4297070083645048]
We present a novel generative adversarial network (GAN) model for data augmentation that can realistically synthesize and remove lesions on mammograms.
With self-attention and semi-supervised learning components, the U-net-based architecture can generate high resolution (256x256px) outputs.
arXiv Detail & Related papers (2020-05-29T21:23:00Z) - Learning from Suspected Target: Bootstrapping Performance for Breast
Cancer Detection in Mammography [6.323318523772466]
We introduce a novel top likelihood loss together with a new sampling procedure to select and train the suspected target regions.
We firstly test our proposed method on a private dense mammogram dataset.
Results show that our proposed method greatly reduce the false positive rate and the specificity is increased by 0.25 on detecting mass type cancer.
arXiv Detail & Related papers (2020-03-01T09:04:24Z) - Two-stage multi-scale breast mass segmentation for full mammogram
analysis without user intervention [2.7490008316742096]
Mammography is the primary imaging modality used for early detection and diagnosis of breast cancer.
Among diverse types of breast abnormalities, masses are the most important clinical findings of breast carcinomas.
We present a two-stage multi-scale pipeline that provides accurate mass contours from high-resolution full mammograms.
arXiv Detail & Related papers (2020-02-27T13:16:22Z) - Stan: Small tumor-aware network for breast ultrasound image segmentation [68.8204255655161]
We propose a novel deep learning architecture called Small Tumor-Aware Network (STAN) to improve the performance of segmenting tumors with different size.
The proposed approach outperformed the state-of-the-art approaches in segmenting small breast tumors.
arXiv Detail & Related papers (2020-02-03T22:25:01Z)
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.