一种基于集成知识蒸馏的食物识别方法及系统

    公开(公告)号:CN118351530A

    公开(公告)日:2024-07-16

    申请号:CN202410446098.7

    申请日:2024-04-15

    Applicant: 江南大学

    Abstract: 本发明公开了一种基于集成知识蒸馏的食物识别方法及系统,属于食物图像识别技术领域。所述方法包括:分别训练CNN和视觉Transformer,用于提取食物图像的特征;集成CNN和视觉Transformer,求解最优权重组合;将基础模型的集成作为教师模型对CNN跟视觉Transformer进行知识蒸馏;再次集成经过知识蒸馏后的CNN跟视觉Transformer。本发明利用集成学习和知识蒸馏的技术,集成学习有效地结合了CNN的局部感知能力和视觉Transformers的全局建模能力,这种协同集成增强了模型在类别内识别微小差异和捕获各类别间各种模式的能力,知识蒸馏进一步提升集成模型的同时实现集成模型的轻量化。

Patent Agency Ranking