Let's Focus on Neuron: Neuron-Level Supervised Fine-tuning for Large Language Model

Haoyun Xu, Runzhe Zhan, Derek F. Wong, Lidia S. Chao
3/18/2024
cs.CL

No official code found. Best implementation: Krypton808/NeFT (10 stars, last commit 2025-01-17).

Abstract

Large Language Models (LLMs) are composed of neurons that exhibit various behaviors and roles, which become increasingly diversified as models scale. Recent studies have revealed that not all neurons are active across different datasets, and this sparsity correlates positively with the task-specific ability, leading to advancements in model pruning and training efficiency. Traditional fine-tuning methods engage all parameters of LLMs, which is computationally expensive and may not be necessary. In contrast, Parameter-Efficient Fine-Tuning (PEFT) approaches aim to minimize the number of trainable parameters, yet they still operate at a relatively macro scale (e.g., layer-level). We introduce Neuron-Level Fine-Tuning (NeFT), a novel approach that refines the granularity of parameter training down to the individual neuron, enabling more precise and computationally efficient model updates. The experimental results show that NeFT not only exceeded the performance of full-parameter fine-tuning and PEFT but also provided insights into the analysis of neurons.

View on arXivView PDF

Code Implementations(1)

Krypton808/NeFTCommunity67%

[COLING 2025] Let's Focus on Neuron: Neuron-Level Supervised Fine-tuning for Large Language Model

104Shell, PythonDec 16, 20241 years ago

Similar Papers

Cite this paper

@article{xu2024lets,
  title  = {Let's Focus on Neuron: Neuron-Level Supervised Fine-tuning for Large Language Model},
  author = {Haoyun Xu and Runzhe Zhan and Derek F. Wong and Lidia S. Chao},
  year   = {2024},
  eprint = {2403.11621},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2403.11621v1}
}

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

Discussion