Teknik KILAVUZ

LoRA Uzmanlarının Karışımı

Mixture of LoRA experts is a family of fine-tuning methods that attach several small low-rank adapters, called LoRA experts, to a frozen language model and use a learned router to decide how much each adapter contributes for each input.

Bu kılavuzu dinleyinAI tarafından anlatılan · ~7 min

Bu sayfada8 min read
  1. Genel Bakış
  2. Önemli çıkarımlar
  3. Derin Dalış
  4. Count the parameters in a small LoRA mixture
  5. Gerçek Dünya Uygulaması
  6. LoRA Uzmanları Karışımının Geleceği
  7. Uygulamaya koymak
  8. Kaynaklar ve ek okuma
  9. Keşfetmeye Devam Edin
  10. Sık sorulan sorular

Genel Bakış

It combines the low cost of LoRA with the specialization of mixture-of-experts models, so one base model can handle many tasks while training under 1% of its parameters.

Önemli çıkarımlar

  1. A mixture of LoRA experts attaches several low-rank adapters to a frozen model and uses a router to weight them for each input.
  2. LoRA cut trainable parameters 10,000-fold and GPU memory threefold compared with fully fine-tuning GPT-3 175B, according to its 2021 paper.
  3. Cohere's 2023 study matched full fine-tuning while updating under 1% of an 11-billion-parameter model using lightweight experts.
  4. Dynamic routing prevents merging adapters into the base weights, so mixtures of LoRA experts add some inference cost that a single merged LoRA avoids.
  5. Load-balancing losses keep routers from sending nearly all tokens to a few experts, a failure called expert collapse.

Derin Dalış

Mixture of LoRA experts combines two established ideas. The first is LoRA, low-rank adaptation, introduced by Edward Hu and colleagues in June 2021. Instead of retraining all the weights of a large model, LoRA freezes them and learns a small correction for selected weight matrices, written as the product of two thin matrices. For GPT-3 175B, the LoRA paper reports 10,000 times fewer trainable parameters and three times less GPU memory than full fine-tuning, with no added inference delay once the correction is merged into the weights. The second idea is the mixture of experts (MoE): a model contains several sub-networks, called experts, and a small gating network, or router, chooses which ones to use for each input. Noam Shazeer and colleagues showed in 2017 that sparsely gated MoE layers with up to thousands of experts could raise model capacity more than 1,000-fold with little extra computation.

A mixture of LoRA experts uses LoRA adapters as the experts. Step by step: a pretrained model is frozen; at chosen layers, usually the attention projections or the feed-forward block, several LoRA adapters are attached side by side; a router, typically one small linear layer followed by a softmax, reads the current token's hidden state and outputs a weight for each adapter; each adapter computes its low-rank correction; and the corrections are added to the frozen layer's output in proportion to the router's weights. With sparse routing, only the top one or two adapters are computed for each token. During training only the adapters and router are updated.

The reason to do this rather than train one bigger adapter is interference. When one adapter is trained on a mixture of very different tasks, such as coding, translation and arithmetic, the tasks compete for the same few parameters. Separate experts let different kinds of input use different parameters. LoRAMoE, posted in December 2023 by Shihan Dou and colleagues, targeted a specific version of this problem: scaling up instruction data can damage the world knowledge a model learned in pretraining. It splits its LoRA experts so some focus on preserving existing knowledge and others on new tasks, and reports gains on downstream tasks while retaining stored knowledge.

The approach developed through several papers. In September 2023 Ted Zadouri and colleagues at Cohere, in "Pushing Mixture of Experts to the Limit", built mixtures of lightweight experts including LoRA-based ones and updated less than 1% of an 11-billion-parameter model, outperforming standard parameter-efficient methods and matching full fine-tuning on instruction tuning. LoraHub, posted in July 2023 by Chengsong Huang and colleagues, took a different route: it combines LoRA modules already trained on separate tasks, choosing fixed mixing weights from a few examples of a new task with a gradient-free search. MixLoRA, from April 2024, added top-k routed LoRA experts inside feed-forward blocks plus separate LoRA adapters on attention, with a load-balancing loss so the router does not overuse a few experts. In April 2024 Xun Wu, Shaohan Huang and Furu Wei described Mixture of LoRA Experts (MoLE), which learns gating over existing trained LoRAs and was tested in both language and vision-and-language settings.

