Summing Up the Facts: Additive Mechanisms Behind Factual Recall in LLMs
Chughtai, Bilal, Cooney, Alan, Nanda, Neel
–arXiv.org Artificial Intelligence
How do transformer-based large language models (LLMs) store and retrieve knowledge? We focus on the most basic form of this task -- factual recall, where the model is tasked with explicitly surfacing stored facts in prompts of form `Fact: The Colosseum is in the country of'. We find that the mechanistic story behind factual recall is more complex than previously thought. It comprises several distinct, independent, and qualitatively different mechanisms that additively combine, constructively interfering on the correct attribute. We term this generic phenomena the additive motif: models compute through summing up multiple independent contributions. Each mechanism's contribution may be insufficient alone, but summing results in constructive interfere on the correct answer. In addition, we extend the method of direct logit attribution to attribute an attention head's output to individual source tokens. We use this technique to unpack what we call `mixed heads' -- which are themselves a pair of two separate additive updates from different source tokens.
arXiv.org Artificial Intelligence
Feb-11-2024
- Country:
- South America
- Brazil (0.14)
- Peru (0.04)
- Chile (0.04)
- Colombia (0.04)
- Ecuador (0.04)
- Argentina > Pampas
- Buenos Aires F.D. > Buenos Aires (0.04)
- Oceania
- New Zealand > North Island
- Auckland Region > Auckland (0.04)
- Australia > Australian Capital Territory
- Canberra (0.04)
- New Zealand > North Island
- North America
- Mexico (0.04)
- Jamaica (0.04)
- Haiti (0.04)
- Guatemala (0.04)
- Cuba (0.04)
- United States
- Alaska (0.04)
- Louisiana (0.04)
- Idaho (0.04)
- Hawaii (0.04)
- California (0.04)
- Arizona (0.04)
- Alabama (0.04)
- Illinois > Cook County
- Chicago (0.04)
- Canada
- Quebec > Montreal (0.04)
- Ontario > Toronto (0.04)
- Manitoba > Winnipeg Metropolitan Region
- Winnipeg (0.04)
- Alberta > Census Division No. 6
- Calgary Metropolitan Region > Calgary (0.04)
- Europe
- Italy (0.05)
- Germany (0.05)
- Bulgaria (0.04)
- Greece (0.04)
- Ireland (0.04)
- Switzerland (0.04)
- Albania (0.04)
- Belgium (0.04)
- Estonia (0.04)
- Romania (0.04)
- Austria > Vienna (0.04)
- Iceland (0.04)
- Croatia (0.04)
- Ukraine (0.04)
- Czechia > Prague (0.04)
- Sweden > Stockholm
- Stockholm (0.04)
- Russia > Central Federal District
- Moscow Oblast > Moscow (0.04)
- Spain > Galicia
- Madrid (0.04)
- Netherlands > North Holland
- Amsterdam (0.04)
- Hungary > Budapest
- Budapest (0.04)
- Denmark > Capital Region
- Copenhagen (0.04)
- Portugal > Lisbon
- Lisbon (0.04)
- Middle East
- Malta (0.04)
- Cyprus (0.04)
- Republic of Türkiye > Istanbul Province
- Istanbul (0.04)
- Norway > Eastern Norway
- Oslo (0.04)
- Finland > Uusimaa
- Helsinki (0.04)
- France > Île-de-France
- United Kingdom
- Scotland (0.04)
- England
- Leicestershire > Leicester (0.04)
- Greater London > London (0.04)
- Buckinghamshire > Milton Keynes (0.04)
- Poland > Masovia Province
- Warsaw (0.04)
- Asia
- Pakistan (0.04)
- Afghanistan (0.04)
- Bangladesh (0.04)
- Nepal (0.04)
- Cambodia (0.04)
- Russia (0.04)
- Mongolia (0.04)
- Myanmar (0.04)
- Taiwan (0.04)
- Philippines > Luzon
- National Capital Region > City of Manila (0.04)
- Indonesia > Java
- India
- Maharashtra > Mumbai (0.04)
- Kerala (0.04)
- Thailand > Bangkok
- Bangkok (0.04)
- South Korea > Seoul
- Seoul (0.04)
- China
- Middle East
- Malaysia > Kuala Lumpur
- Kuala Lumpur (0.04)
- Japan > Honshū
- Kantō > Tokyo Metropolis Prefecture > Tokyo (0.04)
- Africa
- South America
- Genre:
- Research Report (0.65)
- Industry:
- Leisure & Entertainment > Sports
- Soccer (1.00)
- Basketball (0.68)
- Leisure & Entertainment > Sports
- Technology: