paper-with-me

홈 › Papers

AtP*: An efficient and scalable method for localizing LLM behaviour to components

2024-03-01 · János Kramár, Tom Lieberum, Rohin Shah, Neel Nanda

Activation Patching is a method of directly computing causal attributions of behavior to model components. However, applying it exhaustively requires a sweep with cost scaling linearly in the number of model components, which can be prohibitively expensive for SoTA Large Language Models (LLMs). We investigate Attribution Patching (AtP), a fast gradient-based approximation to Activation Patching and find two classes of failure modes of AtP which lead to significant false negatives. We propose a variant of AtP called AtP*, with two changes to address these failure modes while retaining scalability. We present the first systematic study of AtP and alternative methods for faster activation patching and show that AtP significantly outperforms all other investigated methods, with AtP* providing further significant improvement. Finally, we provide a method to bound the probability of remaining false negatives of AtP* estimates.

📄 PDF Abstract BibTeX arXiv:2403.00745

Code (0)

등록된 구현이 없습니다.

Methods 이 논문이 사용한 방법론

Patching Activation patching studies the model's computation by altering its latent representations, the token embeddings in transformer-based language models, during the inference process

Similar Papers 제목 키워드 기반

Head and Tail Localization of C. elegans

2020-01-12 · Mansi Ranjit Mane, Aniket Anand Deshmukh, Adam J. Iliff

C. elegans is commonly used in neuroscience for behaviour analysis because of it's compact nervous system with well-described connectivity. Localizing the animal and distinguishing between its head and tail are important…

BIG-bench Machine Learning

Decomposing and Editing Predictions by Modeling Model Computation

2024-04-17 · Harshay Shah, Andrew Ilyas, Aleksander Madry

How does the internal computation of a machine learning model transform inputs into predictions? In this paper, we introduce a task called component modeling that aims to address this question. The goal of component mode…

counterfactualmodelModel Editing

Localizing Model Behavior with Path Patching

2023-04-12 · Nicholas Goldowsky-Dill, Chris MacLeod, Lucas Sato, Aryaman Arora

Localizing behaviors of neural networks to a subset of the network's components or a subset of interactions between components is a natural first step towards analyzing network mechanisms and possible failure modes. Exis…

model

Anomaly Detection using Autoencoders in High Performance Computing Systems

2018-11-13 · Andrea Borghesi, Andrea Bartolini, Michele Lombardi, Michela Milano 외

Anomaly detection in supercomputers is a very difficult problem due to the big scale of the systems and the high number of components. The current state of the art for automated anomaly detection employs Machine Learning…

Anomaly DetectionVocal Bursts Intensity Prediction

Online Learning and Planning in Cognitive Hierarchies

2023-10-18 · Bernhard Hengst, Maurice Pagnucco, David Rajaratnam, Claude Sammut 외

Complex robot behaviour typically requires the integration of multiple robotic and Artificial Intelligence (AI) techniques and components. Integrating such disparate components into a coherent system, while also ensuring…