vix.ing · top · new · best · stats · spec

Habaek: High-performance water segmentation through dataset expansion and inductive bias optimization

2024/10/21 by Hanseon Joo, Eunji Lee, Joo, Hanseon +3
Computer Science · Earth and Planetary Sciences · Environmental Science · #Artificial Intelligence (cs.AI) #Computer Vision and Pattern Recognition (cs.CV) #FOS: Computer and information sciences #Image Enhancement Techniques #Underwater Acoustics Research #Water Quality Monitoring Technologies

paper · pdf · doi:10.48550/arxiv.2410.15794

openalex publication_date 2024/10/21 · openalex created_date 2024/11/06 · openalex updated_date 2026/07/28

Abstract

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.

Related