基于部分注释CT图像的自监督迁移学习肺结节分类

• 作者: 吴若愚 张久权 彭超 陶俊利 黄鸿
• 来源期刊: 光学学报
• 学科: 工学
• 关键词: 图像处理 肺结节分类 特征提取 自监督学习 部分注释 迁移学习
• 年，卷（期）: 2020,（18）
• 所属期刊栏目: 图像处理
• 页码范围: 93-100
• 页数: 8页
• 分类号: TP391
• 语种: 中文
• DOI: 10.3788/AOS202040.1810003

摘要

深度学习模型训练时需要大量的注释样本,但在医学领域注释数据难以获取.针对此问题,提出了一种结合部分注释数据的自监督学习算法,以提高3D肺结节的分类性能.在传统自监督训练的网络结构基础上,设计了一种多任务学习的网络结构,以同时利用医学图像处理任务中大量未注释数据和少量注释数据.通过先训练未注释数据然后加入注释数据继续训练的方式,实现了注释数据与未注释数据间部分网络结构和参数的共享.相较于传统自监督学习方法,所提算法在保证模型泛化能力的同时能够学习到更多与肺结节相关的鉴别特征,因此将模型迁移学习用于肺结节分类时也能表现出更佳的性能.所提算法在公开数据集LIDC-IDRI上的分类准确率达0.886,曲线下面积(AUC)值达0.929,实验结果表明,所提算法能够有效提升肺结节的分类性能.

作者信息

序号	姓名	单位	发文数	被引次数	H指数	G指数
1	黄鸿		73	422	11.0	15.0
2	彭超		5	2	1.0	1.0
3	吴若愚		1	0	0.0	0.0
4	陶俊利		1	0	0.0	0.0
5	张久权		1	0	0.0	0.0