# RetMIL: Retentive Multiple Instance Learning for Histopathological Whole Slide Image Classification

Source: [RetMIL: Retentive Multiple Instance Learning for Histopathological Whole Slide Image Classification](<https://www.sota2.com/research/paper/68fa72961e209131cea57563>)

Collection index: [llms.txt](<https://www.sota2.com/research/llms.txt>)

Data updated: 2026-01-04T04:48:50.478Z

Authors: Hongbo Chu, Qiehe Sun, Jiawen Li, Yuxuan Chen, Lizhong Zhang, Tian Guan, Anjia Han, Yonghong He

Published: 2024-03-16T08:50:47.000Z

## Abstract

Histopathological whole slide image (WSI) analysis with deep learning has become a research focus in computational pathology. The current paradigm is mainly based on multiple instance learning (MIL), in which approaches with Transformer as the backbone are well discussed. These methods convert WSI tasks into sequence tasks by representing patches as tokens in the WSI sequence. However, the feature complexity brought by high heterogeneity and the ultra-long sequences brought by gigapixel size makes Transformer-based MIL suffer from the challenges of high memory consumption, slow inference speed, and lack of performance. To this end, we propose a retentive MIL method called RetMIL, which processes WSI sequences through hierarchical feature propagation structure. At the local level, the WSI sequence is divided into multiple subsequences. Tokens of each subsequence are updated through a parallel linear retention mechanism and aggregated utilizing an attention layer. At the global level, subsequences are fused into a global sequence, then updated through a serial retention mechanism, and finally the slide-level representation is obtained through a global attention pooling. We conduct experiments on two public CAMELYON and BRACS datasets and an public-internal LUNG dataset, confirming that RetMIL not only achieves state-of-the-art performance but also significantly reduces computational overhead. Our code will be accessed shortly.

## Recorded benchmark results

A research result applies to its stated dataset, split, metric and evaluation configuration. Compare like-for-like settings, preserve the metric's unit and direction, and distinguish paper-reported results from independent reproduction. The best result recorded here is not a claim of exhaustive, current global coverage.

- [WSI Classification on CAMELYON](<https://www.sota2.com/research/sota/wsi-classification-on-camelyon.md>): WSI Classification; CAMELYON; F1 Score: 87.24.

- [WSI Classification on LUNG](<https://www.sota2.com/research/sota/wsi-classification-on-lung.md>): WSI Classification; Lung; F1 Score: 91.51.

## Original sources and implementation

[Original paper](<https://arxiv.org/abs/2403.10858>)

[PDF](<https://arxiv.org/pdf/2403.10858.pdf>)
