基于知识缓存与个性化联邦学习的诊断预测装置及系统

本发明提供的基于知识缓存与个性化联邦学习的诊断预测装置及系统，涉及模型学习与医学影像诊断领域。本发明通过接收多个客户端上传的医学样本索引信息，构建动态知识缓存；根据医学样本索引信息创建标签到索引、索引到哈希值的索引关系映射表；初始化每个医学样本并建立每个医学样本与R个最相似样本的索引关系映射；接收各客户端上传的类别概率向量及其索引，更新知识缓存；然后更新索引关系映射表，并检索R个最相似样本的知识向量；采用加权平均方法生成综合知识表示；将综合知识表示分发至各客户端进行个性化联邦学习；训练好的各客户端模型用于医学诊断预测。本发明解决了通信成本高、模型结构兼容性差及公共数据集获取困难的问题。