AI News HubLIVE
站内改写1 分钟阅读

QFireNet:一种用于 Sentinel-2 影像野火分割的量子增强 U-Net

该论文提出 QFireNet,一种结合量子电路和 U-Net 的混合模型,用于从卫星图像中分割野火。在 Sen2Fire 数据集上,量子增强模型(QB-Net 和 QuFeX)在 F1 分数上优于经典 U-Net 基线,且数据混合技术显著提升了性能。

来源arXiv Machine Learning作者: Jaiman Munshi (IonQ Team, App Dev Club, University of Maryland, College Park), Tanvi Tewary (IonQ Team, App Dev Club, University of Maryland, College Park), Sawyer Bloom (IonQ Team, App Dev Club, University of Maryland, College Park), Aidan Chu (IonQ Team, App Dev Club, University of Maryland, College Park), Chetan Maviti (IonQ Team, App Dev Club, University of Maryland, College Park), Kyon Winston-Bey (IonQ Team, App Dev Club, University of Maryland, College Park), Harshit Badjatia (IonQ Team, App Dev Club, University of Maryland, College Park), Farhan Kittur (IonQ Team, App Dev Club, University of Maryland, College Park), Vardhan Madhavarapu (IonQ Team, App Dev Club, University of Maryland, College Park), Varun Kota (IonQ Team, App Dev Club, University of Maryland, College Park), Joshua Kwon (IonQ Team, App Dev Club, University of Maryland, College Park), Nazia Rangwala-Vohra (IonQ Team, App Dev Club, University of Maryland, College Park), Franz Klein (IonQ Team, App Dev Club, University of Maryland, College Park)

来自 arXiv 的一项新研究提出了一种量子增强的深度学习模型,用于从 Sentinel-2 卫星影像中分割野火区域。该模型名为 QFireNet,基于经典的 U-Net 图像分割架构,并在其瓶颈部分引入了变分量子电路,旨在更好地建模高维光谱特征空间。野火检测是一个具有挑战性的语义分割问题,因为存在类别不平衡、特征复杂和大气干扰等问题。研究者希望借助量子计算的表达能力来提升分割性能。

在实验中,团队测试了两种量子电路设计:QuFeX 和 QB-Net 变分量子电路。这些电路被注入到 U-Net 的瓶颈层,形成量子混合模型。使用 Sen2Fire 数据集进行对比评估,该数据集包含从 Sentinel-2 卫星获取的多光谱图像。结果表明,两种量子增强模型均优于经典的 U-Net 基线:QB-Net 达到了 31.18 的 F1 分数,QuFeX 达到了 30.79,而经典 U-Net 仅为 28.71。此外,他们还测试了经典的特征金字塔网络(FPN),其 F1 分数为 31.13,与量子模型相当。

一个关键的发现是数据混合技术的重要性。由于训练集和测试集来自不同地理区域,存在显著的域偏移。通过均匀混合输入数据,模型能够消除这种偏移,将经典 FPN 的 F1 分数大幅提升至 39.76,提高了约 8.6 个百分点。这一结果强调了数据预处理在泛化中的关键作用。研究还探索了其他经典改进,包括参数压缩和替代损失函数,但数据混合带来的提升最大。

为了验证模型的鲁棒性,团队在加利福尼亚燃烧区域(CaBuAr)数据集上进行了跨数据集迁移测试。结果表明,QFireNet 能够较好地泛化到新的地理区域,进一步证实了量子混合架构的潜力。总体而言,该研究为量子机器学习在遥感图像分割领域的应用提供了有力证据,未来工作将探索更复杂的量子电路和更大规模的数据集。