PEDRA-EFB0: colorectal cancer prognostication using deep learning with patch embeddings and dual residual attention.
Authors
Affiliations (4)
Affiliations (4)
- School of Internet of Things Engineering, Jiangnan University, Wuxi, 214122, China.
- Key Laboratory of Light Industry, Jiangnan University, Wuxi, 214122, China. [email protected].
- Key Laboratory of Light Industry, Jiangnan University, Wuxi, 214122, China.
- Radiol Dept, Jiangnan Univ, Affiliated Hosp, Wuxi, 214122, Jiangsu, People's Republic of China.
Abstract
In computer-aided diagnosis systems, precise feature extraction from CT scans of colorectal cancer using deep learning is essential for effective prognosis. However, existing convolutional neural networks struggle to capture long-range dependencies and contextual information, resulting in incomplete CT feature extraction. To address this, the PEDRA-EFB0 architecture integrates patch embeddings and a dual residual attention mechanism for enhanced feature extraction and survival prediction in colorectal cancer CT scans. A patch embedding method processes CT scans into patches, creating positional features for global representation and guiding spatial attention computation. Additionally, a dual residual attention mechanism during the upsampling stage selectively combines local and global features, enhancing CT data utilization. Furthermore, this paper proposes a feature selection algorithm that combines autoencoders and entropy technology, encoding and compressing high-dimensional data to reduce redundant information and using entropy to assess the importance of features, thereby achieving precise feature selection. Experimental results indicate the PEDRA-EFB0 model outperforms traditional methods on colorectal cancer CT metrics, notably in C-index, BS, MCC, and AUC, enhancing survival prediction accuracy. Our code is freely available at https://github.com/smile0208z/PEDRA .