LoRI: Reducing Cross-Task Interference in Multi-Task Low-Rank Adaptation
작성자
Haebom
저자
Juzheng Zhang, Jiacheng You, Ashwinee Panda, Tom Goldstein
개요
LoRA는 매개변수 효율적인 미세 조정(PEFT) 방법으로 널리 사용되지만, 여전히 상당한 오버헤드가 발생하고 다중 작업 시나리오에서 매개변수 간섭 문제가 발생합니다. 본 논문에서는 간단하면서도 효과적인 접근 방식인 LoRA with Reduced Interference (LoRI)를 제안합니다. LoRI는 투영 행렬 A를 무작위 투영으로 고정하고, B 행렬을 작업별 마스크를 사용하여 희소화합니다. 이 설계는 훈련 가능한 매개변수의 수를 크게 줄이면서 강력한 작업 성능을 유지합니다. 또한, LoRI는 어댑터 하위 공간 간의 직교성을 활용하여 어댑터 병합에서 작업 간 간섭을 최소화하고, 희소성을 사용하여 파국적인 망각을 완화하여 지속적인 학습을 지원합니다. 자연어 이해, 수학적 추론, 코드 생성 및 안전 정렬 작업에 대한 광범위한 실험을 통해 LoRI가 LoRA보다 최대 95% 적은 훈련 가능한 매개변수를 사용하면서 전체 미세 조정 및 기존 PEFT 방법보다 우수한 성능을 보임을 보여줍니다. 다중 작업 실험에서 LoRI는 작업 간 간섭을 줄이고 효과적인 어댑터 병합 및 지속적인 학습을 가능하게 합니다. 코드는 https://github.com/juzhengz/LoRI 에서 확인할 수 있습니다.