All Projects → superxuang → amta-net

superxuang / amta-net

Licence: MIT license
Asymmetric Multi-Task Attention Network for Prostate Bed Segmentation in CT Images

Programming Languages

python
139335 projects - #7 most used programming language

Projects that are alternatives of or similar to amta-net

DAF3D
Deep Attentive Features for Prostate Segmentation in 3D Transrectal Ultrasound
Stars: ✭ 60 (+130.77%)
Mutual labels:  attention-mechanism, prostate-segmentation
Multi-task-Conditional-Attention-Networks
A prototype version of our submitted paper: Conversion Prediction Using Multi-task Conditional Attention Networks to Support the Creation of Effective Ad Creatives.
Stars: ✭ 21 (-19.23%)
Mutual labels:  attention-mechanism, multi-task-learning
Keras Attention Mechanism
Attention mechanism Implementation for Keras.
Stars: ✭ 2,504 (+9530.77%)
Mutual labels:  attention-mechanism
question-generation
Neural Models for Key Phrase Detection and Question Generation
Stars: ✭ 29 (+11.54%)
Mutual labels:  attention-mechanism
Aoanet
Code for paper "Attention on Attention for Image Captioning". ICCV 2019
Stars: ✭ 242 (+830.77%)
Mutual labels:  attention-mechanism
Lightnetplusplus
LightNet++: Boosted Light-weighted Networks for Real-time Semantic Segmentation
Stars: ✭ 218 (+738.46%)
Mutual labels:  attention-mechanism
NeuralMerger
Yi-Min Chou, Yi-Ming Chan, Jia-Hong Lee, Chih-Yi Chiu, Chu-Song Chen, "Unifying and Merging Well-trained Deep Neural Networks for Inference Stage," International Joint Conference on Artificial Intelligence (IJCAI), 2018
Stars: ✭ 20 (-23.08%)
Mutual labels:  multi-task-learning
Guided Attention Inference Network
Contains implementation of Guided Attention Inference Network (GAIN) presented in Tell Me Where to Look(CVPR 2018). This repository aims to apply GAIN on fcn8 architecture used for segmentation.
Stars: ✭ 204 (+684.62%)
Mutual labels:  attention-mechanism
SA-DL
Sentiment Analysis with Deep Learning models. Implemented with Tensorflow and Keras.
Stars: ✭ 35 (+34.62%)
Mutual labels:  attention-mechanism
Linformer Pytorch
My take on a practical implementation of Linformer for Pytorch.
Stars: ✭ 239 (+819.23%)
Mutual labels:  attention-mechanism
Transformers-RL
An easy PyTorch implementation of "Stabilizing Transformers for Reinforcement Learning"
Stars: ✭ 107 (+311.54%)
Mutual labels:  attention-mechanism
Self Attention Cv
Implementation of various self-attention mechanisms focused on computer vision. Ongoing repository.
Stars: ✭ 209 (+703.85%)
Mutual labels:  attention-mechanism
Dalle Pytorch
Implementation / replication of DALL-E, OpenAI's Text to Image Transformer, in Pytorch
Stars: ✭ 3,661 (+13980.77%)
Mutual labels:  attention-mechanism
lstm-attention
Attention-based bidirectional LSTM for Classification Task (ICASSP)
Stars: ✭ 87 (+234.62%)
Mutual labels:  attention-mechanism
Neat Vision
Neat (Neural Attention) Vision, is a visualization tool for the attention mechanisms of deep-learning models for Natural Language Processing (NLP) tasks. (framework-agnostic)
Stars: ✭ 213 (+719.23%)
Mutual labels:  attention-mechanism
TianChi AIEarth
TianChi AIEarth Contest Solution
Stars: ✭ 57 (+119.23%)
Mutual labels:  attention-mechanism
Linear Attention Transformer
Transformer based on a variant of attention that is linear complexity in respect to sequence length
Stars: ✭ 205 (+688.46%)
Mutual labels:  attention-mechanism
Triplet Attention
Official PyTorch Implementation for "Rotate to Attend: Convolutional Triplet Attention Module." [WACV 2021]
Stars: ✭ 222 (+753.85%)
Mutual labels:  attention-mechanism
Attentionalpoolingaction
Code/Model release for NIPS 2017 paper "Attentional Pooling for Action Recognition"
Stars: ✭ 248 (+853.85%)
Mutual labels:  attention-mechanism
Video-Description-with-Spatial-Temporal-Attention
[ACM MM 2017 & IEEE TMM 2020] This is the Theano code for the paper "Video Description with Spatial Temporal Attention"
Stars: ✭ 53 (+103.85%)
Mutual labels:  attention-mechanism

AMTA-Net

This is a python (PyTorch) implementation of Asymmetrical Multi-Task Attention Network proposed in our MICCAI2020 paper "Asymmetrical Multi-Task Attention U-Net for the Segmentation of Prostate Bed in CT Image". An exhaustive extension of this work is published on Medical Image Analysis journal paper "Asymmetric Multi-Task Attention Network for Prostate Bed Segmentation in Computed Tomography Images".

Network architecture

Inner structure of attention module

Citation

@inproceedings{Xu2020PBSeg, 
  title = {Asymmetrical multi-task attention u-net for the segmentation of prostate bed in CT image},
  author = {Xuanang Xu and Chunfeng Lian and Shuai Wang and Andrew Wang and Trevor Royce and Ronald Chen and Jun Lian and Dinggang Shen},
  booktitle={International Conference on Medical Image Computing and Computer-Assisted Intervention},
  year = {2020},
  pages = {470--479},
  organization={Springer}
}

@article{Xu2021PBSeg,
  title = {Asymmetric multi-task attention network for prostate bed segmentation in computed tomography images},
  author = {Xuanang Xu and Chunfeng Lian and Shuai Wang and Tong Zhu and Ronald C. Chen and Andrew Z. Wang and Trevor J. Royce and Pew-Thian Yap and Dinggang Shen and Jun Lian},
  year = {2021},
  publisher = {Elsevier},
  pages = {102116},
  journal = {Medical Image Analysis}
}
Note that the project description data, including the texts, logos, images, and/or trademarks, for each open source project belongs to its rightful owner. If you wish to add or remove any projects, please contact us at [email protected].