基于贝叶斯和cGAN的甲状腺结节生成方法

作者:杨文开; 董云云; 赵涓涓*; 强彦; 刘江; Muhammad Bilal Zia
来源:计算机工程与设计, 2021, 42(04): 1036-1042.
DOI:10.16208/j.issn1000-7024.2021.04.020

摘要

为克服小样本医学图像数据集对训练强大的超声图像中甲状腺结节的自动分割或分类的深度学习模型的限制,提出一种基于贝叶斯神经网络和条件生成对抗网络(cGAN)的数据生成方法。构建cGAN并以修改的结节掩膜作为条件,通过在真实样本上训练生成器生成具有不同特征的结节。使用贝叶斯神经网络识别出最有价值的生成样本,并用其构建新的数据集训练模型进行结节的分割和分类。实验结果表明,提出方法仅使用约35%的真实样本即可实现令人满意的分割和分类性能。

全文