跳转至

GIIM: Graph-based Learning of Inter- and Intra-view Dependencies for Multi-view Medical Image Diagnosis

会议: CVPR 2026
arXiv: 2603.09446
代码: 无
领域: 医学图像分析 / 图神经网络 / 计算机辅助诊断
关键词: 多异构图, 多视图诊断, 视图内/视图间依赖, 缺失视图处理, CADx

一句话总结

提出 GIIM 框架,基于多异构图(MHG)通过四类边关系同时建模同一病灶跨期相动态变化和不同病灶间空间关联,并设计四种缺失视图填充策略,在肝脏 CT、乳腺 X 光和乳腺 MRI 三种模态上均显著优于现有方法。

研究背景与动机

领域现状:临床诊断需要综合多个视图中异常之间的复杂依赖关系——同一病灶在多期增强 CT 中的动态变化、不同病灶间的空间共现关系等。CNN/Transformer/GNN 等方法已在单视图或简单多视图融合上取得进展。

现有痛点

  1. 现有 CADx 方法通常独立处理各视图或简单拼接特征,忽略了视图内(intra-view)多病灶间关系和视图间(inter-view)时序/空间动态
  2. 注意力方法需固定大小输入,无法灵活处理变数量病灶
  3. 临床中常因协议限制、技术故障或患者原因出现视图缺失,现有方法缺乏鲁棒应对

核心矛盾:需要同时建模四类依赖关系(同病灶跨视图、单视图不同病灶、多视图不同病灶、单-多视图聚合),且在视图缺失时保持鲁棒。

本文目标 将多视图医学诊断重构为关系建模问题,用异构图全面捕获四类依赖,同时处理缺失数据。

切入角度:GNN 天然适合变数量节点和异构关系建模,用不同类型节点和边编码不同层次的临床关系。

核心 idea:将每个患者的多病灶多视图数据构建为多异构图,用类型感知的消息传递同时推理四类依赖。

方法详解

整体框架

两阶段训练:(1) 对每个视图独立训练 ConvNeXt 特征提取器(利用 7×7 大核卷积和深度可分离卷积捕获形态和强度细节);(2) 冻结特征提取器,将多病灶多视图特征构建为多异构图(MHG),用异构消息传递 GNN 进行关系推理和分类。

关键设计

  1. 双类型节点表示

    • 单视图节点 \(N_{single}^v = f_v(l_v)\):某个视图下某病灶的特征
    • 多视图节点 \(M_{multi} = \|_{v=1}^V N_{single}^v\):所有视图特征拼接的聚合节点
    • 在乳腺 X 光场景中,因 CC/MLO 视图间病灶对应关系不确定,单视图节点改为该视图所有病灶特征的均值
  2. 四类边关系

    • \(E_{intra}\):同一病灶跨不同视图(如动脉期→静脉期→延迟期)→ 捕获时序增强变化
    • \(E_{s-m}\):单视图节点到其多视图聚合节点 → 整合各期信息
    • \(E_{inter-s}\):同一视图内不同病灶 → 建模空间共现(如 HCC 常与卫星灶共存)
    • \(E_{inter-m}\):不同病灶的聚合节点间 → 高层病灶上下文关系,小病灶可借助附近大病灶的上下文
  3. 异构消息传递

    • 对每个节点分别从 single-view 邻居和 multi-view 邻居聚合(各有独立权重矩阵 \(\mathbf{W}_{single}^k\)\(\mathbf{W}_{multi}^k\)),拼接原始状态后经非线性变换更新:\(h_n^k = \sigma(\mathbf{W}^k \cdot \text{CONCAT}(h_n^{k-1}, h_{N_{single}(n)}^k, h_{M_{multi}(n)}^k))\)
    • 5 层 SAGEConv(512→256→128→64→类数),最后一层直接输出分类概率
  4. 缺失视图四策略

    • Constant:零向量填充——简单但让模型学会忽略缺失节点
    • Learnable:可学习参数,通过 Frobenius 范数归一化
    • RAG-based:检索数据库中可用特征最相似的完整样本,借其缺失特征
    • Covariance-based:基于视图间特征差的协方差矩阵计算样本相似度,选最相似样本填充

