基于双阶段交叉注意力与条件高斯混合损失的MRI肝纤维化纵向预测

    Two-stage Cross Attention and Conditional Gaussian Mixture Loss for MRI-based Liver Fibrosis Longitudinal Prediction

    • 摘要: 肝纤维化纵向变化预测对指导治疗、改善预后具有重要临床意义。然而,肝纤维化病灶呈弥散性分布,且存在时空异质性,相邻肝纤维化分期的影像特征差异十分细微且缺乏一致性,借助无创检查对其纵向变化进行准确评估仍面临较大挑战。针对上述问题,本文提出基于孪生结构的条件高斯混合网络(Conditional Gaussian Mixture Network, CGMNet)。首先,本文设计双阶段交叉注意力模块,促进基线期与随访期特征的交互融合,增强对跨时相关键变化区域的关注能力;为识别相邻分期之间的细微差异、避免特征分布重叠,提出基于高斯混合分布的条件高斯混合损失。该损失以基线期图像的分期标签为条件,扩大类间特征的间距,并促使类内特征更紧凑。本文基于自建的肝纤维化患者配对磁共振成像数据集对所提方法进行验证。实验结果表明,CGMNet在多个评价指标上均优于现有主流基线模型;消融实验结果进一步验证了各模块对模型性能提升的贡献。

       

      Abstract: Predicting longitudinal changes in liver fibrosis is of great significance for guiding treatment decisions and improving patient prognosis. However, the diffuse distribution and spatiotemporal heterogeneity of fibrotic lesions, along with the subtle and inconsistent imaging differences between adjacent stages, pose significant challenges for accurate non-invasive assessment of disease progression. To address these issues, this paper proposes a Siamese-structured Conditional Gaussian Mixture Network (CGMNet) . First, a Two-Stage Cross Attention (TSCA) module is designed to promote feature interaction and fusion between baseline and follow-up images, thereby enhancing the model’s sensitivity to cross-temporal discriminative regions. Furthermore, to capture subtle inter-stage variations and alleviate feature distribution overlap, a Conditional Gaussian Mixture (CGM) Loss based on Gaussian mixture distribution is proposed. By using the staging labels of baseline images as conditions, this loss increases inter-class feature separation and improves intra-class feature compactness. The proposed method is evaluated on a self-constructed paired magnetic resonance imaging dataset of liver fibrosis patients. Experimental results show that CGMNet outperforms existing mainstream baseline models on multiple evaluation metrics, and ablation studies further confirm the individual contribution of each module to overall performance improvement.

       

    /

    返回文章
    返回