Enhancing Building Information Extraction from Remote Sensing Images through Reinforcement Learning from AI Feedback
Deep learning is a standard approach in remote sensing for building information extraction, but it faces challenges like sparse data, large obstacles, and complex light-shadow interactions. To address these, we propose a novel reinforcement learning approach, Reinforcement Learning from AI Feedback (RLAIF), which enhances extraction by using a customized reward network with a deep multi-scale attention mechanism. We also improve the deep deterministic policy gradient algorithm with AI feedback, reducing iterations and boosting robustness. Experimental results show our approach reduces cross-entropy loss per pixel by 3.52% compared to supervised and 7.43% compared to unsupervised methods, while improving recall and accuracy, solidifying its competitive advantage.
Authors
- Lipeng Gao (ORCID: https://orcid.org/0000-0003-0026-3719)
- JingTao Yao (ORCID: https://orcid.org/0000-0002-7823-4136)
- Wenjing Cai (ORCID: https://orcid.org/0000-0001-9503-8519)
- Liang He (ORCID: https://orcid.org/0000-0001-8032-8767)
- Huiqun He
Institutions
- Northwestern Polytechnical University (CN)
- Shanghai Academy of Spaceflight Technology (CN)
Publication Details
- Published
- 2024-12-20
- DOI
- https://doi.org/10.1109/acai63924.2024.10899536
- Primary Topic
- Remote Sensing and Land Use
- Type
- article
- Field-Weighted Citation Impact
- 0.00
Funders
- Research Promotion Foundation
- Aeronautical Science Foundation of China