用量子电路增强U-Net,提升卫星图像野火分割精度
QFireNet: A Quantum-Enhanced U-Net for Wildfire Segmentation from Sentinel-2 Imagery

- 在U-Net瓶颈层引入量子变分电路(QuFeX与QB-Net)
- 量子模型在Sen2Fire数据集上F1达31.18,优于经典U-Net的28.71
- 数据混合显著缓解地理差异,使经典FPN F1升至39.76
从卫星影像中检测野火是语义分割难题,受类别不平衡、特征复杂及大气干扰影响。本文在基础U-Net模型上构建量子混合解决方案,针对Sen2Fire数据集的高维光谱特征空间进行建模。在U-Net瓶颈层注入变分量子电路,采用QuFeX与QB-Net量子线路。对比测试了经典特征金字塔网络(FPN),并改进经典U-Net的参数压缩、损失函数及输入数据均匀混合策略。主发现:在相同条件下,QB-Net(F1=31.18)与QuFeX(F1=30.79)均优于经典U-Net基线(F1=28.71);经典FPN取得31.13的可比结果。关键发现是数据混合有效消除训练与测试集间的地理域偏移,使经典FPN的F1提升至39.76。通过在加州烧毁区域(CaBuAr)数据集上的跨数据集迁移验证了架构的鲁棒性与泛化能力。总体表明,量子机器学习在野火图像分割中具有潜力,后续实验将进一步验证与拓展此发现。
原文摘要 · Abstract (English)
Wildfire detection from satellite imagery is a semantic image segmentation problem that has proven to be difficult due to challenges such as class imbalance, feature complexity, and atmospheric interference. In this paper, we build on the foundational U-Net image segmentation model to develop a quantum-hybrid solution in hopes of more effectively modeling the high-dimensional spectral feature space of the Sen2Fire dataset. We inject a variational quantum circuit in the bottleneck portion of U-Net, specifically the QuFeX and QB-Net ansatzes. We test a classical Feature Pyramid Network (FPN) for further comparative analysis of the model, and we also explore classical improvements to the U-Net model and its training process, including a compression of parameters, alternative loss functions, and uniform mixing of input data. Our primary finding is that under matched conditions, both QB-Net (with an $F_1$ score of 31.18) and QuFeX ($F_1 = 30.79$) outperformed the classical U-Net baseline results ($F_1 = 28.71$). Additionally, the classical FPN achieved a comparable score of 31.13. A crucial finding was that data mixing removed a significant domain shift between the geographically-separated train and test sets, which boosted the classical FPN $F_1$ score to 39.76. We validate the architecture's robustness and generalizability to the wildfire detection problem via cross-dataset transfer on the California Burned Areas (CaBuAr) dataset. Overall, we find that quantum machine learning has potential to provide an advantage in the problem of wildfire image segmentation, and further experiments will continue to validate and expand upon this finding.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。