cs.CVMar 10, 2026

MIL-PF: Multiple Instance Learning on Precomputed Features for Mammography Classification

Authors: Nikola Jovišić, Milica Škipina, Nicola Dall'Asen, Dubravko Ćulibrk

Organizations: Institute for AI R&D of Serbia, Serbia · Faculty of Technical Sciences, University of Novi Sad, Serbia · University of Trento, Italy · University of Pisa, Italy · Fondazione Bruno Kessler, Italy

Abstract

Modern foundation models provide highly expressive visual representations, yet adapting them to high-resolution medical imaging remains challenging due to limited annotations and weak supervision. Mammography, in particular, is characterized by large images, variable multi-view studies and predominantly breast-level labels, making end-to-end fine-tuning computationally expensive and often impractical. We propose Multiple Instance Learning on Precomputed Features (MIL-PF), a scalable framework that combines frozen foundation encoders with a lightweight MIL head for mammography classification. By precomputing the semantic representations and training only a small task-specific aggregation module (40k parameters), the method enables efficient experimentation and adaptation without retraining large backbones. The architecture explicitly models the global tissue context and the sparse local lesion signals through attention-based aggregation. MIL-PF achieves state-of-the-art classification performance at clinical scale while substantially reducing training complexity. We release the code for full reproducibility.

Figures & tables

Explore similar work

CardsList
  1. Benchmarking the Robustness of Foundation Models for Mammography under Domain Shift

    Jul 11, 2026Giang Nguyen, Raghav Mehta, Emma A. M. Stanley +4Distribution Shift RobustnessMedical Imaging Foundation Models

  2. Mixture of Mini Experts: Overcoming the Linear Layer Bottleneck in Multiple Instance Learning

    Mar 23, 2026Daniel Shao, Joel Runevic, Richard J. Chen +4Multiple Instance LearningHistopathology Image Classification

  3. Attend what matters: Leveraging vision foundational models for breast cancer classification using mammograms

    Apr 21, 2026Samyak Sanghvi, Piyush Miglani, Sarvesh Shashikumar +3Medical ImagingVision Foundation Model Adaptation