训练人工智能模型的运算装置与方法

本发明提供一种训练人工智能模型的运算装置与方法，该装置包含储存装置及处理器。该储存装置储存程序码，其中该程序码包含该人工智能模型，以及该人工智能模型包含物体检测器、多层感知器及交互行为解码器。该处理器用以载入并执行该程序码，其中该人工智能模型会执行以下训练操作：在第一阶段，先对该物体检测器进行训练；以及在第二阶段，利用该第一阶段所获得的参数来初始化该物体检测器的一部分，并对该人工智能模型进行训练，其中该人工智能模型的训练包含利用预训练模型进行知识蒸馏训练，以将该预训练模型的知识迁移至该多层感知器的输出。本发明能够提升人工智能模型的泛化性。