Skip to main navigation Skip to search Skip to main content

A lightweight multi-scale multi-angle dynamic interactive transformer-CNN fusion model for 3D medical image segmentation

  • Xin Hua
  • , Zhijiang Du
  • , Hongjian Yu*
  • , Jixin Ma
  • , Fanjun Zheng
  • , Chen Zhang
  • , Qiaohui Lu
  • , Hui Zhao
  • *Corresponding author for this work
  • Harbin Institute of Technology
  • General Hospital of People's Liberation Army

Research output: Contribution to journalArticlepeer-review

Abstract

Combining Convolutional Neural Network(CNN) and Transformer has become one of the mainstream methods for three-dimensional (3D) medical image segmentation. However, the complexity and diversity of target forms in 3D medical images require models to capture complex feature information for segmentation, resulting in an excessive number of parameters which are not conducive to training and deployment. Therefore, we have developed a lightweight 3D multi-target semantic segmentation model. In order to enhance contextual texture connections and reinforce the expression of detailed feature information, we designed a multi-scale and multi-angle feature interaction module to enhance feature representation by interacting multi-scale features from different perspectives. To address the issue of attention collapse in Transformers, leading to the neglect of other detailed feature learning, we utilized local features as dynamic parameters to interact with global features, dynamically grouping and learning critical features from global features, thereby enhancing the model's ability to learn detailed features. While ensuring the segmentation capability of the model, we aimed to keep the model lightweight, resulting in a total of 9.63 M parameters. Extensive experiments were conducted on public datasets ACDC and Brats2018, as well as a private dataset, Temporal Bone CT. The results indicate that our proposed model is more competitive compared to the latest techniques in 3D medical image segmentation.

Original languageEnglish
Article number128417
JournalNeurocomputing
Volume608
DOIs
StatePublished - 1 Dec 2024

Keywords

  • 3D Medical image segmentation,Light-weight,Convolutional Neural Network,Transformer

Fingerprint

Dive into the research topics of 'A lightweight multi-scale multi-angle dynamic interactive transformer-CNN fusion model for 3D medical image segmentation'. Together they form a unique fingerprint.

Cite this