Transformer Based Models for Unsupervised Anomaly Segmentation in Brain MR Images - Archive ouverte HAL
Communication Dans Un Congrès Année : 2023

Transformer Based Models for Unsupervised Anomaly Segmentation in Brain MR Images

Résumé

The quality of patient care associated with diagnostic radiology is proportionate to a physician’s workload. Segmentation is a fundamental limiting precursor to both diagnostic and therapeutic procedures. Advances in machine learning (ML) aim to increase diagnostic efficiency by replacing a single application with generalized algorithms. The goal of unsupervised anomaly detection (UAD) is to identify potential anomalous regions unseen during training, where convolutional neural network (CNN) based autoencoders (AEs), and variational autoencoders (VAEs) are considered a de facto approach for reconstruction based-anomaly segmentation. The restricted receptive field in CNNs limits the CNN to model the global context. Hence, if the anomalous regions cover large parts of the image, the CNN-based AEs are not capable of bringing a semantic understanding of the image. Meanwhile, vision transformers (ViTs) have emerged as a competitive alternative to CNNs. It relies on the self-attention mechanism that can relate image patches to each other. We investigate in this paper Transformer’s capabilities in building AEs for the reconstruction-based UAD task to reconstruct a coherent and more realistic image. We focus on anomaly segmentation for brain magnetic resonance imaging (MRI) and present five Transformer-based models while enabling segmentation performance comparable to or superior to state-of-the-art (SOTA) models. The source code is made publicly available on GitHub. © 2023, The Author(s), under exclusive license to Springer Nature Switzerland AG.

Dates et versions

hal-04282432 , version 1 (13-11-2023)

Identifiants

Citer

Ahmed Ghorbel, Ahmed Aldahdooh, Shadi Albarqouni, Wassim Hamidouche. Transformer Based Models for Unsupervised Anomaly Segmentation in Brain MR Images. Proceedings of the 8th International MICCAI Brainlesion Workshop, BrainLes 2022, Sep 2022, Singapore, Singapore. pp.25-44, ⟨10.1007/978-3-031-33842-7_3⟩. ⟨hal-04282432⟩
19 Consultations
0 Téléchargements

Altmetric

Partager

More