AC-UNet: Adaptive Connection UNet for White Matter Tract Segmentation Through Neural Architecture Search
File version
Author(s)
Tchetchenian, Ari
Yin, Xuefei
Liew, Alan
Song, Yang
Meijering, Erik
Griffith University Author(s)
Primary Supervisor
Other Supervisors
Editor(s)
Date
Size
File type(s)
Location
Athens, Greece
License
Abstract
White matter tract segmentation in diffusion magnetic resonance images is crucial for brain health analysis. A prevailing method for this task is deep learning using U-shaped networks. Several variants of UNet have been proposed to improve its skip connections, resulting in segmentation improvements. In this paper, we propose a novel neural architecture search-based solution for skip connection optimization, named AC-UNet (Adaptive Connection UNet). It can automatically identify the optimal skip connections of a U-shaped network to construct the best architecture for the task. Moreover, we propose to search non-repeatable operations for each layer, which further extends the exploration of feature aggregation for better segmentation. Experimental results on the largest public tractograms dataset demonstrate the superiority of AC-UNet over mainstream UNet-based architectures.
Journal Title
Conference Title
2024 IEEE International Symposium on Biomedical Imaging (ISBI)
Book Title
Edition
Volume
Issue
Thesis Type
Degree Program
School
Publisher link
Patent number
Funder(s)
Grant identifier(s)
Rights Statement
Rights Statement
Item Access Status
Note
Access the data
Related item(s)
Subject
Persistent link to this record
Citation
Zhu, Y; Tchetchenian, A; Yin, X; Liew, A; Song, Y; Meijering, E, AC-UNet: Adaptive Connection UNet for White Matter Tract Segmentation Through Neural Architecture Search, 2024 IEEE International Symposium on Biomedical Imaging (ISBI), 2024