(一)文章信息 |
标题:MCMCNet: A Semi-Supervised Road Extraction Network for High-Resolution Remote Sensing Images via Multiple Consistency and Multitask Constraints
期刊:《IEEE Transactions on Geoscience and Remote Sensing》(中科院1区,IF=7.5)
作者:Lipeng Gao, et al.
doi:10.1109/TGRS.2024.3426561
(二)研究背景 |
(三)研究数据与方法 |
伪标签约束:通过引导对比学习模块(GCLM)增强类间可区分性,促进更精确的伪标签生成。
道路骨架预测:在原有的道路分割预测头的基础上,添加了道路骨架预测头(RSPH)来确保道路提取的连续性和主干道的完整性。
适应性道路增强模块:该模块通过结合局部和全局信息,增强道路的线性特征,避免模型学习冗余信息。
(四)研究结果 |
1. 性能评估与对比
通过对比实验,MCMCNet相较于其他经典的半监督道路提取模型,在这两个指标上分别提高了 3%到5%。这一提升反映了MCMCNet在准确性和完整性方面的优势。F1分数和IoU的提升不仅表现在较为简单的场景下,也在复杂场景中有所体现,特别是道路遮挡和背景干扰较为严重的区域。相比于完全监督的模型,MCMCNet能够在标注数据非常有限的情况下实现优秀的性能,验证了半监督学习策略在遥感影像道路提取任务中的有效性。
2. 伪标签与引导对比学习
伪标签生成是本研究的一个关键创新。通过使用引导对比学习模块(GCLM),MCMCNet能够提升伪标签的质量,并提高网络对未标注数据的学习能力。实验结果表明,这种策略有效提高了伪标签的准确性,减少了因标注数据稀缺而导致的误差。图像分割任务中,通过使用GCLM,MCMCNet增强不同类别之间的可区分性,进一步改善道路提取的精度。通过对比学习模块,模型能够学习到更多关于道路结构的信息,从而提高提取效果。
3. 道路骨架预测(RSPH)
为了保证道路提取的连续性和完整性,MCMCNet引入了道路骨架预测头(RSPH)。这一模块的引入显著提升了模型在长条形道路的提取效果,尤其是在道路被遮挡或背景较为复杂的情况下。通过骨架预测,模型能够保持道路的连贯性,避免出现断裂或遗漏现象。在实验中,RSPH模块帮助模型准确地提取了主干道,并在一些测试场景中处理了复杂的道路形态(如交叉口、曲折道路等)。
4. 适应性道路增强模块(ARAM)
MCMCNet中还加入了适应性道路增强模块(ARAM),该模块通过结合局部和全局信息来增强道路的线性特征,并避免冗余信息的学习。通过该模块,MCMCNet能够更好地处理复杂场景中的道路特征,尤其是在多变的地理环境中表现优异。实验表明,ARAM模块在提高道路提取精度的同时,避免了对无关背景信息的过度学习,提升模型的整体鲁棒性。
5. 复杂场景下的表现
复杂场景(如道路遮挡、背景杂乱、不同道路宽度等)往往是道路提取任务中的一大挑战。MCMCNet通过其独特的模块设计,在这些复杂场景中仍表现出较高的精度。实验显示,尽管遥感影像中存在一些道路被遮挡或具有不规则形状的情况,MCMCNet依然能够成功提取出道路信息,且不会丢失关键细节。对比实验中的可视化结果也验证了MCMCNet在不同场景下的道路提取效果,尤其在解决道路细节的提取和背景噪声的分离上,MCMCNet表现出了较好的能力。
6. 多任务学习的优势
通过引入多任务学习框架,MCMCNet不仅能够进行道路分割,还能够同时进行骨架预测等多项任务。这种多任务约束进一步优化了道路提取的效果,使得MCMCNet能够在处理复杂的道路场景时保持较高的准确率和连续性。实验表明,多任务学习策略有效提高了模型的学习效率和道路提取的精度,尤其是在需要提取细节部分的任务中,表现尤为突出。
(五)研究结论 |
本研究提出的MCMCNet在半监督学习框架下,通过引入多重一致性和多任务约束,有效解决了高分辨率遥感影像中道路提取的挑战。通过引入伪标签约束、道路骨架预测和适应性增强模块,MCMCNet不仅提高了道路提取的精度,还增强了模型的鲁棒性。实验结果表明,MCMCNet在三个不同的数据集上都实现了优于现有方法的表现,尤其是在处理复杂道路场景时,具有更强的适应性和可扩展性。因此,MCMCNet为未来的道路提取研究提供了一个有效的参考框架。
文章来源 :Gao, L., Zhou, Y., Tian, J., Cai, W., & Lv, Z. (2024). MCMCNet: A Semi-supervised Road Extraction Network for High-resolution Remote Sensing Images via Multiple Consistency and Multi-task Constraints. IEEE Transactions on Geoscience and Remote Sensing.
免责声明 :本公众号发布的内容仅供学习交流使用,内容版权归原作者所有。如有侵犯您的权益,请及时与我们联系。
| 回复1096获取全文!
2024-12-3
2024-12-3
2024-10-23