Group and Shuffle: Efficient Structured Orthogonal Parametrization

Mikhail Gorbunov, Nikolay Yudin, Vera Soboleva, Aibek Alanov, Alexey Naumov +1 more
6/14/2024
cs.LGcs.AIcs.CL

No official code found. Best implementation: Skonor/group_and_shuffle (4 stars, MIT, last commit 2025-01-12). 1 more below.

Abstract

The increasing size of neural networks has led to a growing demand for methods of efficient fine-tuning. Recently, an orthogonal fine-tuning paradigm was introduced that uses orthogonal matrices for adapting the weights of a pretrained model. In this paper, we introduce a new class of structured matrices, which unifies and generalizes structured classes from previous works. We examine properties of this class and build a structured orthogonal parametrization upon it. We then use this parametrization to modify the orthogonal fine-tuning framework, improving parameter and computational efficiency. We empirically validate our method on different domains, including adapting of text-to-image diffusion models and downstream task fine-tuning in language modeling. Additionally, we adapt our construction for orthogonal convolutions and conduct experiments with 1-Lipschitz neural networks.

View on arXivView PDF

Code Implementations(2)

MIT

Code for the NeurIPS 2024 paper: Group and Shuffle: Efficient Structured Orthogonal Parametrization

40Shell, PythonOct 30, 20241 years agoMIT
Skonor/gsoft_demoCommunity72%
MIT

Code for the NeurIPS 2024 paper: Group and Shuffle: Efficient Structured Orthogonal Parametrization

00Shell, PythonDec 27, 20241 years agoMIT

Similar Papers

Cite this paper

@article{gorbunov2024group,
  title  = {Group and Shuffle: Efficient Structured Orthogonal Parametrization},
  author = {Mikhail Gorbunov and Nikolay Yudin and Vera Soboleva and Aibek Alanov and Alexey Naumov and Maxim Rakhuba},
  year   = {2024},
  eprint = {2406.10019},
  archivePrefix = {arXiv},
  doi    = {10.48550/arXiv.2406.10019},
  url    = {http://arxiv.org/abs/2406.10019v1},
  journal = {NEURIPS 2024 2024}
}

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/2406.10019.svg)](https://codeofpaper.com/paper/2406.10019)

Discussion