基于任务感知的三维医学图像重建预训练方法

本发明公开了一种基于任务感知的三维医学图像重建预训练方法，包括：对原始三维医学图像进行预处理，得到预处理后的图像；构建掩码重建主任务，在掩码重建主任务中建立主任务网络；主任务网络包括依次串联的掩码模块、重建主任务编码器和重建主任务解码器；其中，掩码模块用于对预处理后的图像进行基于高语义信息掩码策略的选择性掩码；构建分割辅助任务，在分割辅助任务中建立辅助任务网络，用以感知下游分割任务；辅助任务网络包括分割任务编码器和分割任务解码器；对掩码重建主任务和分割辅助任务进行联合学习，以同时优化训练主任务网络和辅助任务网络。该方法提升了预训练效率，且在不增加下游任务复杂度的情况下，提升了分割精度。