mlsanityData to 7 Oct 2026

Task-Aware LoRA Adapter Composition via Similarity Retrieval in Vector Databases

arXiv 2602.21222 ↗First posted 1 Feb 2026cs.CL cs.AI cs.LGcited by 0

Not on the map. Why?

Abstract (arXiv)

Parameter efficient fine tuning methods like LoRA have enabled task specific adaptation of large language models, but efficiently composing multiple specialized adapters for unseen tasks remains challenging. We present a novel framework for dynamic LoRA adapter composition that leverages similarity retrieval in vector databases to enable zero-shot generalization across diverse NLP tasks. Our approach constructs a task-aware vector database by embedding training examples from 22 datasets spanning commonsense reasoning, question answering, natural language inference, and sentiment analysis. At inference time, we retrieve the most similar training examples, compute task similarity distributions via nucleus sampling, and dynamically merge relevant LoRA adapters using retrieval weighted fusion strategies. We evaluated four merging methods Linear, Concatenation, TIES, and Magnitude Prune demonstrating that our dataset centric retrieval approach often matches or exceeds the performance of individually fine-tuned task-specific adapters. Notably, Linear merging achieves 70.95% on PIQA and 77.62% on RTE, substantially outperforming single-task baselines (46% and 52%, respectively). Our framework requires no additional retriever training, operates with frozen embeddings, and enables efficient, interpretable adapter composition. These results suggest that retrieval based dynamic merging offers a promising direction for scalable, parameter-efficient multitask learning without requiring full model retraining for each new task.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 7 of its 11 references that are in the map. First posted on arXiv 1 Feb 2026.

Builds on

7 of 11 references.

References · 7

  1. 2022Editing Models with Task Arithmeticcited by 1,076
  2. 2024Mixture of LoRA Expertscited by 179
  3. 2020Exploring and Predicting Transferability across NLP Taskscited by 126
  4. 2024LoraRetriever: Input-Aware LoRA Retrieval and Composition for Mixed Tasks in the Wildcited by 68
  5. 2024What Matters for Model Merging at Scale?cited by 52
  6. 2024LoRA-Flow: Dynamic LoRA Fusion for Large Language Models in Generative Taskscited by 39
  7. 2024Retrieval-Augmented Mixture of LoRA Experts for Uploadable Machine Learningcited by 14

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2025From Isolation to Alignment: Unified LoRA for Efficient Multi-Task Learning
  2. 2023Token-Level Adaptation of LoRA Adapters for Downstream Task Generalization
  3. 2024LoRA-XS: Low-Rank Adaptation with Extremely Small Number of Parameters
  4. 2026TLoRA: Task-aware Low Rank Adaptation of Large Language Models
  5. 2024MTL-LoRA: Low-Rank Adaptation for Multi-Task Learning
  6. 2024VB-LoRA: Extreme Parameter Efficient Fine-Tuning with Vector Banks
  7. 2026A Unified Study of LoRA Variants: Taxonomy, Review, Codebase, and Empirical Evaluation
  8. 2026SAMoRA: Semantic-Aware Mixture of LoRA Experts for Task-Adaptive Learning