Attention V-Net: A Modified V-Net Architecture for Left Atrial Segmentation
We propose a fully convolutional neural network based on the attention mechanism for 3D medical image segmentation tasks. It can adaptively learn to highlight the salient features of images that are useful for image segmentation tasks. Some prior methods enhance accuracy using multi-scale feature fu...
Main Authors: | , , |
---|---|
Format: | Article |
Language: | English |
Published: |
MDPI AG
2022-04-01
|
Series: | Applied Sciences |
Subjects: | |
Online Access: | https://www.mdpi.com/2076-3417/12/8/3764 |
_version_ | 1797437051801436160 |
---|---|
author | Xiaoli Liu Ruoqi Yin Jianqin Yin |
author_facet | Xiaoli Liu Ruoqi Yin Jianqin Yin |
author_sort | Xiaoli Liu |
collection | DOAJ |
description | We propose a fully convolutional neural network based on the attention mechanism for 3D medical image segmentation tasks. It can adaptively learn to highlight the salient features of images that are useful for image segmentation tasks. Some prior methods enhance accuracy using multi-scale feature fusion or dilated convolution, which is basically artificial and lacks the flexibility of the model itself. Therefore, some works proposed the 2D attention gate module, but these works process 2D medical slice images, ignoring the correlation between 3D image sequences. In contrast, the 3D attention gate can comprehensively use the information of three dimensions of medical images. In this paper, we propose the Attention V-Net architecture, which uses the 3D attention gate module, and applied it to the left atrium segmentation framework based on semi-supervised learning. The proposed method is evaluated on the dataset of the 2018 left atrial challenge. The experimental results show that the Attention V-Net obtains improved performance under evaluation indicators, such as Dice, Jaccard, ASD (Average surface distance), and 95HD (Hausdorff distance). The result indicates that the model in this paper can effectively improve the accuracy of left atrial segmentation, therefore laying the foundation for subsequent work such as in atrial reconstruction. Meanwhile, our model is of great significance for assisting doctors in treating cardiovascular diseases. |
first_indexed | 2024-03-09T11:13:24Z |
format | Article |
id | doaj.art-ddf24fbc4f1646ec859c42b232c14a14 |
institution | Directory Open Access Journal |
issn | 2076-3417 |
language | English |
last_indexed | 2024-03-09T11:13:24Z |
publishDate | 2022-04-01 |
publisher | MDPI AG |
record_format | Article |
series | Applied Sciences |
spelling | doaj.art-ddf24fbc4f1646ec859c42b232c14a142023-12-01T00:38:11ZengMDPI AGApplied Sciences2076-34172022-04-01128376410.3390/app12083764Attention V-Net: A Modified V-Net Architecture for Left Atrial SegmentationXiaoli Liu0Ruoqi Yin1Jianqin Yin2School of Artificial Intelligence, Beijing University of Posts and Telecommunications, Beijing 100876, ChinaSchool of Artificial Intelligence, Beijing University of Posts and Telecommunications, Beijing 100876, ChinaSchool of Artificial Intelligence, Beijing University of Posts and Telecommunications, Beijing 100876, ChinaWe propose a fully convolutional neural network based on the attention mechanism for 3D medical image segmentation tasks. It can adaptively learn to highlight the salient features of images that are useful for image segmentation tasks. Some prior methods enhance accuracy using multi-scale feature fusion or dilated convolution, which is basically artificial and lacks the flexibility of the model itself. Therefore, some works proposed the 2D attention gate module, but these works process 2D medical slice images, ignoring the correlation between 3D image sequences. In contrast, the 3D attention gate can comprehensively use the information of three dimensions of medical images. In this paper, we propose the Attention V-Net architecture, which uses the 3D attention gate module, and applied it to the left atrium segmentation framework based on semi-supervised learning. The proposed method is evaluated on the dataset of the 2018 left atrial challenge. The experimental results show that the Attention V-Net obtains improved performance under evaluation indicators, such as Dice, Jaccard, ASD (Average surface distance), and 95HD (Hausdorff distance). The result indicates that the model in this paper can effectively improve the accuracy of left atrial segmentation, therefore laying the foundation for subsequent work such as in atrial reconstruction. Meanwhile, our model is of great significance for assisting doctors in treating cardiovascular diseases.https://www.mdpi.com/2076-3417/12/8/37643D medical imageattention mechanismsemi-supervised learningleft atrial segmentation |
spellingShingle | Xiaoli Liu Ruoqi Yin Jianqin Yin Attention V-Net: A Modified V-Net Architecture for Left Atrial Segmentation Applied Sciences 3D medical image attention mechanism semi-supervised learning left atrial segmentation |
title | Attention V-Net: A Modified V-Net Architecture for Left Atrial Segmentation |
title_full | Attention V-Net: A Modified V-Net Architecture for Left Atrial Segmentation |
title_fullStr | Attention V-Net: A Modified V-Net Architecture for Left Atrial Segmentation |
title_full_unstemmed | Attention V-Net: A Modified V-Net Architecture for Left Atrial Segmentation |
title_short | Attention V-Net: A Modified V-Net Architecture for Left Atrial Segmentation |
title_sort | attention v net a modified v net architecture for left atrial segmentation |
topic | 3D medical image attention mechanism semi-supervised learning left atrial segmentation |
url | https://www.mdpi.com/2076-3417/12/8/3764 |
work_keys_str_mv | AT xiaoliliu attentionvnetamodifiedvnetarchitectureforleftatrialsegmentation AT ruoqiyin attentionvnetamodifiedvnetarchitectureforleftatrialsegmentation AT jianqinyin attentionvnetamodifiedvnetarchitectureforleftatrialsegmentation |