Semantic-guided Diverse Decoding for Large Language Model

Sirui Han (The Hong Kong University of Science and Technology) · Weijie Shi (Hong Kong University of Science and Technology) · Yue Cui (HKUST) · Yaguang Wu (Alibaba Group) · Jingzhi Fang (Hong Kong University of Science and Technology) · Shibo Zhang (MetaX Integrated Circuit (Shanghai) Co., LTD) · Mengze Li (The Hong Kong University of Science and Technology) · Jia Zhu (Zhejiang Normal University) · Jiajie Xu (Xidian University) · Xiaofang Zhou (The Hong Kong University of Science and Technology)
adaptive gain functionsbest-of-n strategiesconstraint optimizationdata synthesisdiverse beam searchdiverse decodingdynamic inter-group repulsionembedding spacegroup-based reinforcement learningorthogonal directional guidanceposition-debiased probability assessmentrlhf training convergencesemantic diversitytemperature sampling

Diverse decoding of large language models is crucial for applications requiring multiple semantically distinct responses, yet existing methods primarily achieve lexical rather than semantic diversity. This limitation significantly constrains Best-of-N strategies, group-based reinforcement learning, and data synthesis. While temperature sampling and diverse beam search modify token distributions or apply n-gram penalties, they fail to ensure meaningful semantic differentiation. We introduce Semantic-guided Diverse Decoding (SemDiD), operating directly in embedding space that balances quality with diversity through three complementary mechanisms: orthogonal directional guidance, dynamic inter-group repulsion, and position-debiased probability assessment. SemDiD harmonizes these competing objectives using adaptive gain functions and constraint optimization, ensuring both quality thresholds and maximal semantic differentiation. Experiments show SemDiD consistently outperforms existing methods, improving Best-of-N coverage by 1.4-5.2% across diverse tasks and accelerating RLHF training convergence by 15% while increasing accuracy by up to 2.1%.