Aug, 2023

DermoSegDiff: 一种边界感知的皮肤病变分割扩散模型

TL;DR皮肤病变分割在皮肤病的早期检测和准确诊断中起着至关重要的作用。本论文提出了一种新的框架DermoSegDiff,它在学习过程中融入边界信息,并引入了一种新的损失函数来优先考虑边界,并逐渐降低其他区域的重要性。我们还引入了一种基于U-Net的去噪网络,在网络内部高效地集成了噪声和语义信息。在多个皮肤分割数据集上的实验结果表明,DermoSegDiff相比现有的CNN、Transformer和扩散方法在各种情况下均具有优越性,展示了其效果和泛化能力。该实现可在GitHub上公开访问。