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

Learning to Select, Not Relearn: Hard-Routed Mixtures of Reasoning LoRAs

About

Composing independently trained LoRA adapters into a single large language model is useful for multi-domain adaptation, especially when the original training data cannot be shared. A common approach is to use MoE-style routing over LoRA experts, but for frozen pretrained adapters, soft weighted combinations can change the unit-scale additive update under which each LoRA module was originally trained. We propose \textbf{Hard-Routed MoR-LoRA}, a two-stage framework for composing frozen reasoning LoRA experts through unit-scale hard selection. First, domain-specific LoRA adapters are trained independently using reinforcement learning from verifiable feedback to obtain reasoning experts. Then, all experts are frozen, reasoning traces are distilled from them, and only a lightweight shared router together with a small attention LoRA is trained for integration. The router selects exactly one expert per token using hard top-1 routing, while a straight-through estimator enables gradient-based training. Experiments across five benchmarks, multiple model scales, and additional model families show that Hard-Routed MoR-LoRA preserves expert behavior while requiring substantially fewer trainable parameters than soft-routing mixture baselines. Our analysis further shows that normalized soft mixtures often concentrate most routing mass on a single expert, suggesting that hard unit-scale routing provides a simple and efficient abstraction for frozen LoRA expert composition.

Seyed Alireza Molavi, Zhan Su, Yan Hu, Peyman Sheikholharam Mashhadi, Stefan Byttner, Prayag Tiwari• 2026

Related benchmarks

TaskDatasetResultRank
Sentiment ClassificationSST-2
Accuracy90.71
220
Math Word Problem SolvingGSM8K
Accuracy84.69
117
Linguistic AcceptabilityCOLA
Accuracy (CoLA)78.09
108
Mathematical ReasoningSVAMP
Accuracy (%)85.67
71
Boolean Question AnsweringBoolQ
Accuracy84.43
56
Scientific Question AnsweringARC Challenge
Accuracy84.61
16
Multi-task Language EvaluationGSM8K, ARC-C, MedQA, BoolQ, and CoLA Evaluation Average
Average Accuracy79.8
14
Multi-task Language ReasoningReasoning and Language Understanding Suite (GSM8K, ARC-C, MEDQA, BOOLQ, COLA)
GSM8K Accuracy90.67
6
Showing 8 of 8 rows

Other info

Follow for update