损失函数 / 训练策略

  • 单视图阶段:标准分类交叉熵,ConvNeXt 独立训练后冻结
  • 图模型阶段:MHG 端到端训练,逐患者构图(每个 batch 一个患者图)

实验关键数据

主实验

数据集 方法 Accuracy (%) AUC (%)
Liver CT NN-based (多视图) 75.45 89.09
Attention-based 73.41 88.53
GIIM 78.20 91.05
VinDr-Mammo NN-based 67.48 82.21
Attention-based 68.09 81.00
GIIM 71.17 82.54
BreastDM (MRI) NN-based 80.85 87.35
Attention-based 85.10 76.37
GIIM 87.23 89.02

多视图 vs 单视图:Liver 上提升约 12% Acc,Mammo 上提升约 7.8% Acc。

消融实验

缺失视图策略对比(Liver,100% 缺失率测试)

策略 100% miss-view Full-view
NN-based 70.00 75.45
GIIM (Constant) 72.27 78.20
GIIM (Learnable) 72.05 77.05
GIIM (RAG) 71.59 78.41
GIIM (Covariance) 72.05 78.18

边类型消融:四类边缺任何一种都导致性能下降,\(E_{intra}\)(同病灶跨期)影响最大。

关键发现

  • 零向量在缺失测试下最稳定(作为唯一"缺失标记"让模型学会依赖其他视图),RAG/Covariance 在完整数据下更优
  • 多视图一致性提升最大的是 Liver(4 期 CT,同一病灶增强模式变化显著)
  • BI-RADS 分类中因 CC/MLO 视图间病灶对应不确定,采用均值聚合替代逐病灶建图

亮点与洞察

  • 四类边的设计完整覆盖了临床放射科医师的关系推理模式,比简单拆 attention 更有可解释性
  • 缺失视图策略的 trade-off 发现实用:生成式填充在完整数据下更好,零向量在缺失数据下更好
  • 异构消息传递的分类聚合思路(单视图 vs 多视图邻居分别聚合)避免了边类型信息丢失
  • GNN 的灵活性使得框架可处理任意数量的病灶和视图,优于需固定输入的 CNN/Transformer

局限与展望

  • 单视图特征提取器和图模型分阶段训练,端到端联合训练可能进一步提升
  • 图结构由数据硬编码(病灶数/视图数决定),未探索动态图构建或注意力加权边
  • ConvNeXt 作为 backbone 相对保守,ViT 或 SAM 等更强 backbone 可能进一步提升
  • 三个数据集规模相对有限(最大 920 例),大规模验证不足

相关工作与启发

  • vs Phase Attention (Wang et al. 2022):intra-phase + inter-phase attention,但处理固定大小输入且忽略病灶间关系;GIIM 用 GNN 灵活处理变数量病灶
  • vs SSL-MNGCN (Ibrahim et al. 2022):用 GCN 处理乳腺 X 光纹理/空间特征图,但未建模跨视图时序关系
  • vs mmFormer (Zhang et al. 2022):多模态 Transformer 处理不完整脑肿瘤分割,但针对体素级任务而非病灶级分类
  • 启发:异构图的关系建模范式可推广到其他需要多视图/多模态联合推理的场景

评分

  • 新颖性: ⭐⭐⭐⭐ 四类异构边 + 缺失视图策略组合设计完整
  • 实验充分度: ⭐⭐⭐⭐ 三种模态、缺失视图消融、四种填充策略对比
  • 写作质量: ⭐⭐⭐ 内容详实但结构略繁杂
  • 价值: ⭐⭐⭐ 医学多视图诊断的通用框架,但数据集规模限制了说服力