cs.LGSep 28, 2026

Distribution-Conditioned Task Routing for Class-Incremental Learning

Authors: Longhuan Xu, Zhipeng Zhou, Wei Ji, Chunyan Miao, Peilin Zhao, Lijun Zhang

Organizations: Nanjing University · Nanyang Technological University · Shanghai Jiao Tong University

Abstract

Parameter-efficient adaptation enables continual learners to acquire task-specific knowledge through compact model updates while maintaining strong within-task performance. However, class-incremental inference requires each input to be classified among all classes seen so far without access to its task identity. For learners equipped with task-specific parameter-efficient modules, this introduces a critical task-routing challenge beyond catastrophic forgetting. We study post-hoc task routing without retraining the learner or introducing a separately trained router. Such training-free inference-time calibration remains comparatively underexplored in parameter-efficient class-incremental learning. We identify three sources of routing error (feature-level, task-level, and class-level misalignment) and propose Feature Distribution Calibration (FDC). Its three components address these misalignments: Task Subspace Filtering (TSF) suppresses feature components outside each task's principal subspace, Residual Likelihood Calibration (RLC) evaluates the typicality of its subspace residual, and Prototype Affinity Calibration (PAC) measures compatibility with the task's class prototypes. Experiments demonstrate plug-and-play applicability to eight parameter-efficient class-incremental methods using a shared encoder. With one component configuration selected per method across all five benchmarks, FDC improves final accuracy in all 40 method-dataset pairs by 4.39 percentage points on average. Enabling all components improves 35 of the 40 pairs, with an average gain of 4.45 points. When applied to a simple baseline, FDC achieves strong overall performance.

Figures & tables

Appendix figures & tables4 assets

Supplementary material from the paper’s appendix.

Appendix

Explore similar work

CardsList
  1. One Adapter, Many Tasks: Task-Conditioned Feature Transformations for Continual Learning

    Aug 31, 2026Yunxiang Fu, Meng Lou, Yizhou YuClass-Incremental LearningContinual Learning

  2. Decoupled and Distilled: Task-Adaptive LoRA-Teachers with Ensemble Knowledge Transfer for Few-Shot Class-Incremental Learning

    Sep 30, 2026Hongwei Zhao, Rui Liu, Yansong Liu +2Class-Incremental LearningFew-Shot Learning

  3. Hyperbolic Prototype Routing for Rehearsal-Free Class-Incremental Learning

    Sep 30, 2026HongWei Zhao, Rui Liu, Yong ChenClass-Incremental LearningLearnable Prototypes