TMTrans: texture mixed transformers for medical image segmentation

Article type: Research Article

Authors: Chen, Lifang | Wang, Tao^{; *} | Ge, Hongze

Affiliations: Jiangnan University, Wuxi, China

Correspondence: [*] Corresponding author. E-mail: [email protected].

Abstract: Accurate segmentation of skin cancer is crucial for doctors to identify and treat lesions. Researchers are increasingly using auxiliary modules with Transformers to optimize the model’s ability to process global context information and reduce detail loss. Additionally, diseased skin texture differs from normal skin, and pre-processed texture images can reflect the shape and edge information of the diseased area. We propose TMTrans (Texture Mixed Transformers). We have innovatively designed a dual axis attention mechanism (IEDA-Trans) that considers both global context and local information, as well as a multi-scale fusion (MSF) module that associates surface shape information with deep semantics. Additionally, we utilize TE(Texture Enhance) and SK(Skip connection) modules to bridge the semantic gap between encoders and decoders and enhance texture features. Our model was evaluated on multiple skin datasets, including ISIC 2016/2017/2018 and PH2, and outperformed other convolution and Transformer-based models. Furthermore, we conducted a generalization test on the 2018 DSB dataset, which resulted in a nearly 2% improvement in the Dice index, demonstrating the effectiveness of our proposed model.

Keywords: U-Net, texture, transformer, skin lesion, medical image segmentation

DOI: 10.3233/AIC-230089

Journal: AI Communications, vol. 36, no. 4, pp. 325-340, 2023