摘要
arXiv:2412.20651v2 宣传类型: 替换-交叉
摘要:通过在大规模数据集上进行训练,已被证明能够增强生成模型在扩散模型中生成和处理图像的质量和真实性;然而,由于成本和隐私问题,医疗成像领域中并非总是能够获取到这样的大规模数据集,这与这种模型的主要应用之一相矛盾,即在真实数据稀缺的情况下生成合成样本。同时,预先训练的一般模型的微调也因医疗领域和预先训练模型之间的分布转移而成为一个挑战。在这里,我们提出了一种名为Latent Drift(LD)的方法,可以使扩散模型适用于任何微调方法以缓解由分布转移带来的问题,或者在推理时间作为条件使用。Latent Drifting使扩散模型能够条件化于适合复杂任务的医疗图像,包括反事实图像生成,这对于研究性别、年龄以及在患者中添加或移除疾病如何改变医疗图像非常重要。我们使用三种公开的纵向基准数据集(包含脑MRI和胸片)来评估我们的方法在反事实图像生成中的表现。我们的结果证明,在与不同微调方案结合使用时,可以获得显著的性能提升。