Correcting Large Language Model Behavior via Influence Function

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Zhang, Han, Zhang, Zhuo, Zhang, Yi, Zhai, Yuanzhao, Peng, Hanyang, Lei, Yu, Yu, Yue, Wang, Hui, Liang, Bin, Gui, Lin, Xu, Ruifeng
Format: Preprint
Veröffentlicht: 2024
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866916537770704896
author Zhang, Han
Zhang, Zhuo
Zhang, Yi
Zhai, Yuanzhao
Peng, Hanyang
Lei, Yu
Yu, Yue
Wang, Hui
Liang, Bin
Gui, Lin
Xu, Ruifeng
author_facet Zhang, Han
Zhang, Zhuo
Zhang, Yi
Zhai, Yuanzhao
Peng, Hanyang
Lei, Yu
Yu, Yue
Wang, Hui
Liang, Bin
Gui, Lin
Xu, Ruifeng
contents Recent advancements in AI alignment techniques have significantly improved the alignment of large language models (LLMs) with static human preferences. However, the dynamic nature of human preferences can render some prior training data outdated or even erroneous, ultimately causing LLMs to deviate from contemporary human preferences and societal norms. Existing methodologies, whether they involve the curation of new data for continual alignment or the manual correction of outdated data for re-alignment, demand costly human resources. To address this challenge, we propose a novel approach, Large Language Model Behavior Correction with Influence Function Recall and Post-Training (LANCET), which requires no human involvement. LANCET consists of two phases: (1) using influence functions to identify the training data that significantly impact undesirable model outputs, and (2) applying an Influence function-driven Bregman Optimization (IBO) technique to adjust the model's behavior based on these influence distributions. Our experiments demonstrate that LANCET effectively and efficiently correct inappropriate behaviors of LLMs. Furthermore, LANCET can outperform methods that rely on collecting human preferences, and it enhances the interpretability of learning human preferences within LLMs.
format Preprint
id arxiv_https___arxiv_org_abs_2412_16451
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Correcting Large Language Model Behavior via Influence Function
Zhang, Han
Zhang, Zhuo
Zhang, Yi
Zhai, Yuanzhao
Peng, Hanyang
Lei, Yu
Yu, Yue
Wang, Hui
Liang, Bin
Gui, Lin
Xu, Ruifeng
Machine Learning
Artificial Intelligence
Computation and Language
Recent advancements in AI alignment techniques have significantly improved the alignment of large language models (LLMs) with static human preferences. However, the dynamic nature of human preferences can render some prior training data outdated or even erroneous, ultimately causing LLMs to deviate from contemporary human preferences and societal norms. Existing methodologies, whether they involve the curation of new data for continual alignment or the manual correction of outdated data for re-alignment, demand costly human resources. To address this challenge, we propose a novel approach, Large Language Model Behavior Correction with Influence Function Recall and Post-Training (LANCET), which requires no human involvement. LANCET consists of two phases: (1) using influence functions to identify the training data that significantly impact undesirable model outputs, and (2) applying an Influence function-driven Bregman Optimization (IBO) technique to adjust the model's behavior based on these influence distributions. Our experiments demonstrate that LANCET effectively and efficiently correct inappropriate behaviors of LLMs. Furthermore, LANCET can outperform methods that rely on collecting human preferences, and it enhances the interpretability of learning human preferences within LLMs.
title Correcting Large Language Model Behavior via Influence Function
topic Machine Learning
Artificial Intelligence
Computation and Language
url https://arxiv.org/abs/2412.16451