atlas

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

QuanTA: Efficient High-Rank Fine-Tuning of LLMs with Quantum-Informed Tensor Adaptation

NeurIPS 2024 · Large language models › Instruction and fine-tuning · approach: Instruction data selection · 6 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 QuanTA, a quantum-circuit-inspired PEFT method that parameterizes high-rank weight updates via two-axis tensor operations on reshaped hidden states, merged at inference, outperforming LoRA and DoRA on commonsense and arithmetic reasoning with ~0.03-0.2% trainable parameters.

Problem

LoRA's low-rank weight updates may be insufficient for complex downstream tasks, limiting fine-tuning performance relative to full fine-tuning, while existing parameter-efficient methods remain constrained to low-rank representations.

Approach

Reshape the hidden dimension into N qudit-like axes and apply a sequence of trainable two-axis tensors, analogous to a quantum circuit of two-qubit gates, to parameterize full-rank (high-rank) weight updates; a frozen copy of the tensors at initialization guarantees zero initial update, and the result merges into the base weight with no inference overhead.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Use QuanTA instead of the strongest listed baselineLLaMA-7B · Commonsense reasoning (Avg.)accuracy: 78.4 -> 80.3Helped
Use QuanTA instead of the strongest listed baselineLLaMA-13B · Commonsense reasoning (Avg.)accuracy: 81.5 -> 83.1Helped
Use QuanTA instead of full fine-tuningLLaMA2-7B · Commonsense reasoning (Avg.)accuracy: 81.2 -> 81.6Helped
Use QuanTA instead of LoRALLaMA2-13B · Commonsense reasoning (Avg.)accuracy: 82.3 -> 84.8Helped
Use QuanTA instead of DoRALLaMA3-8B · Commonsense reasoning (Avg.)accuracy: 85.2 -> 85.8Helped
Use QuanTA instead of LoRALLaMA2-7B · Arithmetic reasoning (Avg. W/O AQuA)accuracy: 79.6 -> 80.5Helped
Use QuanTA instead of LoRALLaMA2-13B · Arithmetic reasoning (Avg. W/O AQuA)accuracy: 82.5 -> 83.9Helped
Use QuanTA instead of LoRALLaMA2-7B · AQuAaccuracy: 17.5 -> 16.7Hurt
Use QuanTA instead of LoRALLaMA2-13B · AQuAaccuracy: 16.7 -> 18.9Helped

Surprising findings

Limitations

Experimental setup

Datasets: DROP Commonsense170K Math10K BoolQ PIQA SIQA HellaSwag WinoGrande ARC-Easy ARC-Challenge OBQA AQuA

Benchmarks: DROP BoolQ PIQA SIQA HellaSwag WinoGrande ARC-Easy ARC-Challenge OBQA AQuA GSM8K MAWPS

Metrics: F1-score accuracy

Baselines: LoRA DoRA full fine-tuning series adapters parallel adapters prefix tuning GPT-3.5

Builds on: LoRA DoRA KronA

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