Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Queryable LoRA: Instruction-Regularized Routing Over Shared Low-Rank Update Atoms

About

We present a data-adaptive method for parameter-efficient fine-tuning of large neural networks. Standard low-rank adaptation methods improve efficiency by restricting each layer update to a fixed low-rank form, but this static parameterization can be too rigid when the appropriate correction depends on the input and on the evolving depth-wise computation of the network. Our approach replaces a purely layer-local adapter with a shared queryable memory of low-rank update atoms. For each block of layers, the model forms a query from the current low-rank state and a running summary of previous blocks, uses this query to retrieve a content-dependent combination of shared update components via attention, and applies the resulting routed operator within the low-rank bottleneck. In this way, the method retains the efficiency and scalability of low-rank adaptation while allowing the effective update to vary across inputs and to share reusable structure across layers. The resulting architecture provides a principled middle ground between static LoRA-style updates and fully generated parameter updates: it remains compact and parameter-efficient while supporting dynamic, context-sensitive adaptation. Further, we incorporate instruction-regularization by augmenting routing logits with a language-induced prior over update atoms, thereby biasing the selection of low-rank transformations toward semantically relevant directions without generating unconstrained parameter updates. Experiments on noisy non-linear regression tasks and LLM fine-tuning suggest that this queryable update-memory formulation can improve final test performance and training stability compared to standard low-rank adaptation, while using a comparable number of trainable parameters.

Omatharv Bharat Vaidya, Connor T. Jerzak, Nhat Ho, Chandrajit Bajaj• 2026

Related benchmarks

TaskDatasetResultRank
Mathematical ReasoningGSM8K (test)
Accuracy45.3
954
Code GenerationMBPP (test)--
405
Question AnsweringARC (test)
Accuracy65.6
153
Question AnsweringARC Challenge (test)
Accuracy67.2
103
Natural Language UnderstandingSuperGLUE (test)--
74
Question AnsweringGPQA Diamond
Accuracy32.3
61
Reading ComprehensionBoolQ (test)
Accuracy84.4
43
Mathematical ReasoningOrca-Math (test)
Accuracy35.2
31
Reading ComprehensionRACE (test)
Accuracy59.9
11
Mathematical ReasoningNumina-Math (test)
Accuracy23.4
5
Showing 10 of 10 rows

Other info

Follow for update