基于层一致性平均教师模型的半监督岩石薄片图像分类 |
| |
引用本文: | 严子杰,王杨,陈雁,张翀.基于层一致性平均教师模型的半监督岩石薄片图像分类[J].应用科学学报,2024(1):27-38. |
| |
作者姓名: | 严子杰 王杨 陈雁 张翀 |
| |
作者单位: | 西南石油大学计算机科学学院 |
| |
基金项目: | 四川省科技厅科技计划项目(No.20ZDYF1215)资助; |
| |
摘 要: | 传统的岩石薄片图像分类依赖于大量人工标记的图像样本,这种方式受制于标记人员的经验和能力,且无法通过不断增加的未标记岩石薄片图像样本实现分类能力的可扩展式增强。该文提出的在平均教师(mean teacher, MT)模型的基础上,通过在无监督损失中添加层一致性正则化项的方式约束师生模型的层次结构,实现对未标记数据信息的有效利用。消融实验和层一致性平均教师(hierarchy consistency mean teacher, HCMT)模型对比实验结果表明,层一致性正则化方法利用了未标记数据的有效信息,提升了MT模型的分类效果,使得HCMT模型可以在半标记数据集中获得如全标记数据集相似的分类能力。该实验表明,半监督学习模型利用大量未标记岩石薄片图像数据可以提升模型分类的能力。
|
关 键 词: | 半监督学习 平均教师模型 岩石薄片图像分类 层一致性方法 |
|
|