Knowledge Injection Exists in MoE? Exploring Expert-Aware Contrast Decoding in MoE for Mitigating LLMs'Hallucinations
Abstract
Existing LLM hallucination mitigation methods, including prompt engineering and model optimization, either hardly alter models'internal knowledge or have poor cross-domain generalization.
Contrastive decoding mitigates hallucinations by using layer-wise differences in LLMs.
However, prior studies only explore transformer-based models (e.g., GPT), ignoring other effective frameworks like mixture-of-experts (MoE) models.
Since MoE alters the traditional transformer architecture, we conduct empirical studies to investigate whether similar layer-wise differences exist in MoEs.
Our results show that they do not exist in MoE with shared experts; nevertheless, across different MoEs, higher layers exhibit distinct expert activation patterns between factual and non-factual outputs.
Building on these, we propose EAACD, an expert-aware adaptive contrast decoding that uses expert differences in MoE's higher layers to mitigate hallucinations on QA tasks.
EAACD splits high-layer experts into a higher-reliability group and several lower-reliability groups based on their confidence and consistency.
It contrasts the higher-reliability group's prediction with each lower-reliability group's prediction to calibrate the model's original predictions.
To strengthen this contrast, EAACD amplifies hallucinations from lower-reliability experts via attention and masking to provide stronger negative references.
EAACD outperforms all baselines on four datasets.
이 뉴스, 어떠셨어요?
탭 한 번으로 반응 · 로그인 불필요