Cluster-norm for Unsupervised Probing of Knowledge

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Laurito, Walter, Maiya, Sharan, Dhimoïla, Grégoire, Owen, Yeung, Hänni, Kaarel
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866913529995460608
author Laurito, Walter
Maiya, Sharan
Dhimoïla, Grégoire
Owen
Yeung
Hänni, Kaarel
author_facet Laurito, Walter
Maiya, Sharan
Dhimoïla, Grégoire
Owen
Yeung
Hänni, Kaarel
contents The deployment of language models brings challenges in generating reliable information, especially when these models are fine-tuned using human preferences. To extract encoded knowledge without (potentially) biased human labels, unsupervised probing techniques like Contrast-Consistent Search (CCS) have been developed (Burns et al., 2022). However, salient but unrelated features in a given dataset can mislead these probes (Farquhar et al., 2023). Addressing this, we propose a cluster normalization method to minimize the impact of such features by clustering and normalizing activations of contrast pairs before applying unsupervised probing techniques. While this approach does not address the issue of differentiating between knowledge in general and simulated knowledge - a major issue in the literature of latent knowledge elicitation (Christiano et al., 2021) - it significantly improves the ability of unsupervised probes to identify the intended knowledge amidst distractions.
format Preprint
id arxiv_https___arxiv_org_abs_2407_18712
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Cluster-norm for Unsupervised Probing of Knowledge
Laurito, Walter
Maiya, Sharan
Dhimoïla, Grégoire
Owen
Yeung
Hänni, Kaarel
Artificial Intelligence
Computation and Language
Machine Learning
The deployment of language models brings challenges in generating reliable information, especially when these models are fine-tuned using human preferences. To extract encoded knowledge without (potentially) biased human labels, unsupervised probing techniques like Contrast-Consistent Search (CCS) have been developed (Burns et al., 2022). However, salient but unrelated features in a given dataset can mislead these probes (Farquhar et al., 2023). Addressing this, we propose a cluster normalization method to minimize the impact of such features by clustering and normalizing activations of contrast pairs before applying unsupervised probing techniques. While this approach does not address the issue of differentiating between knowledge in general and simulated knowledge - a major issue in the literature of latent knowledge elicitation (Christiano et al., 2021) - it significantly improves the ability of unsupervised probes to identify the intended knowledge amidst distractions.
title Cluster-norm for Unsupervised Probing of Knowledge
topic Artificial Intelligence
Computation and Language
Machine Learning
url https://arxiv.org/abs/2407.18712