12月2日,人工智能领域顶级会议The Thirty-Fifth Association for the Advancement of Artificial Intelligence-AAAI2021 (CCF A) 接收了本实验室博士生罗祥德同学在半监督医学图像分割方面的研究工作“Semi-supervised Medical Image Segmentation through Dual-task Consistency”。该工作针对基于半监督的医学图像算法,提出了一种利用双任务一致性约束的新方法,将同一个分割问题表示成两个不同的任务,并鼓励两个任务在预定义的表示空间内保持一致,进而充分利用未标注的数据提升深度神经网络的性能,同时大大降低训练网络模型所需要的标注成本。
基于深度学习的方法因为在图像处理上优越表现而受到广泛的关注,近年来在图像识别、人工智能领域不断取得了性能突破。但是由于深度神经网络需要依赖大量良好标注的数据,在小数据上很难达到比较好的效果。在医学图像领域,数据标注通常需要大量的时间,也需要医学领域的专业知识,但医生通常很忙,没有充足的时间来标注大量的数据,因此从少量标注数据和大量未标注数据来学习以获得高性能模型变得尤为重要。
基于这样的问题,本文提出了一种基于双任务一致性的半监督学习算法,在现有全监督医学图像分割算法基础上,该算法可以充分利用没有标注的数据进行进一步学习,进而大大提高未标注数据的利用率和促进网络分割性能。实验表明,通过引入双任务一致性,网络能更稳定的从少量标注数据和大量未标注数据中学习,并显著提高分割结果。同时与最新的半监督分割算法相比,此方法需要的训练成本更低,所取得的效果也更好,从而降低了深度神经网络对标注数据的依赖。
本研究成果由电子科技大学与同济大学、商汤科技等单位合作完成。第一作者罗祥德同学是本实验室一年级博士生,本论文也是他取得的首项研究成果。AAAI 2021共有7911篇有效审稿论文,最终接收了1692篇,接受率仅为21%。
预印版论文链接:https://arxiv.org/pdf/2009.04448.pdf
项目代码:https://github.com/HiLab-git/DTC
手机浏览
公众号
访问量: 开通时间:-- 最后更新时间:--
清水河校区:成都市高新区(西区)西源大道2006号 邮编: 611731
沙河校区:成都市建设北路二段四号 邮编:610054
蜀ICP备 05006379 号 I 川公网安备 51019002000280 号