Search papers, labs, and topics across Lattice.
This paper introduces Conditional Layer Normalization (CLN), a novel mechanism that generates affine normalization parameters based on input pixel size, enabling deep learning models to effectively handle continuous variations in magnification levels in whole slide images (WSIs) for histopathology. By training on patches sampled across a range of pixel sizes, the model achieves robust performance in both classification and segmentation tasks without the need for multiple magnification-specific models. The approach not only matches or exceeds the performance of traditional single-magnification models but also significantly reduces training and inference costs by consolidating a five-model ensemble into a single network.
A single model can now outperform ensembles in histopathology, achieving scale-invariance while slashing training costs by up to 80%.
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.