Habaek通过数据扩增与结构优化,实现高精度水体分割
Habaek: High-performance water segmentation through dataset expansion and inductive bias optimization
- 用ADE20K和RIWA数据集扩增训练数据,提升模型泛化能力
- 在遥感影像上达到0.91986~0.94397的IoU,超越现有模型
- 结合LoRA降低计算量,适合实时水体监测应用
水体分割对灾害响应和水资源管理至关重要。利用高分辨率航拍图像监控河流、湖泊和水库,有助于农业、工业及生态保护的主动管理。深度学习提升了洪水监测能力,使CNN、U-Net和Transformer等模型可处理大量卫星与航空影像数据。然而这些模型通常计算开销大,难以用于实时场景。本研究通过引入ADE20K和RIWA数据集进行数据增强,改进SegFormer模型以提升其泛化性能,并探究归纳偏置对注意力模型的影响,发现模型在更大数据集上表现更优。为进一步降低复杂度,采用低秩适应(LoRA)技术,在保持精度的同时减少计算需求。实验表明,所提出的Habaek模型在分割任务中取得0.91986至0.94397的交并比(IoU),F1-score、召回率、准确率和精确率均优于对比模型,展现出实际应用潜力。研究强调了模型结构优化与数据集扩展在高效水体分割中的重要性。
原文摘要 · Abstract (English)
Water segmentation is critical to disaster response and water resource management. Authorities may employ high-resolution photography to monitor rivers, lakes, and reservoirs, allowing for more proactive management in agriculture, industry, and conservation. Deep learning has improved flood monitoring by allowing models like CNNs, U-Nets, and transformers to handle large volumes of satellite and aerial data. However, these models usually have significant processing requirements, limiting their usage in real-time applications. This research proposes upgrading the SegFormer model for water segmentation by data augmentation with datasets such as ADE20K and RIWA to boost generalization. We examine how inductive bias affects attention-based models and discover that SegFormer performs better on bigger datasets. To further demonstrate the function of data augmentation, Low-Rank Adaptation (LoRA) is used to lower processing complexity while preserving accuracy. We show that the suggested Habaek model outperforms current models in segmentation, with an Intersection over Union (IoU) ranging from 0.91986 to 0.94397. In terms of F1-score, recall, accuracy, and precision, Habaek performs better than rival models, indicating its potential for real-world applications. This study highlights the need to enhance structures and include datasets for effective water segmentation.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。