vix.ing · top · new · best · stats · spec

Hybrid Window Attention Based Transformer Architecture for Brain Tumor Segmentation

2022/09/16 by Himashi Peiris, Munawar Hayat, Peiris, Himashi +7 · 1 citation
Computer Science · Neuroscience · #Advanced Neural Network Applications #Brain Tumor Detection and Classification #Computer Vision and Pattern Recognition (cs.CV) #FOS: Computer and information sciences #FOS: Electrical engineering #Image and Video Processing (eess.IV) #Medical Image Segmentation Techniques #electronic engineering #information engineering

paper · pdf · doi:10.48550/arxiv.2209.07704

openalex publication_date 2022/09/16 · openalex created_date 2022/09/20 · openalex updated_date 2026/07/28

Abstract

As intensities of MRI volumes are inconsistent across institutes, it is essential to extract universal features of multi-modal MRIs to precisely segment brain tumors. In this concept, we propose a volumetric vision transformer that follows two windowing strategies in attention for extracting fine features and local distributional smoothness (LDS) during model training inspired by virtual adversarial training (VAT) to make the model robust. We trained and evaluated network architecture on the FeTS Challenge 2022 dataset. Our performance on the online validation dataset is as follows: Dice Similarity Score of 81.71%, 91.38% and 85.40%; Hausdorff Distance (95%) of 14.81 mm, 3.93 mm, 11.18 mm for the enhancing tumor, whole tumor, and tumor core, respectively. Overall, the experimental results verify our method's effectiveness by yielding better performance in segmentation accuracy for each tumor sub-region. Our code implementation is publicly available : https://github.com/himashi92/vizvivafets2022

Cited by

Related