一种面向遥感语义分割任务的跨模型知识蒸馏方法

AITNT
正文
推荐专利
一种面向遥感语义分割任务的跨模型知识蒸馏方法
申请号:CN202411444821
申请日期:2024-10-16
公开号:CN119514595A
公开日期:2025-02-25
类型:发明专利
摘要
一种面向遥感语义分割任务的跨模型知识蒸馏方法,它涉及一种跨模型知识蒸馏方法。本发明为了解决现有遥感图像语义分割模型参数量大、消耗资源大的问题。本发明的步骤包括:选择能力互补的学生模型和教师模型,选取数据集训练教师模型,使用训练好的教师模型输出的特征图作为知识输入学生模型;将输入到学生模型的特征图按照通道进行注意力加权,并计算损失;将由教师模型传递的知识解耦成为目标区域与非目标区域,将这两个独立的目标区域进行预测与知识蒸馏,计算损失;将学生模型语义分割的损失与上述步骤的损失进行加权求和,计算出总损失,通过总损失更新学生模型的参数,最终将学生模型训练完毕。本发明属于遥感图像处理技术领域。
技术关键词
知识蒸馏方法 教师 学生 注意力 遥感图像语义分割 遥感图像处理技术 超参数 正则化策略 标签 像素 通道 数学 数据 编码器 表达式 纹理特征 高层次
添加客服微信openai178,进AITNT官方交流群
驱动智慧未来:提供一站式AI转型解决方案
沪ICP备2023015588号