Towards Tracing Trustworthiness Dynamics: Revisiting Pre-training Period of Large Language Models

Chen Qian, Jie Zhang, Wei Yao, Dongrui Liu, Zhenfei Yin +3 more
2/29/2024
cs.CLcs.AI

Official code: ChnQ/TracingLLM (30 stars, Apache-2.0, last commit 2024-05-22).

Abstract

Ensuring the trustworthiness of large language models (LLMs) is crucial. Most studies concentrate on fully pre-trained LLMs to better understand and improve LLMs' trustworthiness. In this paper, to reveal the untapped potential of pre-training, we pioneer the exploration of LLMs' trustworthiness during this period, focusing on five key dimensions: reliability, privacy, toxicity, fairness, and robustness. To begin with, we apply linear probing to LLMs. The high probing accuracy suggests that \textit{LLMs in early pre-training can already distinguish concepts in each trustworthiness dimension}. Therefore, to further uncover the hidden possibilities of pre-training, we extract steering vectors from a LLM's pre-training checkpoints to enhance the LLM's trustworthiness. Finally, inspired by~\citet{choi2023understanding} that mutual information estimation is bounded by linear probing accuracy, we also probe LLMs with mutual information to investigate the dynamics of trustworthiness during pre-training. We are the first to observe a similar two-phase phenomenon: fitting and compression~\citep{shwartz2017opening}. This research provides an initial exploration of trustworthiness modeling during LLM pre-training, seeking to unveil new insights and spur further developments in the field. We will make our code publicly accessible at \url{https://github.com/ChnQ/TracingLLM}.

View on arXivView PDF

Code Implementations(1)

ChnQ/TracingLLMOfficial100%
Apache-2.0
303Shell, PythonFeb 29, 20242 years agoApache-2.0

Similar Papers

Cite this paper

@article{qian2024towards,
  title  = {Towards Tracing Trustworthiness Dynamics: Revisiting Pre-training Period of Large Language Models},
  author = {Chen Qian and Jie Zhang and Wei Yao and Dongrui Liu and Zhenfei Yin and Yu Qiao and Yong Liu and Jing Shao},
  year   = {2024},
  eprint = {2402.19465},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2402.19465v2}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2402.19465.svg)](https://codeofpaper.com/paper/2402.19465)

Discussion