Feature-enhanced Adversarial Semi-supervised Semantic Segmentation
Network for Pulmonary Embolism Annotation
- URL: http://arxiv.org/abs/2204.04217v1
- Date: Fri, 8 Apr 2022 04:21:02 GMT
- Title: Feature-enhanced Adversarial Semi-supervised Semantic Segmentation
Network for Pulmonary Embolism Annotation
- Authors: Ting-Wei Cheng, Jerry Chang, Ching-Chun Huang, Chin Kuo, Yun-Chien
Cheng
- Abstract summary: This study established a feature-enhanced adversarial semi-supervised semantic segmentation model to automatically annotate pulmonary embolism lesion areas.
In current studies, all of the PEA image segmentation methods are trained by supervised learning.
This study proposed a semi-supervised learning method to make the model applicable to different datasets by adding a small amount of unlabeled images.
- Score: 6.142272540492936
- License: http://creativecommons.org/licenses/by-nc-nd/4.0/
- Abstract: This study established a feature-enhanced adversarial semi-supervised
semantic segmentation model to automatically annotate pulmonary embolism lesion
areas in computed tomography pulmonary angiogram (CTPA) images. In current
studies, all of the PE CTPA image segmentation methods are trained by
supervised learning. However, the supervised learning models need to be
retrained and the images need to be relabeled when the CTPA images come from
different hospitals. This study proposed a semi-supervised learning method to
make the model applicable to different datasets by adding a small amount of
unlabeled images. By training the model with both labeled and unlabeled images,
the accuracy of unlabeled images can be improved and the labeling cost can be
reduced. Our semi-supervised segmentation model includes a segmentation network
and a discriminator network. We added feature information generated from the
encoder of segmentation network to the discriminator so that it can learn the
similarity between predicted mask and ground truth mask. This HRNet-based
architecture can maintain a higher resolution for convolutional operations so
the prediction of small PE lesion areas can be improved. We used the labeled
open-source dataset and the unlabeled National Cheng Kung University Hospital
(NCKUH) (IRB number: B-ER-108-380) dataset to train the semi-supervised
learning model, and the resulting mean intersection over union (mIOU), dice
score, and sensitivity achieved 0.3510, 0.4854, and 0.4253, respectively on the
NCKUH dataset. Then, we fine-tuned and tested the model with a small amount of
unlabeled PE CTPA images from China Medical University Hospital (CMUH) (IRB
number: CMUH110-REC3-173) dataset. Comparing the results of our semi-supervised
model with the supervised model, the mIOU, dice score, and sensitivity improved
from 0.2344, 0.3325, and 0.3151 to 0.3721, 0.5113, and 0.4967, respectively.
Related papers
- LVM-Med: Learning Large-Scale Self-Supervised Vision Models for Medical
Imaging via Second-order Graph Matching [59.01894976615714]
We introduce LVM-Med, the first family of deep networks trained on large-scale medical datasets.
We have collected approximately 1.3 million medical images from 55 publicly available datasets.
LVM-Med empirically outperforms a number of state-of-the-art supervised, self-supervised, and foundation models.
arXiv Detail & Related papers (2023-06-20T22:21:34Z) - Mediastinal Lymph Node Detection and Segmentation Using Deep Learning [1.7188280334580195]
In clinical practice, computed tomography (CT) and positron emission tomography (PET) imaging detect abnormal lymph nodes (LNs)
Deep convolutional neural networks frequently segment items in medical photographs.
A well-established deep learning technique UNet was modified using bilinear and total generalized variation (TGV) based up strategy to segment and detect mediastinal lymph nodes.
The modified UNet maintains texture discontinuities, selects noisy areas, searches appropriate balance points through backpropagation, and recreates image resolution.
arXiv Detail & Related papers (2022-11-24T02:55:20Z) - PCA: Semi-supervised Segmentation with Patch Confidence Adversarial
Training [52.895952593202054]
We propose a new semi-supervised adversarial method called Patch Confidence Adrial Training (PCA) for medical image segmentation.
PCA learns the pixel structure and context information in each patch to get enough gradient feedback, which aids the discriminator in convergent to an optimal state.
Our method outperforms the state-of-the-art semi-supervised methods, which demonstrates its effectiveness for medical image segmentation.
arXiv Detail & Related papers (2022-07-24T07:45:47Z) - Mixed-UNet: Refined Class Activation Mapping for Weakly-Supervised
Semantic Segmentation with Multi-scale Inference [28.409679398886304]
We develop a novel model named Mixed-UNet, which has two parallel branches in the decoding phase.
We evaluate the designed Mixed-UNet against several prevalent deep learning-based segmentation approaches on our dataset collected from the local hospital and public datasets.
arXiv Detail & Related papers (2022-05-06T08:37:02Z) - Automatic size and pose homogenization with spatial transformer network
to improve and accelerate pediatric segmentation [51.916106055115755]
We propose a new CNN architecture that is pose and scale invariant thanks to the use of Spatial Transformer Network (STN)
Our architecture is composed of three sequential modules that are estimated together during training.
We test the proposed method in kidney and renal tumor segmentation on abdominal pediatric CT scanners.
arXiv Detail & Related papers (2021-07-06T14:50:03Z) - 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) - Few-shot Medical Image Segmentation using a Global Correlation Network
with Discriminative Embedding [60.89561661441736]
We propose a novel method for few-shot medical image segmentation.
We construct our few-shot image segmentor using a deep convolutional network trained episodically.
We enhance discriminability of deep embedding to encourage clustering of the feature domains of the same class.
arXiv Detail & Related papers (2020-12-10T04:01:07Z) - Deep Multi-path Network Integrating Incomplete Biomarker and Chest CT
Data for Evaluating Lung Cancer Risk [4.822738153096615]
We propose a new network design, termed as multi-path multi-modal missing network (M3Net)
It integrates the multi-modal data (i.e., CDEs, biomarker and CT image) considering missing modality with multiple paths neural network.
The network can be trained end-to-end with both medical image features and CDEs/biomarkers, or make a prediction with single modality.
arXiv Detail & Related papers (2020-10-19T13:55:40Z) - 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) - An interpretable classifier for high-resolution breast cancer screening
images utilizing weakly supervised localization [45.00998416720726]
We propose a framework to address the unique properties of medical images.
This model first uses a low-capacity, yet memory-efficient, network on the whole image to identify the most informative regions.
It then applies another higher-capacity network to collect details from chosen regions.
Finally, it employs a fusion module that aggregates global and local information to make a final prediction.
arXiv Detail & Related papers (2020-02-13T15:28:42Z)
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.