中文

AFiRe:面向放射学图像的解剖学驱动的自监督学习以实现细粒度表征

计算机视觉与模式识别 2025-04-23 v2

摘要

当前自监督方法(如对比学习)主要聚焦于全局判别,忽视了实现准确放射学分析所必需的关键细粒度解剖细节。为此,我们提出了面向放射学图像分析的解剖学驱动自监督框架,以增强细粒度表征(AFiRe)。AFiRe的核心思想是将解剖学一致性与Vision Transformer的独特token处理特征相结合。具体而言,AFiRe协同进行两种自监督方案:(i)基于解剖结构和范畴一致性对图像token进行对齐的token级解剖学引导对比学习,从而增强细粒度空间-解剖学判别;(ii)针对局部异常的像素级异常消除恢复,以细致的几何信息细化所学判别。此外,我们提出了合成病灶掩码,以增强解剖学多样性同时保持内部一致性,通常被传统数据增强(如裁切和仿射变换)破坏。实验结果表明,AFiRe:(i)提供了稳健的解剖学判别,与最先进的对比学习方法相比,实现了更一致的特征聚类;(ii)在多标签分类任务中表现出优异的泛化能力,超过7种放射学特定自监督方法;(iii)集成细粒度信息,仅使用图像级标注即可实现精确的异常检测。

关键词

引用

@article{arxiv.2504.10972,
  title  = {AFiRe: Anatomy-Driven Self-Supervised Learning for Fine-Grained Representation in Radiographic Images},
  author = {Yihang Liu and Lianghua He and Ying Wen and Longzhen Yang and Hongzhou Chen},
  journal= {arXiv preprint arXiv:2504.10972},
  year   = {2025}
}