Diagnosis of colorectal cancer using residual transformer with mixed attention and explainable AI
Poonam Sharma, Bhisham Sharma, Ajit Noonia, Dhirendra Prasad Yadav, Panos Liatsis, Siamak Pedrammehr, Siamak Pedrammehr, Siamak Pedrammehr, Siamak Pedrammehr
TL;DR
This paper introduces a new AI model for diagnosing colorectal cancer that combines deep learning and attention mechanisms to improve accuracy and provide explainable results.
Contribution
The novel RNTNet model integrates ResNeXt and a vision transformer with mixed attention and Grad-CAM for enhanced CRC diagnosis and interpretability.
Findings
RNTNet achieved 97.96% accuracy on the KvasirV1 dataset and 98.20% on the Kather dataset.
The model's AUC values of 0.9895 and 0.9937 on KvasirV1 and Kather datasets confirm its high diagnostic performance.
Abstract
Colorectal cancer (CRC) is the leading cause of cancer disease and poses a significant threat to global health. Although deep learning models have been utilized to accurately diagnose CRC, they still face challenges in capturing the global correlations of spatial features, especially in complex textures and morphologically similar features. To overcome these challenges, we propose a hybrid model using a residual network and transformer encoder with mixed attention. The Residual Next Transformer Network (RNTNet) extracts spatial features from CRC images using ResNeXt. ResNeXt utilizes group convolution and skip connections to capture fine-grained features. Furthermore, a vision transformer (ViT) encoder containing a mixed attention block is designed using multiscale feature aggregation to provide global attention to the spatial features. In addition, a Grad-CAM module is added to…
Genes, proteins, chemicals, diseases, species, mutations and cell lines named across the full text — each resolved to its canonical identifier and authoritative record.
Click any figure to enlarge with its caption.
Figure 1
Figure 2
Figure 3
Figure 4
Figure 5
Figure 6
Figure 7
Figure 8
Figure 9
Figure 10
Figure 11
Figure 12
Figure 13
Figure 14
Figure 15
Figure 16
Figure 17
Figure 18
Figure 19
Figure 20
Figure 21
Figure 22
Figure 23
Figure 24
Figure 25
Figure 26
Figure 27
Figure 28
Figure 29
Figure 30
Figure 31
Figure 32
Figure 33
Figure 34
Figure 35
Figure 36
Figure 37
Figure 38
Figure 39
Figure 40
Figure 41
Figure 42
Figure 43
Figure 44
Figure 45
Figure 46
Figure 47
Figure 48
Figure 49
Figure 50Peer Reviews
No public reviews on file for this paper yet. If you reviewed it on a platform where reviews are public (OpenReview, ICLR, NeurIPS, ICML), you can paste yours below so the community can read it here.
Videos
No videos yet. Explain this paper in a talk, walkthrough, or lecture? Add one.
Taxonomy
TopicsAI in cancer detection · Colorectal Cancer Screening and Detection · Advanced Neural Network Applications