Token-Importance Guided Direct Preference Optimization

TL;DR

This paper introduces TI-DPO, a novel framework that improves alignment of large language models with human preferences by using a hybrid token importance weighting and triplet loss for more accurate, robust, and diverse responses.

Contribution

The paper presents TI-DPO, combining gradient attribution with a Gaussian prior and triplet loss to enhance token importance estimation and response alignment in LLMs.

Findings

TI-DPO outperforms DPO and RLHF in accuracy and diversity.

It provides more stable and computationally efficient alignment.

The method improves fine-grained semantic control in LLMs.

Abstract

Aligning Large Language Models (LLMs) with human preferences is crucial for safe and effective AI interactions. While popular methods like Direct Preference Optimization (DPO) have simplified alignment, they remain sensitive to data noise and overlook the differential importance of individual tokens. Existing token-level approaches often rely on probability prediction or simplistic weighting schemes to obtain token importance, which still cannot fully address these issues. To solve this problem, we propose the Token-Importance Guided Direct Preference Optimization (TI-DPO), a framework that achieves fine-grained semantic control through two synergistic innovations. First, we propose a novel hybrid weighting mechanism that combines gradient attribution with a Gaussian prior, ensuring both the accuracy and robustness of token importance scores. Second, we employ a triplet loss to provide structured guidance for the optimization, explicitly guiding model outputs to approach preferred responses and diverge from non-preferred ones. Experimental results show that TI-DPO achieves higher accuracy and stronger generative diversity, providing more stable and computationally efficient solutions compared with DPO and other RLHF methods.