###
计算机系统应用英文版:2024,33(12):177-184
本文二维码信息
码上扫一扫!
基于标签噪声对比学习的肺癌淋巴结转移鉴别
(兰州财经大学 信息工程与人工智能学院, 兰州 730020)
Identification of Lung Cancer Lymph Node Metastasis Based on Label Noise Contrastive Learning
(School of Information Engineering and Artificial Intelligence, Lanzhou University of Finance and Economics, Lanzhou 730020, China)
摘要
图/表
参考文献
相似文献
本文已被:浏览 17次   下载 243
Received:May 22, 2024    Revised:June 17, 2024
中文摘要: 基于深度学习的人工智能诊断模型严重依赖于高质量地详尽注释数据进行算法训练, 但受到标签噪声信息的影响. 为了增强模型的鲁棒性并防止有噪声的标签记忆, 提出了一种标签噪声样本选择 (noise label sample selection, NLSS)模型来充分挖掘噪声样本的隐藏信息, 减轻模型过拟合问题. 首先, 通过将混合增强图像作为输入, 提取图像分布式特征表示; 其次, 引入对比损失函数以及比较样本预测标签分布与其真实标签分布的相似性来评估样本, 进行样本选择; 最后, 通过标签重分配模块的伪标签提升策略在样本选择的基础上重新纠正噪声标签的监督信息. 以非小细胞肺癌 (non-small cell lung cancer, NSCLC)患者的 PET/CT 数据集为例进行实验, 结果表明提出的模型均比对比模型有一定的提升, 可降低淋巴结转移状态诊断中标签噪声的干扰.
Abstract:The AI diagnostic model based on deep learning relies heavily on high-quality detailed annotated data for algorithm training, but is affected by label noise information. To enhance the robustness of the model and prevent noisy label memory, a noise label sample selection (NLSS) model is proposed to fully mine the hidden information of noise samples and alleviate model overfitting. Firstly, distributed feature representations of the image are extracted by taking hybrid enhanced images as input. Secondly, the contrasive loss function is introduced to compare the similarity between the predicted label distribution of the sample and the real label distribution for sample evaluation and selection. Finally, based on sample selection, supervised information of the noisy label is re-corrected by the pseudo-label promotion strategy of the label redistribution module. Taking the PET/CT dataset of non-small cell lung cancer (NSCLC) patients as an example, results show that the proposed models outperform comparison models, reducing the interference of label noise in the diagnosis of lymph node metastasis.
文章编号:     中图分类号:    文献标志码:
基金项目:甘肃省高等学校青年博士基金 (2022QB-118)
引用文本:
祁婧,李子荣,刘秀婷,马露,陈俊豪.基于标签噪声对比学习的肺癌淋巴结转移鉴别.计算机系统应用,2024,33(12):177-184
QI Jing,LI Zi-Rong,LIU Xiu-Ting,MA Lu,CHEN Jun-Hao.Identification of Lung Cancer Lymph Node Metastasis Based on Label Noise Contrastive Learning.COMPUTER SYSTEMS APPLICATIONS,2024,33(12):177-184