Compared with the alternatives, the trade-offs are clear. Full fine-tuning updates every weight and adapts best, but needs far more memory and produces a full copy of the model per task. A single LoRA adapter is cheapest to train and serve but can struggle with many mixed tasks. Merging several LoRAs by simple weighted addition, a common shortcut, is free at inference but, according to the MoLE paper, can weaken the base model's generation and blur each LoRA's distinct behavior. Keeping separate adapters and switching between them by hand works when the task is known in advance, but requires an outside decision about which adapter to use. A learned router makes that choice automatically, per token, at the cost of extra machinery.

The approach has real limitations. Routers can collapse, sending almost all tokens to one or two experts and leaving the others untrained, which is why methods such as MixLoRA add load-balancing losses. Input-dependent routing removes one of LoRA's main advantages: because the mixture changes token by token, the adapters cannot be merged into the base weights ahead of time, so inference does extra work. Batching requests that use different adapters complicates serving, which is the problem S-LoRA addresses. Reported gains are mostly self-reported on academic benchmarks with specific base models, and results vary with the number of experts, rank, routing method and which layers receive experts.

In practice the costs are modest. Each LoRA expert adds two thin matrices per adapted weight matrix, so even several experts usually total well under the 1% of model parameters reported in the Cohere work. Training fits on the same hardware used for ordinary LoRA fine-tuning, plus a little memory for the extra adapters and router. The main engineering decisions are how many experts to use, their rank, whether routing is dense or top-k, and which layers to adapt. Serving many adapters efficiently benefits from systems built for it; S-LoRA reports serving thousands of concurrent adapters with up to four times the throughput of naive approaches in Hugging Face PEFT and vLLM.

04Çalışılan örnek

Count the parameters in a small LoRA mixture

  1. Take one attention projection of size 4,096 by 4,096, which has about 16.8 million frozen weights.

  2. Add one LoRA adapter of rank 8: A is 8 by 4,096 and B is 4,096 by 8, giving 65,536 trainable parameters.

  3. Make it a mixture of 4 such experts: 4 times 65,536 is 262,144 parameters.

  4. Add a router that maps the 4,096-dimensional input to 4 gate scores: 4,096 times 4 is 16,384 parameters.

  5. Total trainable parameters: 278,528, about 1.7% of the frozen matrix.

  6. With top-2 routing, compute two adapter corrections per token instead of four and compare the extra work with the base matrix multiply.

Ne gösteriyor?

A concrete budget showing that four experts plus a router cost under 2% of one weight matrix, and that top-2 routing halves adapter compute. Check it by building the layer in PyTorch, freezing W0 and summing numel() over parameters with requires_grad set.

Gerçek Dünya Uygulaması

Cohere researchers showed in 2023 that a mixture of LoRA experts trained on instructions updated less than 1% of an 11-billion-parameter model yet performed on par with full fine-tuning.

LoRAMoE (December 2023) dedicates some LoRA experts to preserving a model's stored world knowledge while others learn new tasks, reducing the forgetting caused by large instruction datasets.

LoraHub (COLM 2024) combines LoRA modules trained on different tasks to tackle unseen Big-Bench Hard tasks from a few examples, without gradients or extra parameters.

MixLoRA adds top-k routed LoRA experts to a frozen model's feed-forward layers and reports about 9% higher multi-task accuracy than standard parameter-efficient methods.

S-LoRA, a serving system built for many adapters, serves thousands of concurrent LoRA adapters with up to four times the throughput of naive serving in Hugging Face PEFT and vLLM.

LoRA Uzmanları Karışımının Geleceği

Research since 2023 has explored several directions at once: jointly trained LoRA mixtures for instruction tuning (Cohere, September 2023), mixtures that protect pretrained knowledge (LoRAMoE, December 2023), composition of independently trained adapters (LoraHub, published at COLM 2024, and MoLE, April 2024) and sparse top-k LoRA experts with load balancing (MixLoRA, 2024). Serving infrastructure has advanced in parallel, with S-LoRA showing that thousands of adapters can share one base model on the same hardware. Open problems include stable routing without expert collapse, principled choices for the number and rank of experts, fair comparison across papers that use different base models and benchmarks, and keeping latency low when adapters cannot be merged into the base weights. LoraHub's authors proposed a platform where users share trained LoRA modules for reuse on new tasks; whether shared adapter libraries with automatic routing become common practice, or whether single well-trained adapters and larger base models remain simpler and good enough, is still an open question that depends on evidence from production use rather than benchmarks alone.

Uygulamaya koymak

Neden önemli?

  • Maliyet ve bütçe. Mimari kararlar yıllarca performansı ve işletme maliyetini etkiler.
  • Daha net kararlar. Teknik eğitim, ekiplerin yalnızca en yenisini değil, doğru yığını seçmesine de yardımcı olur.
  • Kalite kontrolü. Daha iyi mühendislik seçenekleri, üretimdeki güvenilirlik olaylarını azaltır.

