Distributed Multi-Layer Editing for Rule-Level Knowledge in Large Language Models

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Wang, Yating, Zhao, Wenting, Zhao, Yaqi, Gong, Yongshun, Yin, Yilong, Sun, Haoliang
Format: Preprint
Published: 2026
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866914461466492928
author Wang, Yating
Zhao, Wenting
Zhao, Yaqi
Gong, Yongshun
Yin, Yilong
Sun, Haoliang
author_facet Wang, Yating
Zhao, Wenting
Zhao, Yaqi
Gong, Yongshun
Yin, Yilong
Sun, Haoliang
contents Large language models store not only isolated facts but also rules that support reasoning across symbolic expressions, natural language explanations, and concrete instances. Yet most model editing methods are built for fact-level knowledge, assuming that a target edit can be achieved through a localized intervention. This assumption does not hold for rule-level knowledge, where a single rule must remain consistent across multiple interdependent forms. We investigate this problem through a mechanistic study of rule-level knowledge editing. To support this study, we extend the RuleEdit benchmark from 80 to 200 manually verified rules spanning mathematics and physics. Fine-grained causal tracing reveals a form-specific organization of rule knowledge in transformer layers: formulas and descriptions are concentrated in earlier layers, while instances are more associated with middle layers. These results suggest that rule knowledge is not uniformly localized, and therefore cannot be reliably edited by a single-layer or contiguous-block intervention. Based on this insight, we propose Distributed Multi-Layer Editing (DMLE), which applies a shared early-layer update to formulas and descriptions and a separate middle-layer update to instances. While remaining competitive on standard editing metrics, DMLE achieves substantially stronger rule-level editing performance. On average, it improves instance portability and rule understanding by 13.91 and 50.19 percentage points, respectively, over the strongest baseline across GPT-J-6B, Qwen2.5-7B, Qwen2-7B, and LLaMA-3-8B. The code is available at https://github.com/Pepper66/DMLE.
format Preprint
id arxiv_https___arxiv_org_abs_2604_08284
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle Distributed Multi-Layer Editing for Rule-Level Knowledge in Large Language Models
Wang, Yating
Zhao, Wenting
Zhao, Yaqi
Gong, Yongshun
Yin, Yilong
Sun, Haoliang
Computation and Language
Artificial Intelligence
I.2.4; I.2.6
Large language models store not only isolated facts but also rules that support reasoning across symbolic expressions, natural language explanations, and concrete instances. Yet most model editing methods are built for fact-level knowledge, assuming that a target edit can be achieved through a localized intervention. This assumption does not hold for rule-level knowledge, where a single rule must remain consistent across multiple interdependent forms. We investigate this problem through a mechanistic study of rule-level knowledge editing. To support this study, we extend the RuleEdit benchmark from 80 to 200 manually verified rules spanning mathematics and physics. Fine-grained causal tracing reveals a form-specific organization of rule knowledge in transformer layers: formulas and descriptions are concentrated in earlier layers, while instances are more associated with middle layers. These results suggest that rule knowledge is not uniformly localized, and therefore cannot be reliably edited by a single-layer or contiguous-block intervention. Based on this insight, we propose Distributed Multi-Layer Editing (DMLE), which applies a shared early-layer update to formulas and descriptions and a separate middle-layer update to instances. While remaining competitive on standard editing metrics, DMLE achieves substantially stronger rule-level editing performance. On average, it improves instance portability and rule understanding by 13.91 and 50.19 percentage points, respectively, over the strongest baseline across GPT-J-6B, Qwen2.5-7B, Qwen2-7B, and LLaMA-3-8B. The code is available at https://github.com/Pepper66/DMLE.
title Distributed Multi-Layer Editing for Rule-Level Knowledge in Large Language Models
topic Computation and Language
Artificial Intelligence
I.2.4; I.2.6
url https://arxiv.org/abs/2604.08284