OrchMoE: Efficient Multi-Adapter Learning with Task-Skill Synergy

Haowen Wang, Tao Sun, Kaixiang Ji, Jian Wang, Cong Fan +1 more
1/19/2024
cs.LGcs.AIcs.CL

No confident code implementation found for this paper.

Abstract

We advance the field of Parameter-Efficient Fine-Tuning (PEFT) with our novel multi-adapter method, OrchMoE, which capitalizes on modular skill architecture for enhanced forward transfer in neural networks. Unlike prior models that depend on explicit task identification inputs, OrchMoE automatically discerns task categories, streamlining the learning process. This is achieved through an integrated mechanism comprising an Automatic Task Classification module and a Task-Skill Allocation module, which collectively deduce task-specific classifications and tailor skill allocation matrices. Our extensive evaluations on the 'Super Natural Instructions' dataset, featuring 1,600 diverse instructional tasks, indicate that OrchMoE substantially outperforms comparable multi-adapter baselines in terms of both performance and sample utilization efficiency, all while operating within the same parameter constraints. These findings suggest that OrchMoE offers a significant leap forward in multi-task learning efficiency.

View on arXivView PDF

Code Implementations

No confident code match yet

We couldn't find an author-owned or strongly-evidenced community implementation for this paper. Any repos shown below are weak matches — verify before relying on them.

No code implementations found yet.

Know of an implementation? Let us know in the comments below!

Similar Papers

Cite this paper

@article{wang2024orchmoe,
  title  = {OrchMoE: Efficient Multi-Adapter Learning with Task-Skill Synergy},
  author = {Haowen Wang and Tao Sun and Kaixiang Ji and Jian Wang and Cong Fan and Jinjie Gu},
  year   = {2024},
  eprint = {2401.10559},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2401.10559v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2401.10559.svg)](https://codeofpaper.com/paper/2401.10559)

Discussion