Long-Tail Learning with Foundation Model: Heavy Fine-Tuning Hurts

Jiang-Xin Shi, Tong Wei, Zhi Zhou, Xin-Yan Han, Jiejing Shao +1 more
9/1/2023

Official code: shijxcs/LIFT (109 stars, last commit 2024-10-22).

Abstract

The fine-tuning paradigm in addressing long-tail learning tasks has sparked significant interest since the emergence of foundation models. Nonetheless, how fine-tuning impacts performance in long-tail learning was not explicitly quantified. In this paper, we disclose that heavy fine-tuning may even lead to non-negligible performance deterioration on tail classes, and lightweight fine-tuning is more effective. The reason is attributed to inconsistent class conditions caused by heavy fine-tuning. With the observation above, we develop a low-complexity and accurate long-tail learning algorithms LIFT with the goal of facilitating fast prediction and compact models by adaptive lightweight fine-tuning. Experiments clearly verify that both the training time and the learned parameters are significantly reduced with more accurate predictive performance compared with state-of-the-art approaches. The implementation code is available at https://github.com/shijxcs/LIFT.

View on arXivView PDF

Code Implementations(1)

shijxcs/LIFTOfficial100%

Source code for the paper "Long-Tail Learning with Foundation Model: Heavy Fine-Tuning Hurts" (ICML 2024)

10911Shell, PythonSep 7, 20231 years ago

Cite this paper

@article{shi2023longtail,
  title  = {Long-Tail Learning with Foundation Model: Heavy Fine-Tuning Hurts},
  author = {Jiang-Xin Shi and Tong Wei and Zhi Zhou and Xin-Yan Han and Jiejing Shao and Yu-Feng Li},
  year   = {2023},
  eprint = {2309.10019},
  archivePrefix = {arXiv},
  url    = {https://arxiv.org/abs/2309.10019},
  journal = {ICML 2023 2023}
}

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

Discussion