3D Attention U-Net with Pretraining: A Solution to CADA-Aneurysm Segmentation Challenge

Ziyu Su, Yizhuan Jia, Weibin Liao, Yi Lv, Jiaqi Dou, Zhongwei Sun, Xuesong Li*

*Corresponding author for this work

Research output: Chapter in Book/Report/Conference proceedingConference contributionpeer-review

4 Citations (Scopus)

Abstract

Early detection and accurate segmentation of cerebral aneurysm is important for clinical diagnosis and prevention of rupture, which would be life threatening. 3D images can provide abundant information for characterizing the aneurysm. But the traditional manual segmentation of aneurysms takes lots of time and effort. Therefore, accurate and rapid automatic algorithm for 3D segmentation of aneurysm is needed. U-Net is a widely used deep learning network in medical image segmentation, but its performance is limited by the amount of data. In this challenge of aneurysm segmentation, we proposed to add attention gate and Models Genesis pretraining mechanisms to the classical U-Net model to improve the results. The dice of 3D U-net, 3D Attention U-Net, pretrained 3D U-Net and pretrained 3D Attention U-Net are 0.881, 0.884, 0.890 and 0.907, respectively. The experimental results show that the use of attention gate and Models Genesis can significantly improve the performance of U-Net model in segmenting aneurysms. This work achieved rank one in CADA 2020- Aneurysm Segmentation Challenge.

Original languageEnglish
Title of host publicationCerebral Aneurysm Detection - First Challenge, CADA 2020, Held in Conjunction with MICCAI 2020, Proceedings
EditorsAnja Hennemuth, Leonid Goubergrits, Matthias Ivantsits, Jan-Martin Kuhnigk
PublisherSpringer Science and Business Media Deutschland GmbH
Pages58-67
Number of pages10
ISBN (Print)9783030728618
DOIs
Publication statusPublished - 2021
Event1st Cerebral Aneurysm Detection and Analysis challenge, CADA 2020 held in Conjunction with 23rd International Conference on Medical Image Computing and Computer Assisted Intervention, MICCAI 2020 - Virtual, Online
Duration: 8 Oct 20208 Oct 2020

Publication series

NameLecture Notes in Computer Science (including subseries Lecture Notes in Artificial Intelligence and Lecture Notes in Bioinformatics)
Volume12643 LNCS
ISSN (Print)0302-9743
ISSN (Electronic)1611-3349

Conference

Conference1st Cerebral Aneurysm Detection and Analysis challenge, CADA 2020 held in Conjunction with 23rd International Conference on Medical Image Computing and Computer Assisted Intervention, MICCAI 2020
CityVirtual, Online
Period8/10/208/10/20

Keywords

  • 3D Attention U-Net
  • Image segmentation
  • Transfer learning

Fingerprint

Dive into the research topics of '3D Attention U-Net with Pretraining: A Solution to CADA-Aneurysm Segmentation Challenge'. Together they form a unique fingerprint.

Cite this