论文检索
期刊
全部知识仓储预印本开放期刊机构
高级检索

基于动态阈值增强原型网络的联邦半监督学习模型OA北大核心CSTPCD

中文摘要

目前,联邦半监督学习面临着有效利用训练过程中大量无标签数据的挑战。尽管通过轻量级的原型网络实现客户端之间的知识共享可以缓解伪标签质量问题,但仍然有瓶颈。本文提出一种新的动态阈值增强下的原型网络联邦半监督学习算法。通过引入课程伪标签技术,其核心是对不同类别样本的学习状态动态调整阈值,使模型能够学习高质量的样本,显著提高模型的预测性能。实验结果表明,本算法在多个数据集上均取得优异的测试性能。在CIFAR-10数据集上,本算法相对于同类算法至少提高3%的测试精度。此外在SVHN和STL-10数据集上也有1%~7%的领先优势。值得注意的是,本算法在处理异质性和同质性数据时表现出色,且对于不同比例的有标签和无标签数据都具有良好的适应性。本算法不仅提高测试精度,而且未带来额外的通信开销和计算成本。这些结果表明本算法在联邦半监督学习领域具有巨大潜力,并为实际应用提供了一个性能卓越且高效的解决方案。

陈涛;谢在鹏;屈志昊;

河海大学计算机与信息学院,江苏南京211100

计算机与自动化

联邦学习;半监督学习;知识共享;原型网络;伪标签;动态阈值;无标签数据;数据异质性

《智能系统学报》 2024 (003)

P.534-545 / 12

水灾害防御全国重点实验室“一带一路”水资源与可持续发展科技基金项目(2021490811);国家自然科学基金青年项目(62102131);江苏省自然科学基金青年项目(BK20210361).

10.11992/tis.202311015

评论

下载量:0
点击量:0