AI Navigate

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

arXiv cs.CV / 3/11/2026

Ideas & Deep AnalysisTools & Practical UsageModels & Research

Key Points

  • MIL-PF is a new scalable framework for mammography classification that uses Multiple Instance Learning on Precomputed Features, addressing challenges of large image sizes and limited annotations in medical imaging.
  • The approach leverages frozen foundation models for feature extraction and trains only a small, lightweight MIL aggregation head, which significantly reduces computational expense and enables efficient adaptation.
  • MIL-PF uses attention-based aggregation to model both global tissue context and sparse local lesion signals, resulting in state-of-the-art classification performance at a clinical scale.
  • This framework facilitates faster experimentation without retraining large backbone models and contributes to reproducibility by releasing the full code.
  • MIL-PF represents a practical innovation bridging high-capacity foundation models and medical imaging tasks with weak supervision and limited annotation data.

Computer Science > Computer Vision and Pattern Recognition

arXiv:2603.09374 (cs)
[Submitted on 10 Mar 2026]

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

View a PDF of the paper titled MIL-PF: Multiple Instance Learning on Precomputed Features for Mammography Classification, by Nikola Jovi\v{s}i\'c and 3 other authors
View PDF HTML (experimental)
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.
Comments:
Subjects: Computer Vision and Pattern Recognition (cs.CV); Artificial Intelligence (cs.AI)
Cite as: arXiv:2603.09374 [cs.CV]
  (or arXiv:2603.09374v1 [cs.CV] for this version)
  https://doi.org/10.48550/arXiv.2603.09374
Focus to learn more
arXiv-issued DOI via DataCite

Submission history

From: Nikola Jovišić [view email]
[v1] Tue, 10 Mar 2026 08:49:33 UTC (16,451 KB)
Full-text links:

Access Paper:

    View a PDF of the paper titled MIL-PF: Multiple Instance Learning on Precomputed Features for Mammography Classification, by Nikola Jovi\v{s}i\'c and 3 other authors
  • View PDF
  • HTML (experimental)
  • TeX Source
Current browse context:
cs.CV
< prev   |   next >
Change to browse by:

References & Citations

export BibTeX citation Loading...

BibTeX formatted citation

×
Data provided by:

Bookmark

BibSonomy logo Reddit logo
Bibliographic Tools

Bibliographic and Citation Tools

Bibliographic Explorer Toggle
Bibliographic Explorer (What is the Explorer?)
Connected Papers Toggle
Connected Papers (What is Connected Papers?)
Litmaps Toggle
Litmaps (What is Litmaps?)
scite.ai Toggle
scite Smart Citations (What are Smart Citations?)
Code, Data, Media

Code, Data and Media Associated with this Article

alphaXiv Toggle
alphaXiv (What is alphaXiv?)
Links to Code Toggle
CatalyzeX Code Finder for Papers (What is CatalyzeX?)
DagsHub Toggle
DagsHub (What is DagsHub?)
GotitPub Toggle
Gotit.pub (What is GotitPub?)
Huggingface Toggle
Hugging Face (What is Huggingface?)
Links to Code Toggle
Papers with Code (What is Papers with Code?)
ScienceCast Toggle
ScienceCast (What is ScienceCast?)
Demos

Demos

Replicate Toggle
Replicate (What is Replicate?)
Spaces Toggle
Hugging Face Spaces (What is Spaces?)
Spaces Toggle
TXYZ.AI (What is TXYZ.AI?)
Related Papers

Recommenders and Search Tools

Link to Influence Flower
Influence Flower (What are Influence Flowers?)
Core recommender toggle
CORE Recommender (What is CORE?)
About arXivLabs

arXivLabs: experimental projects with community collaborators

arXivLabs is a framework that allows collaborators to develop and share new arXiv features directly on our website.

Both individuals and organizations that work with arXivLabs have embraced and accepted our values of openness, community, excellence, and user data privacy. arXiv is committed to these values and only works with partners that adhere to them.

Have an idea for a project that will add value for arXiv's community? Learn more about arXivLabs.