Differentiable Attention Unet-like Nerual Architecture Search for Multimodal Magnetic Resonance Imaging-based Glioma Segmentaion DAUNAS for Multimodal MRI-based Glioma Segmentation

Jing Qin,Dongyuan Wu,Erqiang Zhou,Zhen Qin,Zhiguang Qin
DOI: https://doi.org/10.1145/3644116.3644119
2023-01-01
Abstract:The utilization of artificial intelligence has had a profound impact on diverse areas within the medical field day by day. Specifically, the identification and management of gliomas, a specific category of brain tumors characterized by a challenging prognosis, heavily depend on the computer-assisted analysis of magnetic resonance imaging (MRI) scans. In the present study, we introduce a novel approach based on neural architecture search (NAS) for accurately segmenting brain tumors utilizing multimodal volumetric MRI scans. Our method employs three classes of candidate operations for different cells, with each operation having a learnable probabilistic parameter. By iteratively updating the operation weights and other network parameters, we discover optimal structure for the encoder and decoder cells. Additionally, we introduce an attention module connection to the automatic search, complementing the connection between the encoder cells and decoder cells for brain MRI processing. Through extensive experiments conducted on the BraTS 2019 dataset, we validate the effectiveness and scalability of our proposed algorithm. Our approach not only relieves the burden of manual architecture design but also achieves competitive performance in terms of brain tumor segmentation.
What problem does this paper attempt to address?