面向数据稀缺脑电 FLAIR MRI 的 SLIM-Diff:共享潜在图像-掩码扩散模型
计算机视觉与模式识别
2026-02-04 v1 人工智能
摘要
脑电 FLAIR MRI 中的焦点皮层畸形(FCD)病灶细微且稀少,导致联合图像-掩码生成模型易产生不稳定和记忆现象。我们提出 SLIM-Diff,一个紧凑的联合扩散模型,其主要贡献在于:(i)单一共享瓶颈 U-Net 网络,从 2 通道图像+掩码表示中强制解剖学与病灶几何之间的紧密耦合;(ii)通过可调 目标进行损失-几何调优。作为内部基线,我们包括标准 DDPM 风格目标(-预测,使用 损失),并在匹配设置下隔离预测参数化和 几何的影响。实验表明,-预测在联合合成中始终是最佳选择, 的分数次二次惩罚在提高图像保真度方面表现更佳,而 更能保留病灶掩码的形态学特征。我们的代码和模型权重已公开于 https://github.com/MarioPasc/slim-diff。
引用
@article{arxiv.2602.03372,
title = {SLIM-Diff: Shared Latent Image-Mask Diffusion with Lp loss for Data-Scarce Epilepsy FLAIR MRI},
author = {Mario Pascual-González and Ariadna Jiménez-Partinen and R. M. Luque-Baena and Fátima Nagib-Raya and Ezequiel López-Rubio},
journal= {arXiv preprint arXiv:2602.03372},
year = {2026}
}
备注
6 pages, 2 figures, 1 table, conference paper