arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2406.09397cs.CVcs.AI

在检索中使视觉模型与人类审美对齐:基准与算法

Aligning Vision Models with Human Aesthetics in Retrieval: Benchmarks and Algorithms

Southeast University(东南大学);Tsinghua University(清华大学);Fudan University(复旦大学);Microsoft Research Asia(微软亚洲研究院);Microsoft(微软公司)

Miaosen Zhang, Yixuan Wei, Zhen Xing, Yifei Ma, Zuxuan Wu, Ji Li, Zheng Zhang, Qi Dai, Chong Luo, Xin Geng, Baining Guo

AI总结:

针对检索中视觉模型难以符合人类审美的问题,提出利用 LLM 推理扩展查询,并通过基于偏好的强化学习蒸馏 LLM 与审美模型知识,同时构建 HPIR 基准验证效果。

摘要:

现代视觉模型是在非常庞大且含噪的数据集上训练的。尽管这些模型获得了强大能力,但在某些方面,它们可能不会遵循用户意图来输出期望结果,例如视觉审美、偏好风格和责任。本文聚焦视觉审美领域,旨在使检索系统中的视觉模型与人类审美标准对齐。先进检索系统通常采用级联的审美模型作为重排序器或过滤器,但这些模型局限于饱和度等低层特征,并且在涉及风格、文化或知识语境时表现不佳。我们发现,利用大语言模型(LLM)的推理能力来改写搜索查询并扩展审美预期,可以弥补这一缺陷。基于上述发现,我们提出一种基于偏好的强化学习方法,对视觉模型进行微调,以从 LLM 推理和审美模型中蒸馏知识,从而使视觉模型更好地与人类审美对齐。同时,由于专门用于评估检索系统的基准较少,我们利用大型多模态模型(LMM)的强大能力来评估审美性能。鉴于审美评估是最具主观性的任务之一,为验证 LMM 的鲁棒性,我们进一步提出了一个名为 HPIR 的新型数据集,用于对人类审美对齐进行基准测试。实验表明,在多项指标下,我们的方法显著增强了视觉模型的审美行为。我们相信,所提出的算法可以成为使视觉模型与人类价值观对齐的一种通用实践。

英文摘要:

Modern vision models are trained on very large noisy datasets. While these models acquire strong capabilities, they may not follow the user's intent to output the desired results in certain aspects, e.g., visual aesthetic, preferred style, and responsibility. In this paper, we target the realm of visual aesthetics and aim to align vision models with human aesthetic standards in a retrieval system. Advanced retrieval systems usually adopt a cascade of aesthetic models as re-rankers or filters, which are limited to low-level features like saturation and perform poorly when stylistic, cultural or knowledge contexts are involved. We find that utilizing the reasoning ability of large language models (LLMs) to rephrase the search query and extend the aesthetic expectations can make up for this shortcoming. Based on the above findings, we propose a preference-based reinforcement learning method that fine-tunes the vision models to distill the knowledge from both LLMs reasoning and the aesthetic models to better align the vision models with human aesthetics. Meanwhile, with rare benchmarks designed for evaluating retrieval systems, we leverage large multi-modality model (LMM) to evaluate the aesthetic performance with their strong abilities. As aesthetic assessment is one of the most subjective tasks, to validate the robustness of LMM, we further propose a novel dataset named HPIR to benchmark the alignment with human aesthetics. Experiments demonstrate that our method significantly enhances the aesthetic behaviors of the vision models, under several metrics. We believe the proposed algorithm can be a general practice for aligning vision models with human values.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