Self-supervised XCiT encoders for prostate histopathology, pretrained on 48 million tissue tiles so that features cluster by histological pattern.
No providers recorded yet. Browse all providers
Annotating a digital pathology archive is a tile-by-tile problem: a slide cut into 512-pixel squares yields tens of thousands of images, and a pathologist has to look at each one. HistoEncoder makes it a cluster-by-cluster problem instead. Because its self-supervised objective pushes visually similar histology to nearby points in feature space, tiles can be clustered first and labelled in bulk — inspect a sample from each cluster, then apply that label to every tile it contains. On the Radboud split of the PANDA cohort, 68.8% of tiles fell into clusters that were more than 90% pure for cancerous versus benign tissue, and 2.1 million tiles were sorted after visually checking 256 clusters.
The models were built by Joona Pohjonen and colleagues at the
University of Helsinki and Helsinki University
Hospital, with the iCAN precision cancer medicine flagship and Skåne University
Hospital, and trained on Europe's LUMI supercomputer. Two checkpoints were released
publicly in 2023 — prostate_small and prostate_medium — with the describing preprint
following in November 2024.
HistoEncoder is deliberately narrower than the pan-cancer encoders it sits alongside. Where UNI, Virchow and Phikon span dozens of tissue sites, HistoEncoder is trained on prostate tissue only, trading breadth for a representation tuned to one organ's morphology and to the clinical questions asked of it.
pip install histoencoder
and expose an extract/cluster command line; the paper's downstream results were
produced with the data and compute of a single laptop.Both checkpoints are cross-covariance image transformers (XCiT), chosen because their
attention is linear in the number of tokens and so handles high-resolution tiles
efficiently: prostate_small is an XCiT-S12 with 25.9 million parameters in the encoder,
prostate_medium the deeper and wider XCiT-M24 at roughly 84 million. Pretraining used
DINO on HelsinkiProstate — 11,226 needle biopsy and prostatectomy slides from 1,307
Helsinki University Hospital patients treated between 2013 and 2021, tiled at 20x, 10x
and 5x, cleaned with HistoPrep, and balanced to 48 million tile images. Evaluation used
cohorts the encoder never saw: PESO and the Radboud and Karolinska splits of PANDA
(2,113 patients), plus HelsinkiTMA, 1,769 tissue microarray spots from 432 patients with
a median 19-year follow-up. Fine-tuned prostate-s beat an identically fine-tuned
natural-image XCiT-S12 on every cohort and at every parameter and data budget tested; it
reached comparable accuracy from 1,024 tile images to what the natural-image baseline
needed hundreds of thousands to match. In the survival analysis, adding 32-cluster
patient-level fractions raised concordance over the baseline model in 84.9%, 89.2% and
67.4% of 1,000 stratified splits for Gleason, CAPRA-S and MSKCC-S respectively.
The primary use is annotating and mining prostate pathology archives: extract features, cluster, label clusters, and either ship the labelled dataset or fine-tune a classifier on it for cancer detection and Gleason grading. The second workflow converts slides into patient-level histomic vectors that join clinical variables in prognostic models, including prostate cancer-specific mortality. Because features can be extracted for arbitrary tissue regions, the encoders also provide a way to attach morphology to region-level assays such as spatial transcriptomics or tissue microarray spots.
HistoEncoder is a concrete argument for organ-specific pretraining: a single institution's biopsy archive, run once on a supercomputer, yields an encoder that a laboratory can then apply and fine-tune on ordinary hardware. The interpretability angle matters too, since the feature clusters correspond to morphologies a pathologist can inspect and reason about, rather than to an opaque score. The limitations are stated plainly in the paper: coverage is prostate tissue only, with other tissue types framed as future work; the survival analysis has no external validation cohort; and the work remains a preprint. The weights are distributed through personal file-hosting links hardcoded into the loader rather than a model hub, which makes long-term availability a practical concern.
Much of this page is generated or calculated automatically. Flag anything that looks off and we will re-run it.