Knowledge Injection Exists in MoE? Exploring Expert-Aware Contrast Decoding in MoE for Mitigating LLMs'Hallucinations

· Source: cs.CL updates on arXiv.org · Field: Technology & Digital — Artificial Intelligence & Machine Learning, Software Development & Engineering · Depth: Expert, extended

Summary

EAACD, an Expert-Aware Adaptive Contrast Decoding strategy, effectively mitigates hallucinations in Mixture-of-Experts (MoE) Large Language Models. Empirical studies revealed that "knowledge injection," a phenomenon used by traditional contrastive decoding, only occurs in MoE models without shared experts like LLaMA-MoE, but not in those with shared experts such as Qwen-MoE. However, the research identified that higher layers in all MoE architectures consistently show distinct expert activation patterns between factual and non-factual outputs. EAACD utilizes this by partitioning higher-layer experts into reliability groups, amplifying hallucinations in lower-reliability experts as negative references, and adaptively calibrating predictions. The method achieved state-of-the-art performance, outperforming all baselines on four datasets, including a nearly 13% improvement on HellaSwag for Qwen-MoE.

Key takeaway

For AI Scientists and Machine Learning Engineers developing or deploying MoE LLMs, this research offers a robust method to combat factual hallucinations. Your teams should consider integrating expert-aware adaptive contrast decoding (EAACD) to enhance model reliability, especially given its state-of-the-art performance without external resources. Implementing EAACD's hallucination amplification requires careful sandboxing to prevent risks, but it significantly improves factual accuracy in QA tasks.

Key insights

EAACD mitigates MoE LLM hallucinations by using expert activation differences and amplifying low-reliability expert errors for contrastive decoding.

Principles

Method

EAACD partitions higher-layer experts by confidence/consistency, amplifies lower-reliability expert hallucinations via attention/masking, then adaptively contrasts predictions to calibrate output.

In practice

Topics

Best for: AI Engineer, Research Scientist, AI Scientist, Machine Learning Engineer, NLP Engineer

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by cs.CL updates on arXiv.org.