DPO Kernels: A Semantically-Aware, Kernel-Enhanced, and Divergence-Rich Paradigm for Direct Preference Optimization

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Das, Amitava, Trivedy, Suranjana, Khanna, Danush, Roy, Rajarshi, Singh, Gurpreet, Ghosh, Basab, Narsupalli, Yaswanth, Jain, Vinija, Sharma, Vasu, Reganti, Aishwarya Naresh, Chadha, Aman
Formato: Preprint
Publicado: 2025
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866916571220279296
author Das, Amitava
Trivedy, Suranjana
Khanna, Danush
Roy, Rajarshi
Singh, Gurpreet
Ghosh, Basab
Narsupalli, Yaswanth
Jain, Vinija
Sharma, Vasu
Reganti, Aishwarya Naresh
Chadha, Aman
author_facet Das, Amitava
Trivedy, Suranjana
Khanna, Danush
Roy, Rajarshi
Singh, Gurpreet
Ghosh, Basab
Narsupalli, Yaswanth
Jain, Vinija
Sharma, Vasu
Reganti, Aishwarya Naresh
Chadha, Aman
contents The rapid rise of large language models (LLMs) has unlocked many applications but also underscores the challenge of aligning them with diverse values and preferences. Direct Preference Optimization (DPO) is central to alignment but constrained by fixed divergences and limited feature transformations. We propose DPO-Kernels, which integrates kernel methods to address these issues through four key contributions: (i) Kernelized Representations with polynomial, RBF, Mahalanobis, and spectral kernels for richer transformations, plus a hybrid loss combining embedding-based and probability-based objectives; (ii) Divergence Alternatives (Jensen-Shannon, Hellinger, Renyi, Bhattacharyya, Wasserstein, and f-divergences) for greater stability; (iii) Data-Driven Selection metrics that automatically choose the best kernel-divergence pair; and (iv) a Hierarchical Mixture of Kernels for both local precision and global modeling. Evaluations on 12 datasets demonstrate state-of-the-art performance in factuality, safety, reasoning, and instruction following. Grounded in Heavy-Tailed Self-Regularization, DPO-Kernels maintains robust generalization for LLMs, offering a comprehensive resource for further alignment research.
format Preprint
id arxiv_https___arxiv_org_abs_2501_03271
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle DPO Kernels: A Semantically-Aware, Kernel-Enhanced, and Divergence-Rich Paradigm for Direct Preference Optimization
Das, Amitava
Trivedy, Suranjana
Khanna, Danush
Roy, Rajarshi
Singh, Gurpreet
Ghosh, Basab
Narsupalli, Yaswanth
Jain, Vinija
Sharma, Vasu
Reganti, Aishwarya Naresh
Chadha, Aman
Machine Learning
Artificial Intelligence
Computation and Language
68T45
The rapid rise of large language models (LLMs) has unlocked many applications but also underscores the challenge of aligning them with diverse values and preferences. Direct Preference Optimization (DPO) is central to alignment but constrained by fixed divergences and limited feature transformations. We propose DPO-Kernels, which integrates kernel methods to address these issues through four key contributions: (i) Kernelized Representations with polynomial, RBF, Mahalanobis, and spectral kernels for richer transformations, plus a hybrid loss combining embedding-based and probability-based objectives; (ii) Divergence Alternatives (Jensen-Shannon, Hellinger, Renyi, Bhattacharyya, Wasserstein, and f-divergences) for greater stability; (iii) Data-Driven Selection metrics that automatically choose the best kernel-divergence pair; and (iv) a Hierarchical Mixture of Kernels for both local precision and global modeling. Evaluations on 12 datasets demonstrate state-of-the-art performance in factuality, safety, reasoning, and instruction following. Grounded in Heavy-Tailed Self-Regularization, DPO-Kernels maintains robust generalization for LLMs, offering a comprehensive resource for further alignment research.
title DPO Kernels: A Semantically-Aware, Kernel-Enhanced, and Divergence-Rich Paradigm for Direct Preference Optimization
topic Machine Learning
Artificial Intelligence
Computation and Language
68T45
url https://arxiv.org/abs/2501.03271