arXiv:2509.01898cs.CV2025-09被引 8

针对无人机热成像超分辨率中的少样本过拟合问题,提出新表征学习方法提升模型泛化能力。

DroneSR: Rethinking Few-shot Thermal Image Super-Resolution from Drone-based Perspective

论文配图:DroneSR: Rethinking Few-shot Thermal Image Super-Resolution from Drone-based Perspective
图 1 · 摘自论文原文
  • 采用高斯量化表征学习缓解扩散模型在小样本下的过拟合
  • 在自建无人机红外数据集上验证,显著降低大模型过拟合现象
  • 适合研究少样本图像重建与无人机视觉的学者参考

尽管大规模模型在性能上取得显著进步,但过拟合问题仍严重削弱其泛化能力。在图像超分辨率任务中,扩散模型作为生成模型的代表通常采用大规模架构,然而无人机捕获的少量红外训练数据常导致大规模架构严重过拟合。为此,本文提出一种面向扩散模型的新型高斯量化表示学习方法,有效缓解过拟合并增强鲁棒性。同时设计有效的监控机制,在训练过程中检测大规模架构的过拟合迹象。通过引入高斯量化表示学习,在保持模型复杂度的同时显著减少过拟合。在此基础上,构建了一个多源无人机红外图像基准数据集用于检测,并强调了在少样本、多样化的无人机图像重建场景中大规模架构的过拟合问题。为验证方法缓解过拟合的有效性,实验在所构建的基准数据集上进行。结果表明,该方法优于现有超分辨率方法,并在复杂条件下显著减轻大规模架构的过拟合。代码与DroneSR数据集将公开于:https://github.com/wengzp1/GARLSR。

原文摘要 · Abstract (English)

Although large scale models achieve significant improvements in performance, the overfitting challenge still frequently undermines their generalization ability. In super resolution tasks on images, diffusion models as representatives of generative models typically adopt large scale architectures. However, few-shot drone-captured infrared training data frequently induces severe overfitting in large-scale architectures. To address this key challenge, our method proposes a new Gaussian quantization representation learning method oriented to diffusion models that alleviates overfitting and enhances robustness. At the same time, an effective monitoring mechanism tracks large scale architectures during training to detect signs of overfitting. By introducing Gaussian quantization representation learning, our method effectively reduces overfitting while maintaining architecture complexity. On this basis, we construct a multi source drone-based infrared image benchmark dataset for detection and use it to emphasize overfitting issues of large scale architectures in few sample, drone-based diverse drone-based image reconstruction scenarios. To verify the efficacy of the method in mitigating overfitting, experiments are conducted on the constructed benchmark. Experimental results demonstrate that our method outperforms existing super resolution approaches and significantly mitigates overfitting of large scale architectures under complex conditions. The code and DroneSR dataset will be available at: https://github.com/wengzp1/GARLSR.

超分辨率无人机扩散模型少样本

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。