Dikkat edin

  • Bir kıyaslamayı optimize etmek daha geniş sistem zayıflıklarını gizleyebilir.
  • Altyapı ve bakım maliyetleri genellikle hafife alınır.
  • Sistemler karmaşıklaştıkça güvenlik ve gözlemlenebilirlik boşlukları büyüyebilir.

Sonraki adımlar

  1. Uygulamadan önce gecikmeyi, kaliteyi ve maliyet hedeflerini tanımlayın.
  2. Gerçekçi yük ve veri koşulları altında kıyaslama yapın.
  3. Hatalar, sapmalar ve kullanıcı etkisi için cihaz izleme.
  4. Ölçeklendirmeden önce geri alma ve olay müdahale yollarını hazırlayın.

Kaynaklar ve ek okuma

  1. arXivLoRA: Low-Rank Adaptation of Large Language Models
  2. arXivOutrageously Large Neural Networks: The Sparsely-Gated Mixture-of-Experts Layer
  3. arXiv (Cohere)Pushing Mixture of Experts to the Limit: Extremely Parameter Efficient MoE for Instruction Tuning
  4. arXivLoRAMoE: Alleviate World Knowledge Forgetting in Large Language Models via MoE-Style Plugin
  5. arXiv (COLM 2024)LoraHub: Efficient Cross-Task Generalization via Dynamic LoRA Composition
  6. arXivLoRA Uzmanlarının Karışımı
  7. arXivMixLoRA: Enhancing Large Language Models Fine-Tuning with LoRA-based Mixture of Experts
  8. arXivS-LoRA: Serving Thousands of Concurrent LoRA Adapters

Keşfetmeye Devam Edin

Ücretsiz bülten

Günlük AI brifingini alın

Hafta içi her sabah sade İngilizce yazılmış, doğrulanmış üç yapay zeka hikayesi. Sonsuza kadar ücretsiz, reklam yok.

Hafta içi her gün bir e-posta. Tek tıklamayla aboneliğinizi iptal edin. Adresinizi asla satmayız veya paylaşmayız.

Kendinizi test edin

Take the Mixture of LoRA Experts quiz

Her yanıtla ilgili anında geri bildirim ve bir kursu geçtiğinizde doğrulanabilir bir kimliğe sahip paylaşılabilir bir sertifika.

Testi başlat

Ücretsiz AI eğitimini destekleyin. AI Understanding, bir 501(c)(3) kâr amacı gütmeyen kuruluştur; hiçbir zaman reklam yoktur, ödeme duvarı yoktur. Bağış yap

Sık sorulan sorular

LoRA Uzmanlarının Karışımı Nedir?

Mixture of LoRA experts is a family of fine-tuning methods that attach several small low-rank adapters, called LoRA experts, to a frozen language model and use a learned router to decide how much each adapter contributes for each input. It combines the low cost of LoRA with the specialization of mixture-of-experts models, so one base model can handle many tasks while training under 1% of its parameters.

LoRA Uzmanları Karışımı'nın 'LoRA' kısmı neyi ifade ediyor?

LoRA, Düşük Sıralı Adaptasyon anlamına gelir: temel modeli dondurur ve davranışı ucuza ayarlayan kompakt düşük dereceli matrisleri eğitir.

MoLE'de geçit/yönlendirici ağının görevi nedir?

Yönlendirici, mevcut LoRA uzmanlarına göre ağırlıklar üretir ve bunları girdi başına (ve genellikle katman veya simge başına) seçip harmanlar.

MoLE neden çoğu zaman tek bir bağdaştırıcıyı birçok karma görev için eğitmekten daha iyidir?

Farklı uzmanlar, bir bağdaştırıcının birçok çelişkili beceriyi aynı anda öğrenmesi gerektiğinde ortaya çıkan yıkıcı unutma ve görev müdahalesinden kaçınır.

MoLE eğitimi sırasında temel modelin önceden eğitilmiş ağırlıklarına genel olarak ne olur?

Omurga donmuş halde kalır; yalnızca küçük LoRA uzmanları ve geçiş ağı, kademeli güncellemeleri alır.

Difüzyon görüntü modellerinde, MoLE'deki hiyerarşik/katman başına geçiş hangi sorunun çözülmesine yardımcı olur?

Katman başına geçiş, her bağdaştırıcının her katmana ne kadar güçlü katkıda bulunduğunu öğrenir ve birden fazla konsept LoRA'nın tek bir baskın olmadan birleşmesine olanak tanır.