摘要
本发明涉及一种世界模型构建方法,特别涉及一种端到端自动驾驶测试的高保真轻量化世界模型构建方法,构建高保真世界模型,针对世界模型参数庞大、推理效率低下等问题,对世界模型进行知识蒸馏,在保留世界模型生成能力的基础上降低模型参数,提升推理效率;针对世界模型计算瓶颈部分自定义开发CUDA算子、优化显存分配,并基于单设备多线程调度及多设备协同计算方法,提升高保真世界模型的推理效率。本发明可构建一种面向端到端自动驾驶测试的高保真轻量化世界模型,有效解决现有世界模型多模态信息对齐精度不高、跨视图及跨帧一致性差、推理效率低等问题,提升端到端自动驾驶系统测试进程的置信度,大幅加快端到端自动驾驶系统的测试效率。