A computation-efficient network with feature aggregation for cancer subtype classification on histopathological images

作者：Zong Min Fan, Chaojie Zhang, Lulu Sun, Wade L. Thorstad, Hiram A. Gay, Xiaowei Wang, Hua Li · 发表于：Engineering Applications of Artificial Intelligence · 年份：2025 · DOI：10.1016/j.engappai.2025.111913 · 被引用次数：4 · 研究领域：AI in cancer detection、Radiomics and Machine Learning in Medical Imaging、Medical Imaging and Analysis

Histopathology whole-slide images (WSI) capture detailed structural and morphological features of tumor tissue, offering rich histological and molecular information. Deep learning (DL) methods have emerged to assist in automatically examining histopathology WSIs and supporting tumor classification. Traditional DL approaches for WSI images face challenges due to the intrinsic complexity of tumor tissue characteristics and the extremely large image size. Multiple instance learning (MIL) methods have been proposed to address these issues by splitting the WSI images into small non-overlapping tiles and aggregating predictions from selected informative tiles for the final classification outcome. However, MIL methods still face challenges such as the need for accurate pseudo-labels, the risk of losing local information, or the failure to learn explicit class-relevant information. To address these limitations, we propose a novel framework that uses a lightweight convolutional neural network (CNN)-based tile encoder (CTE) to extract local tile features and a Transformer-based feature aggregator (TFA) to fuse local features into a representative global feature for WSI classification. Three key contributions of our framework are as follows. Firstly, we design a two-stage training strategy that decouples a lightweight CTE pre-training (using sparsely sampled tiles for efficiency) and TFA fine-tuning (using all tiles for accuracy). It significantly reduces computational costs compared to...