TS-MOF: Two-Stage Multi-Objective Fine-tuning for Long-Tailed Recognition
Zhe Zhao, Zhiheng Gong, Pengkun Wang, Haibin Wen, Cankun Guo, Bo Xue, Xi Lin, Zhenkun Wang, Qingfu Zhang, Yang Wang
Abstract
Long-Tailed Recognition (LTR) presents a significant challenge due to extreme class imbalance, where existing methods often struggle to balance performance across head and tail classes. Directly applying multi-objective optimization (MOO) to leverage multiple LTR strategies can be complex and unstable. To address this, we propose TS-MOF (Two-Stage Multi-Objective Fine-tuning), a novel framework that strategically decouples feature learning from classifier adaptation. After standard pre-training, TS-MOF freezes the feature backbone and focuses on an efficient multi-objective fine-tuning of specialized classifier heads. The core of TS-MOF's second stage lies in two innovations: Refined Performance Level Agreement for adaptive task weighting based on real-time per-class performance, and Robust Deterministic Projective Conflict Gradient for stable gradient conflict resolution and constructive fusion. This approach enables effective synergy between diverse LTR strategies, leading to significant and balanced performance improvements. Extensive experiments on CIFAR100-LT, ImageNet-LT, and iNaturalist 2018 demonstrate that TS-MOF achieves state-of-the-art results, particularly enhancing tail class accuracy (e.g., +3.3% on CIFAR100-LT IR=100 tail) while improving head class performance, all within a remarkably short fine-tuning period of 20 epochs. We provide the detailed code in https://github.com/DataLab-atom/TS-MOF.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext ddbb8fa5-64ed-489d-b389-d48b317a856dCited by top-tier papers1
Ask how each one uses itBuilds on18
- Gradient Surgery for Multi-Task LearningTianhe Yu, Saurabh Kumar, Abhishek Gupta, Sergey Levine et al.NeurIPS 2020 · 2,261 citations
- Decoupling Representation and Classifier for Long-Tailed RecognitionBingyi Kang, Saining Xie, Marcus Rohrbach, Zhicheng Yan et al.ICLR 2020 · 1,496 citations
- Long-tail learning via logit adjustmentAditya Krishna Menon, Sadeep Jayasumana, Ankit Singh Rawat, Himanshu Jain et al.ICLR 2021 · 937 citations
- Balanced Meta-Softmax for Long-Tailed Visual RecognitionJiawei Ren, Cunjun Yu, Shunan Sheng, Xiao Ma et al.NeurIPS 2020 · 861 citations
- Long-tailed Recognition by Routing Diverse Distribution-Aware ExpertsXudong Wang, Long Lian, Zhongqi Miao, Ziwei Liu et al.ICLR 2021 · 481 citations
Related papers
- Two Fists, One Heart: Multi-Objective Optimization Based Strategy Fusion for Long-tailed LearningZhe Zhao, Pengkun Wang, Haibin Wen, Wei Xu et al.ICML 2024 · 9 citations
- Conflict-Buffering Optimization by Symmetry Teleportation for Deep Long-Tailed RecognitionMianzimei Yang, Zhipeng Zhou, Jin Zhang, Yuanhao Pu et al.ACM MM 2025
- Pareto Deep Long-Tailed Recognition: A Conflict-Averse SolutionZhipeng Zhou, Liu Liu, Peilin Zhao, Wei GongICLR 2024 · 12 citations
- Distribution Alignment: A Unified Framework for Long-Tail Visual RecognitionSongyang Zhang, Zeming Li, Shipeng Yan, Xuming He et al.CVPR 2021
- Bag of Tricks for Long-Tailed Visual Recognition with Deep Convolutional Neural NetworksYongshun Zhang, Xiu-Shen Wei, Boyan Zhou, Jianxin WuAAAI 2021 · 162 citations
