AC-UNet: Adaptive Connection UNet for White Matter Tract Segmentation Through Neural Architecture Search

No Thumbnail Available
File version
Author(s)
Zhu, Yanming
Tchetchenian, Ari
Yin, Xuefei
Liew, Alan
Song, Yang
Meijering, Erik
Primary Supervisor
Other Supervisors
Editor(s)
Date
2024
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