Skip to content

Latest commit

 

History

History
72 lines (49 loc) · 2.86 KB

README.md

File metadata and controls

72 lines (49 loc) · 2.86 KB
Raw

🌻: LANet: Lightweight Attention Network for Medical Image Segmentation

This is the official implementation for article "LANet: Lightweight Attention Network for Medical Image Segmentation".

31.07.2024 - Sign the agreement, the article will be published on Springer

24.04.2024 - Attended the conference, waiting for final publication Image 0

11.03.2024 - The article is accepted and will be published after the conference which will be held on Azerbaijan.

20.12.2023 - The article is submitted in Springer proceedings of the ITTA-2024 conference (https://itta.cyber.az).

Overview

LANet, a Lightweight Attention Network, are presented in the paper and incorporates an Efficient Fusion Attention (EFA) block and an Adaptive Feature Fusion (AFF) decoding block. The model adopts MobileViT as a lightweight backbone network with a small number of parameters, facilitating easy training and faster predictive inference.

Image 1

Efficient Fusion Attention

The EFA block enhances the model's feature extraction capability by capturing task-relevant information while reducing redundancy in channel and spatial locations.

Image 5

Adaptive Feature Fusion

The AFF decoding block fuses the purified low-level features from the encoder with the sampled features from the decoder, enhancing the network's understanding and expression of input features.

Image 6

📝 Requirements

  • torch == 2.1.1+cu121
  • tensorboard == 2.11.2
  • numpy == 1.24.1
  • python == 3.9.18
  • torchvision == 0.16.1+cu121
  • ...

📊 Datasets

The efficiency of LANet was evaluated using four public datasets: kvasir-SEG, CVC-clinicDB, CVC-colonDB, and the Data Science Bowl 2018. All datasets used in paper are public, you can download online.

Split the datasets for train, validation and test with ratio 8:1:1

📈 Results

Quantitative results

Dataset mDC mIoU mRec mPrec
Kvasir-SEG 0.911 0.851 0.903 0.949
CVC_clinicDB 0.944 0.896 0.926 0.966
CVC_ColonDB 0.771 0.712 0.758 0.894
2018 DSB 0.930 0.871 0.918 0.946

Qualitative results

Image 2

Ablation study

Image 3

✒️ For citation

@inproceedings{tang2024lanet, title={LANet: Lightweight Attention Network for Medical Image Segmentation}, author={Tang, Yi and Pertsau, Dmitry and Zhao, Di and Kupryianava, Dziana and Tatur, Mikhail}, booktitle={International Conference on Information Technologies and Their Applications}, pages={213--227}, year={2024}, organization={Springer} }

❗ 👀 The codes can not be used for commercial purposes!!!