UniGD Unifies Generative and Discriminative Retrieval for Ads.

Shujie Ji, Yawei Kong, Yilin Zhao, Li Wang, Xialong Liu, Peng Jiang· August 5, 2026 View original

Key takeaways

  • UniGD unifies generative retrieval and relevance scoring in one model.
  • It improves ad revenue by 5.78% and reduces inference latency by 33%.
  • Key components include Conflict-Aware Gradient Enhancement and Codebook-Anchored Representations.
  • The framework supports heterogeneous ad materials like video, product, and live-stream ads.

Who benefits

AdTechE-commerceSearch EnginesSocial MediaRetail

Summary

UniGD is a new framework that integrates generative retrieval and relevance scoring into a single model for industrial search advertising, improving effectiveness and reducing serving costs. It uses conflict-aware gradient enhancement, codebook-anchored representations, and heterogeneous ad-material modeling.

This paper introduces UniGD, a Unified Generative-Discriminative framework designed to revolutionize industrial search advertising. Current generative retrieval (GR) systems often separate the generative likelihood objective from query-ad relevance discrimination, leading to suboptimal effectiveness and higher serving costs. UniGD addresses this by integrating both retrieval and relevance scoring within a single, cohesive model. To manage the potential gradient interference during joint optimization, UniGD incorporates Conflict-Aware Gradient Enhancement (CAGE), which adaptively coordinates the two objectives. Furthermore, it features a Codebook-Anchored Representation Module (CAM) that grounds item representations in frozen hierarchical codebooks derived from multimodal pretrained models, enriching them with robust semantic priors. For diverse ad types like short-video, product, and live-stream ads, UniGD employs Heterogeneous Ad-material Modeling (HAM). This component captures cross-type semantic commonality through a shared backbone while preserving type-specific modeling capabilities. Online A/B tests on a major search advertising platform demonstrated significant improvements, including a 5.78% increase in ad revenue, a 33% reduction in inference latency, and enhanced discriminative relevance estimation.

Why it matters

For professionals in ad tech, e-commerce, or search platforms, UniGD offers a significant leap in efficiency and effectiveness for industrial retrieval systems, directly impacting revenue and operational costs.

How to implement this in your domain

  1. 1Evaluate your current search advertising or recommendation system for opportunities to unify generative and discriminative models.
  2. 2Investigate the UniGD framework's components (CAGE, CAM, HAM) for potential integration into your existing AI infrastructure.
  3. 3Conduct A/B tests on a subset of your advertising traffic to validate the revenue and latency benefits.
  4. 4Train internal teams on advanced retrieval techniques that combine generative and discriminative approaches.

Original post by Shujie Ji, Yawei Kong, Yilin Zhao, Li Wang, Xialong Liu, Peng Jiang

"arXiv:2608.03150v1 Announce Type: new Abstract: Generative retrieval (GR) is a promising paradigm for industrial search advertising, yet its deployment is constrained by strict relevance and latency requirements. Existing systems cascade GR with an independent relevance model, de…"

View on X

Originally posted by Shujie Ji, Yawei Kong, Yilin Zhao, Li Wang, Xialong Liu, Peng Jiang on X · view source

Want to go deeper?

Turn these trends into skills with Learnijoy's hands-on AI & tech courses.

Explore courses