A Fuzzy Transformer Fusion Network (FuzzyTransNet) for Medical Image Segmentation: The Case of Rectal Polyps and Skin Lesions
Source abstract
Abstract
Skin melanoma, one of the deadliest forms of cancer worldwide, demands precise diagnosis to mitigate cancer-related mortality. While histopathological examination, characterized by its cost-effectiveness and efficiency, remains the primary diagnostic approach, the development of an accurate detection system is pressing due to melanoma’s varying sizes, shapes, and indistinct boundaries shared with normal tissues. To address the efficient segmentation of skin melanoma, we propose an innovative hybrid neural network approach in this study. Initially, a fuzzy neural network is constructed using fuzzy logic to preprocess medical images, supplemented by wavelet transformation for image enhancement. Subsequently, the Swin Transformer V2 and ResNet50 networks are introduced to parallelly extract features and apply them to the task of skin melanoma segmentation. Extensive experimental comparisons are conducted with other classic and advanced medical segmentation algorithms on publicly available skin datasets, namely ISIC 2017 and ISIC 2018. Experimental results reveal that our method outperforms the optimal algorithms by 1.3% in the Dice coefficient and 1.3% in accuracy on the ISIC 2018 dataset. The evaluation metrics indicate the effectiveness of the constructed fuzzy block in identifying uncertain lesion boundaries, while the Transformer–CNN branch adeptly extracts global features while accurately capturing underlying details. Additionally, we successfully apply our method to colon polyp segmentation tasks with similar indistinct boundaries, achieving remarkable segmentation outcomes.
Source: crossref · Quality: baseline
Research context
OpenAlex topics: Cutaneous Melanoma Detection and Management · AI in cancer detection · Digital Imaging for Blood Diseases
SciVX discovery tags are terminology-based. OpenAlex topics are source-provided classifications.
Affiliations
School of Artificial Intelligence, Chongqing University of Technology, Chongqing 401135, China
Funding
Chongqing Natural Science Foundation Project · Science and Technology Project of Chongqing Municipal Education Commission
References (44)
Show source-provided references
1. Cancer statistics in China and United States, 2022: Profiles, trends, and determinants10.1097/CM9.0000000000002108 ↗
2. New artificial intelligence system: First validation study versus experienced endoscopists for colorectal polyp detection10.1136/gutjnl-2019-319914 ↗
3. Telerobotic neurovascular interventions with magnetic manipulation10.1126/scirobotics.abg9907 ↗
4. Swarming self-adhesive microgels enabled aneurysm on-demand embolization in physiological blood flow10.1126/sciadv.adf9278 ↗
5. Personalized magnetic tentacles for targeted photothermal cancer therapy in peripheral lungs10.1038/s44172-023-00098-9 ↗
6. Ronneberger, O., Fischer, P., and Brox, T. (2015, January 5–9). U-net: Convolutional networks for biomedical image segmentation. Proceedings of the Medical Image Computing and Computer-Assisted Intervention–MICCAI 2015: 18th International Conference, Munich, Germany.10.1007/978-3-319-24574-4_28 ↗
7. Unet++: Redesigning skip connections to exploit multiscale features in image segmentation10.1109/TMI.2019.2959609 ↗
8. He, K., Zhang, X., Ren, S., and Sun, J. (2016, January 27–30). Deep residual learning for image recognition. Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, Las Vegas, NV, USA.10.1109/CVPR.2016.90 ↗
9. H-DenseUNet: Hybrid densely connected UNet for liver and tumor segmentation from CT volumes10.1109/TMI.2018.2845918 ↗
10. Liang, J., Cao, J., Sun, G., Zhang, K., Van Gool, L., and Timofte, R. (2021, January 10–17). Swinir: Image restoration using swin transformer. Proceedings of the IEEE/CVF International Conference on Computer Vision, Montreal, BC, Canada.10.1109/ICCVW54120.2021.00210 ↗
11. Vaswani, A., Shazeer, N., Parmar, N., Uszkoreit, J., Jones, L., Gomez, A.N., Kaiser, Ł., and Polosukhin, I. (2017, January 4–9). Attention is all you need. Proceedings of the Advances in Neural Information Processing Systems, Long Beach, CA, USA.
12. Liu, Z., Lin, Y., Cao, Y., Hu, H., Wei, Y., Zhang, Z., Lin, S., and Guo, B. (2021, January 10–17). Swin transformer: Hierarchical vision transformer using shifted windows. Proceedings of the IEEE/CVF International Conference on Computer Vision, Montreal, BC, Canada.10.1109/ICCV48922.2021.00986 ↗
13. Liu, Z., Hu, H., Lin, Y., Yao, Z., Xie, Z., Wei, Y., Ning, J., Cao, Y., Zhang, Z., and Dong, L. (2022, January 18–24). Swin transformer v2: Scaling up capacity and resolution. Proceedings of the IEEE/CVF conference on Computer Vision and Pattern Recognition, New Orleans, LA, USA.10.1109/CVPR52688.2022.01170 ↗
14. SwinE-Net: Hybrid deep learning approach to novel polyp segmentation using convolutional neural network and Swin Transformer
15. Dosovitskiy, A., Beyer, L., Kolesnikov, A., Weissenborn, D., Zhai, X., Unterthiner, T., Dehghani, M., Minderer, M., Heigold, G., and Gelly, S. (2020). An image is worth 16 × 16 words: Transformers for image recognition at scale. arXiv.
16. Attention gated networks: Learning to leverage salient regions in medical images10.1016/j.media.2019.01.012 ↗
17. Adaptive wireless millirobotic locomotion into distal vasculature10.1038/s41467-022-32059-9 ↗
18. Müller, D., and Kramer, F. (2021). MIScnn: A framework for medical image segmentation with convolutional neural networks and deep learning. BMC Med. Imaging, 21.10.1186/s12880-020-00543-7 ↗
19. Semantic segmentation of computed tomography for radiotherapy with deep learning: Compensating insufficient annotation quality using contour augmentation
20. Segmenting brain tumors from FLAIR MRI using fully convolutional neural networks10.1016/j.cmpb.2019.05.006 ↗
21. Multimodal brain tumor image segmentation using WRN-PPNet10.1016/j.compmedimag.2019.04.001 ↗
22. Test-time adaptable neural networks for robust medical image segmentation10.1016/j.media.2020.101907 ↗
23. Hu, J., Shen, L., and Sun, G. (2018, January 18–22). Squeeze-and-excitation networks. Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, Salt Lake City, UT, USA.10.1109/CVPR.2018.00745 ↗
24. Woo, S., Park, J., Lee, J.Y., and Kweon, I.S. (2018, January 8–14). Cbam: Convolutional block attention module. Proceedings of the European Conference on Computer Vision (ECCV), Munich, Gtermany.10.1007/978-3-030-01234-2_1 ↗
25. Codella, N.C., Gutman, D., Celebi, M.E., Helba, B., Marchetti, M.A., Dusza, S.W., Kalloo, A., Liopyris, K., Mishra, N., and Kittler, H. (2018, January 4–7). Skin lesion analysis toward melanoma detection: A challenge at the 2017 international symposium on biomedical imaging (isbi), hosted by the international skin imaging collaboration (isic). Proceedings of the 2018 IEEE 15th International Symposium on Biomedical Imaging (ISBI 2018), Washington, DC, USA.10.1109/ISBI.2018.8363547 ↗
26. Codella, N., Rotemberg, V., Tschandl, P., Celebi, M.E., Dusza, S., Gutman, D., Helba, B., Kalloo, A., Liopyris, K., and Marchetti, M. (2019). Skin lesion analysis toward melanoma detection 2018: A challenge hosted by the international skin imaging collaboration (isic). arXiv.
27. Improving dermoscopic image segmentation with enhanced convolutional-deconvolutional networks10.1109/JBHI.2017.2787487 ↗
28. Dense deconvolutional network for skin lesion segmentation10.1109/JBHI.2018.2859898 ↗
29. Skin lesion segmentation in dermoscopy images via deep full resolution convolutional networks10.1016/j.cmpb.2018.05.027 ↗
30. Sarker, M.M.K., Rashwan, H.A., Akram, F., Banu, S.F., Saleh, A., Singh, V.K., Chowdhury, F.U., Abdulwahab, S., Romani, S., and Radeva, P. (2018, January 16–20). SLSDeep: Skin lesion segmentation based on dilated residual and pyramid pooling networks. Proceedings of the Medical Image Computing and Computer Assisted Intervention–MICCAI 2018: 21st International Conference, Granada, Spain.10.1007/978-3-030-00934-2_3 ↗
31. Step-wise integration of deep class-specific learning for dermoscopic image segmentation10.1016/j.patcog.2018.08.001 ↗
32. Zhang, Y., Liu, H., and Hu, Q. (October, January 27). Transfuse: Fusing transformers and cnns for medical image segmentation. Proceedings of the Medical Image Computing and Computer Assisted Intervention–MICCAI 2021: 24th International Conference, Strasbourg, France.10.1007/978-3-030-87193-2_2 ↗
33. Ms RED: A novel multi-scale residual encoding and decoding network for skin lesion segmentation10.1016/j.media.2021.102293 ↗
34. Wang, Y., Wei, Y., Qian, X., Zhu, L., and Yang, Y. (2020). DONet: Dual objective networks for skin lesion segmentation. arXiv.
35. CA-Net: Comprehensive attention convolutional neural networks for explainable medical image segmentation10.1109/TMI.2020.3035253 ↗
36. Touvron, H., Cord, M., Douze, M., Massa, F., Sablayrolles, A., and Jégou, H. (2021, January 18–24). Training data-efficient image transformers & distillation through attention. Proceedings of the International Conference on Machine Learning, Virtual.
37. Selvaraju, R.R., Cogswell, M., Das, A., Vedantam, R., Parikh, D., and Batra, D. (2017, January 22–29). Grad-cam: Visual explanations from deep networks via gradient-based localization. Proceedings of the IEEE International Conference on Computer Vision, Venice, Italy.10.1109/ICCV.2017.74 ↗
38. Automated polyp detection in colonoscopy videos using shape and context information10.1109/TMI.2015.2487997 ↗
39. Jha, D., Smedsrud, P.H., Riegler, M.A., Halvorsen, P., de Lange, T., Johansen, D., and Johansen, H.D. (2020, January 5–8). Kvasir-seg: A segmented polyp dataset. Proceedings of the MultiMedia Modeling: 26th International Conference, MMM 2020, Daejeon, Republic of Korea.10.1007/978-3-030-37734-2_37 ↗
40. WM-DOVA maps for accurate polyp highlighting in colonoscopy: Validation vs. saliency maps from physicians10.1016/j.compmedimag.2015.02.007 ↗
41. Jha, D., Smedsrud, P.H., Riegler, M.A., Johansen, D., De Lange, T., Halvorsen, P., and Johansen, H.D. (2019, January 9–11). Resunet++: An advanced architecture for medical image segmentation. Proceedings of the 2019 IEEE International Symposium on Multimedia (ISM), San Diego, CA, USA.10.1109/ISM46123.2019.00049 ↗
42. Fan, D.P., Ji, G.P., Zhou, T., Chen, G., Fu, H., Shen, J., and Shao, L. (2020, January 4–8). Pranet: Parallel reverse attention network for polyp segmentation. Proceedings of the International Conference on Medical Image Computing and Computer-Assisted Intervention, Lima, Peru.10.1007/978-3-030-59725-2_26 ↗
43. HMA-Net: A deep U-shaped network combined with HarDNet and multi-attention mechanism for medical image segmentation10.1002/mp.16065 ↗
44. Chen, J., Lu, Y., Yu, Q., Luo, X., Adeli, E., Wang, Y., Lu, L., Yuille, A.L., and Zhou, Y. (2021). Transunet: Transformers make strong encoders for medical image segmentation. arXiv.