arXiv:2412.04090cs.CV2024-12被引 1

用大模型当损失函数,让图像处理能按用户自定义目标优化。

LossAgent: Towards Any Optimization Objectives for Image Processing with LLM Agents

  • 用大模型作损失代理,动态组合多种损失函数。
  • 在超分辨率等任务中实现文本描述、主观反馈等复杂目标优化。
  • 适合需要灵活定制图像质量评估标准的开发者与研究者。

我们提出首个用于低层图像处理任务(如图像超分辨率与修复)的损失代理——LossAgent,旨在实现不同实际应用中任意自定义的优化目标。现有损失函数(如MSE)难以支持复杂手工设计的感知度量、文本描述及精细的人类反馈,导致端到端优化困难。为此,我们的LossAgent引入大语言模型(LLM)作为损失代理,利用其强大的文本理解能力,使损失代理可在优化过程中理解复杂的优化目标、轨迹与外部环境反馈。我们构建了包含多种支持端到端优化的损失函数的损失库,并设计面向优化的提示工程,使损失代理在每次优化交互中智能决定库中各损失的组合权重,从而实现任意定制化优化路径。在三个典型低层图像处理任务和多个优化目标上的大量实验表明,所提LossAgent具有有效性与普适性。

原文摘要 · Abstract (English)

We present the first loss agent, dubbed LossAgent, for low-level image processing tasks, e.g., image super-resolution and restoration, intending to achieve any customized optimization objectives of low-level image processing in different practical applications. Notably, not all optimization objectives, such as complex hand-crafted perceptual metrics, text description, and intricate human feedback, can be instantiated with existing low-level losses, e.g., MSE loss, which presents a crucial challenge in optimizing image processing networks in an end-to-end manner. To eliminate this, our LossAgent introduces the powerful large language model (LLM) as the loss agent, where the rich textual understanding of prior knowledge empowers the loss agent with the potential to understand complex optimization objectives, trajectory, and state feedback from external environments in the optimization process of the low-level image processing networks. In particular, we establish the loss repository by incorporating existing loss functions that support the end-to-end optimization for low-level image processing. Then, we design the optimization-oriented prompt engineering for the loss agent to actively and intelligently decide the compositional weights for each loss in the repository at each optimization interaction, thereby achieving the required optimization trajectory for any customized optimization objectives. Extensive experiments on three typical low-level image processing tasks and multiple optimization objectives have shown the effectiveness and applicability of our proposed LossAgent.

图像处理大模型损失函数自定义优化

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