DualIFM:面向视网膜眼底图像的可解释基础模型

HuggingFace Daily Papers(社区热门论文)·2026-08-04 08:00·20天前
AI 导读

DualIFM 通过 BagNet 骨干网络实现设计上的可解释性,其小感受野生成的类别证据图忠实反映模型决策过程,并在预训练中加入 2D 投影层以直接可视化表征空间。该模型在超过 80 万张彩色眼底照片上训练,性能与参数多 16 倍的 RETFound 相当,且在分布外数据上提供可解释预测。代码与预训练模型已开源。

HuggingFace Daily Papers(社区热门论文)
54AI 编辑部评分,满分 100

DualIFM:面向视网膜眼底图像的可解释基础模型

2026-08-04 08:00· 20天前
AI 导读

DualIFM 通过 BagNet 骨干网络实现设计上的可解释性,其小感受野生成的类别证据图忠实反映模型决策过程,并在预训练中加入 2D 投影层以直接可视化表征空间。该模型在超过 80 万张彩色眼底照片上训练,性能与参数多 16 倍的 RETFound 相当,且在分布外数据上提供可解释预测。代码与预训练模型已开源。

Foundation models are used to extract transferable representations from large amounts of unlabeled data, typically via self-supervised learning (SSL). However, many of these models rely on architectures that offer limited interpretability, a critical issue in high-stakes domains such as medical imaging. We propose DualIFM, a foundation model that is interpretable-by-design via a BagNet backbone whose small receptive fields generate class evidence maps that are faithful to the model's decision-making process. Additionally, DualIFM incorporates a 2D projection layer during pretraining that enables direct visualization of the representation space, providing a dataset-level view of the learned structure including meaningful clinical clusters as well as potential spurious correlations. We trained DualIFM on over 800,000 color fundus photographs from various sources to learn generalizable representations for different downstream tasks. Our model achieves performance comparable to RETFound, which has 16times more parameters, while providing interpretable predictions on out-of-distribution data. These results suggest that large-scale SSL pretraining paired with inherent interpretability can lead to robust representations for retinal imaging. Code and pretrained models are available at github.com/berenslab/interpretable_FM.

来源:HuggingFace Daily Papers(社区热门论文)· arxiv.org

arXivGitHub论文/研究