跳到正文
DRIVE: Distributional and Retrieval-Augmented Bidding with Value Evaluation
cs.LG · · 收录于 arXiv 日报 20260616

DRIVE: Distributional and Retrieval-Augmented Bidding with Value Evaluation

作者:Miduo Cui, Haochen Wang, Shangqin Mao, Xun Yang, Qianlong Xie

作者提出 DRIVE 框架,通过分布动作建模、检索增强候选生成和基于价值的评价,解耦动作生成与决策,以优化离线自动出价策略。

论文信息摘要

论文要解决什么

解决现有 Transformer 离线出价方法因单模态和纯参数化公式导致的“平均动作”陷阱,以及在稀疏或长尾流量下动作不可靠的问题。

论文方法概览

  • Backbone: Transformer-based sequence modeling (Decision Transformer style)
  • 关键模块: 分布动作建模 (GMM), 检索增强候选生成,价值评价
  • 损失函数: 离线强化学习损失 (如 CQL/DT 变体)

读后要点

个人理解,欢迎指正

实验表明 DRIVE 在 AuctionNet 和多个离线 RL 基准上持续改进性能,且能泛化到多种 Transformer 方法;解耦设计有效缓解了多模态坍缩。

局限与疑问

未明确提及具体局限,但离线强化学习通常受限于日志数据分布覆盖范围,检索模块在超大规模数据下的效率待验证。

阅读笔记

以下为个人阅读笔记,原文见 arXiv:2606.14192v1。

arXiv 阅读笔记:DRIVE - 分布与检索增强的自动出价价值评估框架

1. 研究背景与动机

在线广告已成为数字流量变现的主要渠道,实时竞价(RTB)机制要求广告商在预算和成本约束下优化长期表现。由于在线探索(Online Exploration)在真实广告拍卖中风险过高且成本昂贵,离线强化学习(Offline RL)成为学习出价策略的必要范式。近年来,基于 Transformer 的序列建模方法(如 Decision Transformer, DT)通过利用长程时间依赖关系,在从日志数据中学习策略方面展现出潜力。

然而,论文指出直接将此类 DT 风格架构应用于现实出价场景仍面临显著挑战。作者认为,现有的方法通常采用单模态(unimodal)和纯参数化(purely parametric)的公式,这导致了两个核心问题:

  1. “平均动作”陷阱(Average Action Trap):相似的市场状态往往对应多种截然不同但有效的出价策略(例如激进的高价或保守的低价)。单模态建模倾向于将这些多样化的行为坍缩为次优的平均动作,导致出价既不够激进以赢得拍卖,也不够保守以控制成本。
  2. 稀疏数据与长尾流量:纯参数化模型缺乏保留高质量历史决策的显式机制,在低密度区域(稀疏数据或长尾流量)容易产生不可靠的动作,尽管数据集中可能存在高质量决策。

论文通过图 1 直观地展示了 DT 风格方法在现实出价中面临的这两个挑战:

2. DRIVE 框架核心设计

为了解决上述局限性,作者提出了 DRIVE(Distributional and Retrieval-Augmented Bidding with Value Evaluation)框架。这是一个统一的基于 Transformer 的离线自动出价框架,其核心思想是将候选动作生成与决策制定解耦。

DRIVE 框架主要包含三个关键组件:

  1. 分布动作建模(Distributional Action Modeling): 作者提出使用高斯混合模型(GMM)对动作空间进行建模。这使得策略能够捕捉多样且有效的出价模式,避免了单模态回归导致的动作坍缩问题,从而保留多模态的出价策略。

  2. 检索增强候选生成(Retrieval-Augmented Candidate Generation): 借鉴自然语言处理中的检索增强生成(RAG)思想,DRIVE 引入检索机制。该机制编码当前状态,并从类似状态中检索高质量的历史动作作为补充候选。这提供了显式的非参数化支持,缓解了稀疏数据 regime 下动作不可靠的问题。

  3. 基于价值的评价(Value-based Evaluation): 在推理阶段,框架引入一个价值批评家(Value Critic)来评估生成的和检索到的候选动作,并选择最有希望的出价。这种基于价值的评价机制确保了最终决策的鲁棒性。

论文通过图 3 展示了 DRIVE 的整体架构,该框架建立在基于 Transformer 的离线 RL 范式之上,整合了多模态动作建模、检索增强和价值评估模块:

3. 实验验证与结果

为了验证 DRIVE 的有效性,作者在 AuctionNet(一个代表性的离线出价基准)以及额外的离线强化学习基准上进行了广泛实验。

实验结果表明,DRIVE 能够 consistently 提高出价性能,并且在多种基于 Transformer 的方法上表现出良好的泛化能力。作者进一步证明,DRIVE 可以无缝集成到多种 DT 风格的方法中,从而在不同离线 RL 基准上持续改进性能。

论文通过图 5 展示了 AuctionNet 和 D4RL 基准的可视化结果,直观反映了模型在不同环境下的表现分布:

4. 总结

DRIVE 论文针对离线自动出价中的多模态策略坍缩和稀疏数据不可靠问题,提出了一种结合分布建模、检索增强和价值评估的统一框架。通过解耦动作生成与决策,DRIVE 不仅保留了多样化的有效出价模式,还利用历史高质量决策增强了模型的鲁棒性。实验结果证实了该方法在 AuctionNet 及多个离线 RL 基准上的优越性,为实时广告系统中的自动出价提供了新的解决方案。

本篇笔记收录于 arXiv 日报 20260616。更多综述见 博客归档 。