Pan-FM: A Pan-Organ Foundation Model with Saliency-Guided Masking for Missing Robustness

· Source: Takara TLDR - Daily AI Papers · Field: Science & Research — Artificial Intelligence & Machine Learning, Health & Medical Research, Life Sciences & Biology · Depth: Expert, quick

Summary

Pan-FM is a novel pan-organ foundation model designed for medical imaging, pre-trained on imaging data from seven organs: Brain, Heart, Adipose, Liver, Kidney, Spleen, and Pancreas. Unlike most existing foundation models that focus on unimodal data, Pan-FM addresses the challenge of realistic missing-organ scenarios in multimodal biomedical data. It employs a unified backbone to manage organ missingness during both training and inference, utilizing masking-based self-distillation. The model introduces Saliency-Guided Masking (SGM) to counteract "dominant-organ shortcut learning bias," where models over-rely on organs like adipose and heart. SGM adaptively masks dominant organs during pre-training based on model attention, fostering balanced cross-organ learning with negligible computational overhead. Evaluated on the UK Biobank, Pan-FM demonstrates superior prediction across 13 disease categories and 14 single disease entities, showing enhanced robustness in settings with missing organs.

Key takeaway

For AI Scientists developing multimodal medical imaging models, Pan-FM offers a robust approach to handling real-world missing data. Its Saliency-Guided Masking technique effectively mitigates dominant-organ bias, leading to more balanced and generalizable whole-body representations. You should consider incorporating similar adaptive masking strategies to improve model performance and robustness in complex, incomplete datasets.

Key insights

Pan-FM is a pan-organ foundation model robust to missing data via saliency-guided masking.

Principles

Method

Pan-FM uses a unified backbone for missing data, pre-trained with masking-based self-distillation. Saliency-Guided Masking (SGM) adaptively masks dominant organs using attention distribution to balance cross-organ learning.

In practice

Topics

Best for: AI Scientist, Research Scientist, Machine Learning Engineer

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by Takara TLDR - Daily AI Papers.