Block-fused attention-driven adaptively-pooled ResNet model for improved cervical cancer classification
摘要
Cervical cancer is the second most common cancer among women and a leading cause of mortality. Many attempts have been made to develop an effective Computer-Aided Diagnosis (CAD) system, however, their performance remains limited. Using pretrained ResNet-50/101/152, we propose a novel CAD system that significantly outperforms prior approaches. Our model has three key components. First, as common in a lot of cancer classification systems, we extract the detailed features (color, edges, and texture) from early convolution blocks and the abstract features (shapes and objects) from later blocks, as both are equally important. Second, a non-parametric 3D attention module is embedded within each block for feature enhancement. Embedding attention within convolution blocks is also common in cancer classification, however, the way we integrate it with the dual-level feature extraction technique is new. Third, we design a theoretically motivated adaptive pooling strategy for feature selection that applies Global Max Pooling to detailed features and Global Average Pooling to abstract features. Typically, these pooling layers are interspersed between convolution blocks, as common in different cancer systems. However, the way we adapt these pooling strategy for dual-level feature extraction technique is also new. These components form our proposed Block-Fused Attention-Driven Adaptively-Pooled ResNet (BF-AD-AP-ResNet) model. To further strengthen learning, we introduce a Tri-Stream model, as common in the cancer literature, which unifies the enhanced features from three BF-AD-AP-ResNets. An SVM classifier is employed for final classification. We evaluate our models on two public datasets, IARC and AnnoCerv. On IARC, the base ResNets achieve an average performance of