首页|期刊导航|东华大学学报(英文版)|RoCoNet:一种用于宫颈细胞图像半监督目标检测的旋转对比网络

RoCoNet:一种用于宫颈细胞图像半监督目标检测的旋转对比网络OA

RoCoNet:Rotational Contrastive Network for Semi-Supervised Cervical Cell Image Object Detection

中文摘要英文摘要

本研究提出了一种基于旋转不变性、对比学习和自适应混合阈值的半监督学习框架——RoCoNet,旨在提升半监督学习在医学细胞数据集上的适用性.细胞数据集独特的采样方式使得输入图片的旋转角度不定,现有的半监督检测器所使用的传统卷积核难以有效工作.为解决该问题,本研究提出了一种旋转注意力卷积机制,以增强模型对旋转变换的稳健性.同时,通过对有监督学习中的对比损失进行改进,提出了跨特征对比损失,并将其运用在半监督学习中,解决了细胞重叠和聚集导致模型分类性能较差的问题.此外,对于训练早期伪标签数量不稳定的情况,我们提出了自适应混合阈值,在局部阈值的基础上使用高斯混合模型(Gaussian mixture model,GMM)计算全局阈值对其进行修正,使伪标签的生成兼顾了数量和质量.在宫颈细胞学液基薄层细胞检测(ThinPrep cytology test,TCT)数据集上的实验显示,仅使用 10%的标签数据,RoCoNet 的平均精度(mean average precision,mAP)就能达到 31.6%,较基线方法的 mAP高出8.4 个百分点.

A semi-supervised learning framework integrating rotational invariance,contrastive learning,and adaptive hybrid thresholds,named rotational contrastive network(RoCoNet),is proposed to enhance the applicability of semi-supervised learning for medical cell datasets.Due to the unique sampling approach of cell datasets,input images often contain uncertain rotation angles,which render traditional convolution kernels ineffective in existing semi-supervised detectors.To address this challenge,rotational attention convolution is introduced,offering robustness to rotational transformations.Additionally,cross-feature contrastive loss is proposed to improve upon the contrastive loss used in supervised learning,tackling issues of poor classification performance caused by cell overlap and clustering.An adaptive hybrid threshold is also introduced to stabilize pseudo-label generation during early training.A global threshold,computed by using Gaussian mixture models(GMMs),is applied to refine the local threshold,which helps balance the quantity and quality of pseudo-labels.Experiments on the ThinPrep cytology test(TCT)dataset for cervical cytopathology show that RoCoNet achieves a mean average precision(mAP)of 31.6%with only 10%labeled data,outperforming the baseline method by 8.4%in mAP.

黄秋波;龚润泽;陈德华

东华大学 计算机科学与技术学院,上海 201620东华大学 计算机科学与技术学院,上海 201620东华大学 计算机科学与技术学院,上海 201620

信息技术与安全科学

半监督学习旋转不变性对比学习目标检测宫颈细胞

semi-supervised learningrotational invariancecontrastive learningobject detectioncervical cell

《东华大学学报(英文版)》 2026 (2)

82-93,12

10.19884/j.1672-5220.202502001

评论