One Model to Magnify Them All: Efficient Scale-Invariant Histopathology via Conditional Normalization and Continuous Magnification Training
2026-08-10 • Computer Vision and Pattern Recognition
Computer Vision and Pattern Recognition
AI summaryⓘ
The authors address the problem that digital pathology images come at various zoom levels, making it hard for AI models to work well across all scales. They introduce Conditional Layer Normalization (CLN), a simple method that adjusts the model based on the image's pixel size, letting one model handle many zoom levels seamlessly. Their approach works well on prostate cancer images, matching or beating models trained specifically for one zoom level, and cuts down the need to train multiple models. This makes both training and diagnosis faster without extra computational cost.
Whole Slide ImagesDigital HistopathologyDeep LearningMagnification InvarianceConditional Layer NormalizationConvolutional Neural NetworksMulti-scale LearningProstate Cancer DatasetAffine NormalizationInference Generalization
Authors
Agnieszka Florkowska, Henning Müller, Marek Wodzinski
Abstract
Whole slide images (WSIs) in digital histopathology are acquired at discrete magnification levels encoding complementary diagnostic information from global tissue architecture to fine-grained cellular morphology. Yet, deep learning models remain sensitive to scale variation. Existing magnification-invariant methods rely on multi-scale architectures at predefined discrete resolutions, while in clinical deployment the acquisition magnification varies continuously, rarely aligns with a model's fixed training resolution, and intermediate scales are common, so robust coverage otherwise demands a costly ensemble of magnification-specific models. We propose Conditional Layer Normalization (CLN), a lightweight mechanism that generates affine normalization parameters from input pixel size via a small MLP, integrated into standard CNN architectures for both WSI classification and segmentation. Trained on patches sampled continuously across a range of pixel sizes, the model decouples inference from scanner-dependent magnification and generalizes to arbitrary, previously unseen scales at test time. On the PANDA prostate cancer dataset, our approach on average matches or exceeds independently trained single-magnification models and ranks among the top three performers at every evaluated magnification, including those unseen during training. This collapses a five-model ensemble into a single network and reduces training, and inference cost roughly 4-5 times, while leaving the multiply-accumulate count unchanged. The code is available at: https://github.com/aflorkowska/OneModelToMagnifyThemAll.