Deep Expectation-Maximization for Semi-Supervised Lung Cancer Screening
- URL: http://arxiv.org/abs/2010.01173v1
- Date: Fri, 2 Oct 2020 19:17:07 GMT
- Title: Deep Expectation-Maximization for Semi-Supervised Lung Cancer Screening
- Authors: Sumeet Menon, David Chapman, Phuong Nguyen, Yelena Yesha, Michael
Morris, Babak Saboury
- Abstract summary: We present a semi-supervised algorithm for lung cancer screening.
A 3D Convolutional Neural Network (CNN) is trained using the Expectation-Maximization (EM) meta-algorithm.
We show that the Semi-Supervised EM algorithm greatly improves the classification accuracy of the cross-domain lung cancer screening.
- Score: 1.5379084885764847
- License: http://arxiv.org/licenses/nonexclusive-distrib/1.0/
- Abstract: We present a semi-supervised algorithm for lung cancer screening in which a
3D Convolutional Neural Network (CNN) is trained using the
Expectation-Maximization (EM) meta-algorithm. Semi-supervised learning allows a
smaller labelled data-set to be combined with an unlabeled data-set in order to
provide a larger and more diverse training sample. EM allows the algorithm to
simultaneously calculate a maximum likelihood estimate of the CNN training
coefficients along with the labels for the unlabeled training set which are
defined as a latent variable space. We evaluate the model performance of the
Semi-Supervised EM algorithm for CNNs through cross-domain training of the
Kaggle Data Science Bowl 2017 (Kaggle17) data-set with the National Lung
Screening Trial (NLST) data-set. Our results show that the Semi-Supervised EM
algorithm greatly improves the classification accuracy of the cross-domain lung
cancer screening, although results are lower than a fully supervised approach
with the advantage of additional labelled data from the unsupervised sample. As
such, we demonstrate that Semi-Supervised EM is a valuable technique to improve
the accuracy of lung cancer screening models using 3D CNNs.
Related papers
- Improving Lung Cancer Diagnosis and Survival Prediction with Deep Learning and CT Imaging [12.276877277186284]
Lung cancer is a major cancer-related deaths, and early diagnosis and treatment are crucial for improving patients' survival outcomes.
We propose to employ neural convolutional networks of networks obtained between the risk of lung cancer and the lungs in CT experiments.
Results demonstrate the effectiveness of both the mini-batched loss and binary cross-entropy to predict both lung cancer and the risk of the occurrence.
arXiv Detail & Related papers (2024-08-18T05:45:08Z) - The effect of data augmentation and 3D-CNN depth on Alzheimer's Disease
detection [51.697248252191265]
This work summarizes and strictly observes best practices regarding data handling, experimental design, and model evaluation.
We focus on Alzheimer's Disease (AD) detection, which serves as a paradigmatic example of challenging problem in healthcare.
Within this framework, we train predictive 15 models, considering three different data augmentation strategies and five distinct 3D CNN architectures.
arXiv Detail & Related papers (2023-09-13T10:40:41Z) - MESAHA-Net: Multi-Encoders based Self-Adaptive Hard Attention Network
with Maximum Intensity Projections for Lung Nodule Segmentation in CT Scan [6.266053305874546]
We propose an efficient end-to-end framework, the multi-encoder-based self-adaptive hard attention network (MESAHA-Net) for precise lung nodule segmentation in CT scans.
MESAHA-Net iteratively performs slice-by-slice 2D segmentation of lung nodules, focusing on the nodule region in each slice to generate 3D segmentation of lung nodules.
The proposed framework has been evaluated on the LIDC-IDRI dataset, the largest publicly available dataset for lung nodule segmentation.
arXiv Detail & Related papers (2023-04-04T07:05:15Z) - Dual Multi-scale Mean Teacher Network for Semi-supervised Infection
Segmentation in Chest CT Volume for COVID-19 [76.51091445670596]
Automated detecting lung infections from computed tomography (CT) data plays an important role for combating COVID-19.
Most current COVID-19 infection segmentation methods mainly relied on 2D CT images, which lack 3D sequential constraint.
Existing 3D CT segmentation methods focus on single-scale representations, which do not achieve the multiple level receptive field sizes on 3D volume.
arXiv Detail & Related papers (2022-11-10T13:11:21Z) - Improving Classification Model Performance on Chest X-Rays through Lung
Segmentation [63.45024974079371]
We propose a deep learning approach to enhance abnormal chest x-ray (CXR) identification performance through segmentations.
Our approach is designed in a cascaded manner and incorporates two modules: a deep neural network with criss-cross attention modules (XLSor) for localizing lung region in CXR images and a CXR classification model with a backbone of a self-supervised momentum contrast (MoCo) model pre-trained on large-scale CXR data sets.
arXiv Detail & Related papers (2022-02-22T15:24:06Z) - Localized Perturbations For Weakly-Supervised Segmentation of Glioma
Brain Tumours [0.5801621787540266]
This work proposes the use of localized perturbations as a weakly-supervised solution to extract segmentation masks of brain tumours from a pretrained 3D classification model.
We also propose a novel optimal perturbation method that exploits 3D superpixels to find the most relevant area for a given classification using a U-net architecture.
arXiv Detail & Related papers (2021-11-29T21:01:20Z) - Lung Cancer Lesion Detection in Histopathology Images Using Graph-Based
Sparse PCA Network [93.22587316229954]
We propose a graph-based sparse principal component analysis (GS-PCA) network, for automated detection of cancerous lesions on histological lung slides stained by hematoxylin and eosin (H&E)
We evaluate the performance of the proposed algorithm on H&E slides obtained from an SVM K-rasG12D lung cancer mouse model using precision/recall rates, F-score, Tanimoto coefficient, and area under the curve (AUC) of the receiver operator characteristic (ROC)
arXiv Detail & Related papers (2021-10-27T19:28:36Z) - Many-to-One Distribution Learning and K-Nearest Neighbor Smoothing for
Thoracic Disease Identification [83.6017225363714]
deep learning has become the most powerful computer-aided diagnosis technology for improving disease identification performance.
For chest X-ray imaging, annotating large-scale data requires professional domain knowledge and is time-consuming.
In this paper, we propose many-to-one distribution learning (MODL) and K-nearest neighbor smoothing (KNNS) methods to improve a single model's disease identification performance.
arXiv Detail & Related papers (2021-02-26T02:29:30Z) - A new semi-supervised self-training method for lung cancer prediction [0.28734453162509355]
There are only relatively few methods that simultaneously detect and classify nodules from computed tomography (CT) scans.
This study presents a complete end-to-end scheme to detect and classify lung nodules using the state-of-the-art Self-training with Noisy Student method.
arXiv Detail & Related papers (2020-12-17T09:53:51Z) - Improved Slice-wise Tumour Detection in Brain MRIs by Computing
Dissimilarities between Latent Representations [68.8204255655161]
Anomaly detection for Magnetic Resonance Images (MRIs) can be solved with unsupervised methods.
We have proposed a slice-wise semi-supervised method for tumour detection based on the computation of a dissimilarity function in the latent space of a Variational AutoEncoder.
We show that by training the models on higher resolution images and by improving the quality of the reconstructions, we obtain results which are comparable with different baselines.
arXiv Detail & Related papers (2020-07-24T14:02:09Z) - Primary Tumor Origin Classification of Lung Nodules in Spectral CT using
Transfer Learning [4.0657540412774935]
Early detection of lung cancer has been proven to decrease mortality significantly.
Recent development in computed tomography (CT), spectral CT, can potentially improve diagnostic accuracy.
We propose a detection and classification system for lung nodules in CT scans.
arXiv Detail & Related papers (2020-06-30T09:56:18Z)
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.