提出残差分解框架,让模型更准地排序长尾类别。
Beyond Logit Adjustment: A Residual Decomposition Framework for Long-Tailed Reranking

- 将纠错分为类固定项和输入相关项,动态调整排名
- 在9个数据集上提升长尾分类性能,尤其在复杂场景下
- 适合需要精准排序的罕见类别任务,如医疗诊断
长尾分类中,少数高频类主导多数稀有类,模型在推理时倾向于偏好高频类。现有后处理方法如logit adjustment通过为每个类别添加固定偏移来修正,但该修正无需对所有输入恒定,固定偏移无法适应变化。本文从贝叶斯最优重排序出发,研究基模型top-k候选列表上的重排序问题。发现最优得分与基得分之间的残差可分解为类内恒定的类间项和依赖输入及竞争标签的成对项。当残差仅为类间项时,固定偏移足以恢复最优排序;若同一标签对在不同上下文产生冲突排序约束,则固定偏移无法实现恢复。该分解可预测成对修正是否有效。我们提出REPAIR(基于成对残差修正的重排序),结合收缩稳定的类间项与由短名单竞争特征驱动的线性成对项。九个基准测试验证了该分解能准确判断成对修正何时有效、何时仅需类间修正。涵盖文本分类、视觉识别和多模态罕见病诊断任务。
原文摘要 · Abstract (English)
Long-tailed classification, where a small number of frequent classes dominate many rare ones, remains challenging because models systematically favor frequent classes at inference time. Existing post-hoc methods such as logit adjustment address this by adding a fixed classwise offset to the base-model logits. However, the correction required to restore the relative ranking of two classes need not be constant across inputs, and a fixed offset cannot adapt to such variation. We study this problem through Bayes-optimal reranking on a base-model top-k shortlist. The gap between the optimal score and the base score, the residual correction, decomposes into a classwise component that is constant within each class, and a pairwise component that depends on the input and competing labels. When the residual is purely classwise, a fixed offset suffices to recover the Bayes-optimal ordering. We further show that when the same label pair induces incompatible ordering constraints across contexts, no fixed offset can achieve this recovery. This decomposition leads to testable predictions regarding when pairwise correction can improve performance and when cannot. We develop REPAIR (Reranking via Pairwise residual correction), a lightweight post-hoc reranker that combines a shrinkage-stabilized classwise term with a linear pairwise term driven by competition features on the shortlist. Experiments on nine benchmarks confirm that the decomposition explains where pairwise correction helps and where classwise correction alone suffices. These span text classification, visual recognition, and multimodal rare-disease diagnosis.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。