Consecutive Batch Model Editing with HooK Layers

TL;DR

CoachHooK is a novel model editing method that efficiently supports both sequential and batch editing scenarios with minimal memory requirements, outperforming existing methods in stability and effectiveness.

Contribution

The paper introduces CoachHooK, a memory-efficient model editing approach that supports both sequential and batch editing, addressing limitations of prior methods.

Findings

Outperforms existing batch-supportive editing methods.

Maintains stability over multiple consecutive editing steps.

Requires only small, fixed memory for hook layers.

Abstract

As the typical retraining paradigm is unacceptably time- and resource-consuming, researchers are turning to model editing to find an effective way that supports both consecutive and batch scenarios to edit the model behavior directly. Despite all these practical expectations, existing model editing methods fail to realize all of them. Furthermore, the memory demands for such sequential model editing approaches tend to be prohibitive, frequently necessitating an external memory that grows incrementally over time. To cope with these challenges, we propose CoachHooK, a model editing method that simultaneously supports sequential and batch editing. CoachHooK is memory-friendly as it only needs a small amount of it to store several hook layers whose size remains unchanged over time. Experimental results demonstrate the superiority of our method over other batch-supportive model editing methods under both single-round and consecutive batch editing scenarios. Extensive analyses of CoachHooK have been conducted to verify the stability of our method over a number of consecutive steps.