atlas

Research map / Large language models / Instruction and fine-tuning

3-in-1: 2D Rotary Adaptation for Efficient Finetuning, Efficient Batching and Composability

NeurIPS 2024 · Large language models › Instruction and fine-tuning · approach: Calibration and knowledge in LLM tuning · 2 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Proposes RoAd, a PEFT method that applies trainable 2D rotations (with scaling) to pairs of hidden dimensions, enabling element-wise batched multi-adapter serving and composability via distributed interchange intervention.

Problem

PEFT adapters are costly to serve in heterogeneous batches (batch matrix multiplication overhead), and PEFT methods offer limited interpretability/composability of learned subspaces.

Approach

Finetuning mainly changes angular rather than magnitude of representations, so RoAd rotates adjacent dimension pairs by trainable angles with trainable scaling; the block-sparse rotation is reformulated as two element-wise products, giving batching overhead comparable to (IA)^3 and orthogonal subspaces usable for composition.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Use RoAd1 rather than LoReFTLLaMA-7B · Eight commonsense reasoning tasksAvg. accuracy: 80.2 vs. 79.2No clear effect
Increase RoAd1 parameterization to RoAd2 (ablation)RoAd on LLaMA-7B · Eight commonsense reasoning tasksAvg. accuracy: 79.2 -> 80.2Helped
Increase RoAd1 parameterization to RoAd4 (ablation)RoAd on LLaMA-7B · Four arithmetic reasoning tasksAvg. accuracy: 44.0 -> 45.8Helped
Increase RoAd1 parameterization to RoAd2 (ablation)RoAd on LLaMA-7B · Four arithmetic reasoning tasksAvg. accuracy: 44.0 -> 46.2Helped
Increase RoAd1 parameterization to RoAd4 (ablation)RoAd on LLaMA-13B · Four arithmetic reasoning tasksAvg. accuracy: 51.9 -> 52.3Helped
Apply RoAd1 only to the first feed-forward layer (ablation)RoBERTa-base on GLUE · GLUE devAvg.: 85.6 -> 85.1Hurt
Combine RoAd1 with LoRA instead of using LoRA aloneLLaVA1.5-7B · Visual instruction tuningAvg.: 68.5 -> 68.5No clear effect
Use RoAd1 instead of OFT n=2048LLaMA-7B finetuning · 100 iterationsTraining time (s): 1249 -> 25Helped
Use RoAd1 instead of OFT n=2048LLaMA-7B finetuning · 100 iterationsPeak GPU memory (GB): 40 -> 23Helped
Use RoAd1 rather than LoReFTLLaMA-13B · Eight commonsense reasoning tasksAvg. accuracy: 83.3 vs. 83.0Hurt

Surprising findings

Limitations

Experimental setup

Datasets: GLUE Math10K BoolQ PIQA SIQA HellaSwag WinoGrande ARC OBQA AQuA GSM8K MAWPS

Benchmarks: GLUE BoolQ PIQA SIQA HellaSwag WinoGrande ARC-e ARC-c OBQA AQuA GSM8K MAWPS

Metrics: accuracy Matthews correlation Pearson correlation win rate throughput peak GPU memory training time

Baselines: LoRA Adapter BOFT OFT BitFit (IA)^3 RED LoReFT Prefix-tuning DoRA FLoRA

Builds on: LoRA OFT BOFT LoReFT RoPE

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy