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

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Jovišić, Nikola, Škipina, Milica, Dall'Asen, Nicola, Ćulibrk, Dubravko
Natura: Preprint
Pubblicazione: 2026
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866915850205790208
author Jovišić, Nikola
Škipina, Milica
Dall'Asen, Nicola
Ćulibrk, Dubravko
author_facet Jovišić, Nikola
Škipina, Milica
Dall'Asen, Nicola
Ćulibrk, Dubravko
contents 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.
format Preprint
id arxiv_https___arxiv_org_abs_2603_09374
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle MIL-PF: Multiple Instance Learning on Precomputed Features for Mammography Classification
Jovišić, Nikola
Škipina, Milica
Dall'Asen, Nicola
Ćulibrk, Dubravko
Computer Vision and Pattern Recognition
Artificial Intelligence
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.
title MIL-PF: Multiple Instance Learning on Precomputed Features for Mammography Classification
topic Computer Vision and Pattern Recognition
Artificial Intelligence
url https://arxiv.org/abs/2603.09374