-
公开(公告)号:CN119478549A
公开(公告)日:2025-02-18
申请号:CN202411822578.5
申请日:2024-12-11
Applicant: 支付宝(杭州)信息技术有限公司
IPC: G06V10/764 , G06N20/00
Abstract: 本说明书公开了一种图像分类模型的训练方法、装置、介质及电子设备,在各边缘节点训练适用于自身的图像分类模型时,中心节点可采用知识蒸馏技术,根据获取到的各边缘节点的图像分类模型的局部编码端的参数,调整中心节点中的全局编码端的参数,得到调整后的全局编码端的参数。进而中心节点针对每个边缘节点,根据调整后的全局编码端以及该边缘节点的局部编码端,调整该边缘节点的局部编码端的参数,并将调整后的局部编码端的参数发送至该边缘节点。在各边缘节点可在数据共享的情况下,即便每个边缘节点的图像分类模型的架构不同,每个边缘节点仍旧能够得到适应于自身模型架构的图像分类模型,提高了训练出的图像分类模型的性能。