CVAD: A generic medical anomaly detector based on Cascade VAE
- URL: http://arxiv.org/abs/2110.15811v1
- Date: Fri, 29 Oct 2021 14:20:43 GMT
- Title: CVAD: A generic medical anomaly detector based on Cascade VAE
- Authors: Xiaoyuan Guo, Judy Wawira Gichoya, Saptarshi Purkayastha and Imon
Banerjee
- Abstract summary: We focus on the generalizability of OOD detection for medical images and propose a self-supervised Cascade Variational autoencoder-based Anomaly Detector (CVAD)
We use a variational autoencoders' cascade architecture, which combines latent representation at multiple scales, before being fed to a discriminator to distinguish the OOD data from the in-distribution (ID) data.
We compare the performance with the state-of-the-art deep learning models to demonstrate our model's efficacy on various open-access medical imaging datasets for both intra- and inter-class OOD.
- Score: 2.647674705784439
- License: http://creativecommons.org/licenses/by-nc-sa/4.0/
- Abstract: Detecting out-of-distribution (OOD) samples in medical imaging plays an
important role for downstream medical diagnosis. However, existing OOD
detectors are demonstrated on natural images composed of inter-classes and have
difficulty generalizing to medical images. The key issue is the granularity of
OOD data in the medical domain, where intra-class OOD samples are predominant.
We focus on the generalizability of OOD detection for medical images and
propose a self-supervised Cascade Variational autoencoder-based Anomaly
Detector (CVAD). We use a variational autoencoders' cascade architecture, which
combines latent representation at multiple scales, before being fed to a
discriminator to distinguish the OOD data from the in-distribution (ID) data.
Finally, both the reconstruction error and the OOD probability predicted by the
binary discriminator are used to determine the anomalies. We compare the
performance with the state-of-the-art deep learning models to demonstrate our
model's efficacy on various open-access medical imaging datasets for both
intra- and inter-class OOD. Further extensive results on datasets including
common natural datasets show our model's effectiveness and generalizability.
The code is available at https://github.com/XiaoyuanGuo/CVAD.
Related papers
- Unsupervised Hybrid framework for ANomaly Detection (HAND) -- applied to Screening Mammogram [5.387300498478745]
Out-of-distribution (OOD) detection is crucial for enhancing the generalization of AI models used in mammogram screening.
We developed a novel backbone - HAND - for detecting OOD from large-scale digital screening mammogram studies.
Hand pipeline offers an automated efficient computational solution for domain-specific quality checks in external screening mammograms.
arXiv Detail & Related papers (2024-09-17T20:12:50Z) - TTA-OOD: Test-time Augmentation for Improving Out-of-Distribution Detection in Gastrointestinal Vision [6.290783164114315]
We introduce a test-time augmentation segment into the OOD detection pipeline.
This augmentation shifts the pixel space, which translates into a more distinct semantic representation for OOD examples.
We evaluate our method against existing state-of-the-art OOD scores.
arXiv Detail & Related papers (2024-07-19T04:50:54Z) - Adapting Visual-Language Models for Generalizable Anomaly Detection in Medical Images [68.42215385041114]
This paper introduces a novel lightweight multi-level adaptation and comparison framework to repurpose the CLIP model for medical anomaly detection.
Our approach integrates multiple residual adapters into the pre-trained visual encoder, enabling a stepwise enhancement of visual features across different levels.
Our experiments on medical anomaly detection benchmarks demonstrate that our method significantly surpasses current state-of-the-art models.
arXiv Detail & Related papers (2024-03-19T09:28:19Z) - From Global to Local: Multi-scale Out-of-distribution Detection [129.37607313927458]
Out-of-distribution (OOD) detection aims to detect "unknown" data whose labels have not been seen during the in-distribution (ID) training process.
Recent progress in representation learning gives rise to distance-based OOD detection.
We propose Multi-scale OOD DEtection (MODE), a first framework leveraging both global visual information and local region details.
arXiv Detail & Related papers (2023-08-20T11:56:25Z) - Unleashing Mask: Explore the Intrinsic Out-of-Distribution Detection
Capability [70.72426887518517]
Out-of-distribution (OOD) detection is an indispensable aspect of secure AI when deploying machine learning models in real-world applications.
We propose a novel method, Unleashing Mask, which aims to restore the OOD discriminative capabilities of the well-trained model with ID data.
Our method utilizes a mask to figure out the memorized atypical samples, and then finetune the model or prune it with the introduced mask to forget them.
arXiv Detail & Related papers (2023-06-06T14:23:34Z) - Out-of-distribution Detection with Implicit Outlier Transformation [72.73711947366377]
Outlier exposure (OE) is powerful in out-of-distribution (OOD) detection.
We propose a novel OE-based approach that makes the model perform well for unseen OOD situations.
arXiv Detail & Related papers (2023-03-09T04:36:38Z) - A Multi-Scale Framework for Out-of-Distribution Detection in Dermoscopic
Images [10.20384144853726]
We propose a multi-scale detection framework to detect out-of-distribution skin disease image data.
Our framework extracts features from different layers of the neural network.
Experiments show that the proposed framework achieves superior performance when compared with other state-of-the-art methods.
arXiv Detail & Related papers (2023-01-18T13:49:35Z) - DIAGNOSE: Avoiding Out-of-distribution Data using Submodular Information
Measures [13.492292022589918]
We propose Diagnose, a novel active learning framework that can jointly model similarity and dissimilarity.
Our experiments verify the superiority of Diagnose over the state-of-the-art AL methods across multiple domains of medical imaging.
arXiv Detail & Related papers (2022-10-04T11:07:48Z) - A knee cannot have lung disease: out-of-distribution detection with
in-distribution voting using the medical example of chest X-ray
classification [58.720142291102135]
The study employed the commonly used chest X-ray classification model, CheXnet, trained on the chest X-ray 14 data set.
To detect OOD data for multi-label classification, we proposed in-distribution voting (IDV)
The proposed IDV approach trained on ID (chest X-ray 14) and OOD data (IRMA and ImageNet) achieved, on average, 0.999 OOD detection AUC across the three data sets.
arXiv Detail & Related papers (2022-08-01T18:20:36Z) - OODformer: Out-Of-Distribution Detection Transformer [15.17006322500865]
In real-world safety-critical applications, it is important to be aware if a new data point is OOD.
This paper proposes a first-of-its-kind OOD detection architecture named OODformer.
arXiv Detail & Related papers (2021-07-19T15:46:38Z) - 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)
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.