TARE: Lightweight Token-Aware Representation Editing for Fine-tuning Transformer-like Models

Yulong Wang, Siyu Zhao


Abstract
Parameter-efficient fine-tuning (PEFT) must balance effectiveness and efficiency: low-rank methods can be costly, while global representation edits often underfit token-level contexts. We propose **Token-Aware Representation Editing (TARE)**, a PEFT method that performs fine-grained, token-specific edits with a small additional inference overhead and minimal tuning.After each FFN block in a transformer-like model, we adopt a lightweight selector that scores a small pool of hidden representation editors for each token, activates only the top-k editors, and mixes their element-wise scaling/bias updates. This design achieves superior performance while maintaining computational efficiency, yielding a more favorable Pareto frontier compared to state-of-the-art (SOTA) methods.Across LLaMA-3-8B (eight knowledge reasoning and seven mathematical reasoning tasks) and RoBERTa-base/large (GLUE), TARE outperforms SOTAs (LoRA, DoRA, MiLoRA, LoReFT, and RED), achieving 86.7% (knowledge reasoning), 76.7% (mathematical reasoning), and 88.3% (GLUE) while tuning only 0.0392% of parameters using about 20 GiB peak GPU memory during training.An implementation is available at: <https://github.com/PatriciaPulec/tare>.
Anthology ID:
2026.acl-long.841
Volume:
Proceedings of the 64th Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers)
Month:
July
Year:
2026
Address:
San Diego, California, United States
Editors:
Maria Liakata, Viviane P. Moreira, Jiajun Zhang, David Jurgens
Venue:
ACL
SIG:
Publisher:
Association for Computational Linguistics
Note:
Pages:
18452–18471
Language:
URL:
https://preview.aclanthology.org/ingest-acl/2026.acl-long.841/
DOI:
Bibkey:
Cite (ACL):
Yulong Wang and Siyu Zhao. 2026. TARE: Lightweight Token-Aware Representation Editing for Fine-tuning Transformer-like Models. In Proceedings of the 64th Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers), pages 18452–18471, San Diego, California, United States. Association for Computational Linguistics.
Cite (Informal):
TARE: Lightweight Token-Aware Representation Editing for Fine-tuning Transformer-like Models (Wang & Zhao, ACL 2026)
Copy Citation:
PDF:
https://preview.aclanthology.org/ingest-acl/2026.acl-long.841.pdf
Checklist:
 2026.acl-long.841.checklist.pdf