AI Insight
PathRWKV is a new computational model designed to analyze whole slide images (WSI) used in cancer diagnosis more efficiently. The model addresses key limitations in current approaches by using an asymmetric structure that allows fast training while maintaining constant memory usage during inference, incorporates position encoding to preserve spatial relationships between tissue samples, and integrates multi-scale feature modeling to capture comprehensive tissue representations. Testing on 29,073 WSIs across 11 datasets showed PathRWKV outperformed 11 existing state-of-the-art methods on 10 datasets.
Why it matters
This technology could significantly improve the efficiency and accuracy of digital pathology workflows in clinical settings, potentially enabling faster cancer diagnoses while reducing computational costs. The model's ability to work with limited training data and maintain spatial context makes it particularly valuable for medical institutions with smaller datasets or limited computational resources.
Understand the Science
⚠️ Preprint – Noch nicht peer-reviewed
Dieser Artikel wurde noch nicht von unabhängigen Experten begutachtet. Die Ergebnisse sind vorläufig und sollten mit Vorsicht interpretiert werden.
-cross
Abstract: Whole Slide Imaging (WSI) has become a gold standard in cancer diagnosis, inspecting multi-scale information from cellular to tissue levels. Processing an entire WSI directly is infeasible due to GPU memory constraints; thus, Multiple Instance Learning (MIL) has emerged as the standard solution by partitioning WSIs into tiles. While recent two-stage MIL frameworks partially achieve memory efficiency by decoupling tile-level extraction from slide-level modeling, they still face four limitations: (1) the conflict between training throughput and inference memory efficiency, (2) the high susceptibility to overfitting on small-scale WSI datasets with sparse supervision, (3) the disruption of spatial structural integrity during sampling-based training, and (4) the inadequate modeling of multi-scale feature interactions within long sequences. We therefore introduce PathRWKV, a novel State Space Model designed for efficient and robust WSI analysis. To resolve the computational trade-off, we propose an asymmetric structure utilizing max pooling aggregation, enabling parallelized training for high throughput and recurrent inference with constant (O(1)) memory complexity. To mitigate overfitting, we employ random sampling to enhance data diversity, with a multi-task learning module to regularize feature learning on limited data. To restore spatial context, we introduce 2D sinusoidal position encoding to perceive the relative locations of tissue tiles. To capture comprehensive representations, we integrate TimeMix and ChannelMix modules, enabling dynamic multi-scale feature modeling across temporal and spatial dimensions. Experiments on 29,073 WSIs across 11 datasets demonstrate that PathRWKV outperforms 11 state-of-the-art methods on 10 datasets, establishing it as a scalable and solution with application potential.
Source: PathRWKV: Enhancing Whole Slide Image Inference with Asymmetric Recurrent Modeling