一种图像分类模型的训练方法、装置、介质及电子设备

AITNT
正文
推荐专利
一种图像分类模型的训练方法、装置、介质及电子设备
申请号:CN202411822578
申请日期:2024-12-11
公开号:CN119478549A
公开日期:2025-02-18
类型:发明专利
摘要
本说明书公开了一种图像分类模型的训练方法、装置、介质及电子设备,在各边缘节点训练适用于自身的图像分类模型时,中心节点可采用知识蒸馏技术,根据获取到的各边缘节点的图像分类模型的局部编码端的参数,调整中心节点中的全局编码端的参数,得到调整后的全局编码端的参数。进而中心节点针对每个边缘节点,根据调整后的全局编码端以及该边缘节点的局部编码端,调整该边缘节点的局部编码端的参数,并将调整后的局部编码端的参数发送至该边缘节点。在各边缘节点可在数据共享的情况下,即便每个边缘节点的图像分类模型的架构不同,每个边缘节点仍旧能够得到适应于自身模型架构的图像分类模型,提高了训练出的图像分类模型的性能。
技术关键词
编码 节点 参数 训练图像分类模型 在线 样本 学生 移动平均算法 知识蒸馏技术 教师 电子设备 处理器 注意力机制 训练装置 可读存储介质 模块 存储器
添加客服微信openai178,进AITNT官方交流群
驱动智慧未来:提供一站式AI转型解决方案
沪ICP备2023015588号