Benchmarking Feature Extractors for Prostate Cancer Detection Using Graph Attention Networks: A Focus on Foundation Models
Accurate prostate cancer detection from Whole-Slide Images (WSIs) is critical for improving diagnostics. This study benchmarks 10 foundation models as feature extractors for graph-based classification using Graph Attention Networks (GATs). We compare them to CNN-based extractors, including ResNet-50, VGG-19, and DenseNet-121, on the AGGC22 dataset. Results show that Virchow2 and UNI achieve the highest F1-scores, while Prov-GigaPath performs best on rare Gleason grades. These findings highlight the potential of foundation models for improving WSI-based cancer classification.
Authors
- Vincenzo Della Mea (ORCID: https://orcid.org/0000-0002-0144-3802)
- Kevin Roitero (ORCID: https://orcid.org/0000-0002-9191-3280)
- Hafsa Akebli (ORCID: https://orcid.org/0009-0002-3879-876X)
Institutions
- University of Udine (IT)
Publication Details
- Published
- 2025-06-18
- DOI
- https://doi.org/10.1109/ichi64645.2025.00099
- Primary Topic
- AI in cancer detection
- Type
- article
- Field-Weighted Citation Impact
- 0.00