Aug, 2023

CS-UNet模型的显微结构分割的迁移学习:一种基于Transformer和CNN编码器的混合算法

TL;DR转移学习通过使用在更大数据集上预先训练的参数来提高深度学习模型的性能。最近NASA的一项研究表明,使用在显微图像上预先训练的CNN编码器比使用在自然图像上预先训练的CNN编码器更有益于显微结构分割。然而,仅使用CNN模型仅能捕捉图像中的局部空间关系。本研究对比了在显微图像和自然图像上预先训练的Transformers和CNN模型的分割性能,结果部分确认了NASA研究的结论。我们还发现,在图像分割中,预先训练的Transformers和CNN编码器的组合一直优于仅预先训练的CNN编码器。这些结果表明,Transformers和CNN相互补充,并且当在显微图像上预先训练时,它们对下游任务更有益。