paper-with-me

홈 › Papers

Parametric Memory Decoding for Zero-Shot Routing in LoRA-Based External Parametric Memory

2026-07-05 · Fengxian Ji, Zhuohan Xie, Jingpu Yang, Fan Zhang, Zirui Song, Xiuying Chen arxiv

With the rise of parametric memory, LoRA-based External Parametric Memory (EPM) has emerged as a modular solution, but existing routing methods often introduce additional training, deployment, and maintenance overhead. This raises a natural question: can a LoRA-based EPM bank be routed without maintaining an additional routing component? However, existing zero-shot LoRA routing methods still face two problems under the EPM setting: (1) their evaluations are scattered across different task settings rather than organized around EPM access, and (2) their routing signals lack a unified perspective to guide systematic improvement. To address these problems, we organize PMD-Bench, covering document-level, domain-level knowledge, and task-skill, and propose Parametric Memory Decoding (PMD), the first framework designed to systematically improve zero-shot LoRA routing by reframing it as decoding activations over external parametric memory. Based on PMD, we further instantiate PMDRouter, which scores each LoRA by its response magnitude from a single base-model prefill. Experiments on PMD-Bench show that PMDRouter achieves the strongest internal-signal performance across multiple zero-shot routing settings. These results demonstrate the feasibility of zero-shot LoRA routing and suggest that PMD can serve as a general framework for improving zero-shot routing methods. Sources: Github (https://anonymous.4open.science/r/Parametric-Memory-Decoding-872A/)

📄 PDF Abstract BibTeX arXiv:2607.04118

Code (0)

등록된 구현이 없습니다.

Similar Papers 제목 키워드 기반

Nonparametric Decoding for Generative Retrieval

2022-10-05 · Hyunji Lee, Jaeyoung Kim, Hoyeon Chang, Hanseok Oh 외

The generative retrieval model depends solely on the information encoded in its model parameters without external memory, its information capacity is limited and fixed. To overcome the limitation, we propose Nonparametri…

DecoderLanguage ModellingRetrievalText Retrieval

Routing by Analogy: kNN-Augmented Expert Assignment for Mixture-of-Experts

2026-01-05 · Boxuan Lyu, Soichiro Murakami, Hidetaka Kamigaito, Peinan Zhang arxiv

Mixture-of-Experts (MoE) architectures scale large language models efficiently by employing a parametric ``router'' to dispatch tokens to a sparse subset of experts. Typically, this router is trained once and then frozen…

SinkRouter: Sink-Aware Routing for Efficient Long-Context Decoding in Large Language and Multimodal Models

2026-04-18 · Junnan Liu, Xinyan Liu, Peifeng Gao, Zhaobo Qi 외 arxiv

In long-context decoding for LLMs and LMMs, attention becomes increasingly memory-bound because each decoding step must load a large amount of KV-cache data from GPU memory. Existing acceleration strategies often trade e…

A Parametric Memory Head for Continual Generative Retrieval

2026-04-25 · Kidist Amde Mekonnen, Yubao Tang, Maarten de Rijke arxiv

Generative information retrieval (GenIR) consolidates retrieval into a single neural model that decodes document identifiers (docids) directly from queries. While this model-as-index paradigm offers architectural simplic…

parameter-efficient fine-tuningInformation RetrievalNatural Questions

UniMem: Complementary Episodic-to-Parametric Memory for Boundary-Agnostic Task Streams

2026-07-28 · Siyu Xia, Chenheng Zhang, Yanting Wu, Haoxuan Li 외 arxiv

Memory is essential for LLM agents to accumulate task experience and reuse task-specific execution strategies. However, real-world deployment over boundary-agnostic and evolving task streams exposes a fundamental stabili…