AREA: Attribute Extraction and Aggregation for CLIP-Based Class-Incremental Learning
Abstract
Class-Incremental Learning (CIL) is important in building real-world learning systems. In CLIP-based CIL, the model performs classification by comparing similarity between visual and textual embeddings obtained from template prompts, e.g., ``a photo of a [CLASS]''. This seemingly monolithic matching process can be decomposed into two conceptually distinct stages: attribute extraction and attribute aggregation. For example, a model may recognize cat using attributes such as fur texture and whiskers. When learning a new class like car, the model must extract additional attributes like wheels and adjust how they are aggregated in the shared representation space. However, since only data from the current task is available, incremental updates can bias both attribute extraction and aggregation toward new classes, leading to catastrophic forgetting. Therefore, we propose AREA for attribute extraction and aggregation in CLIP-based CIL. To stabilize extraction, we anchor class-level visual and textual attributes on the hyperspherical embedding space via principal geodesic analysis. To stabilize aggregation, we learn lightweight task-specific experts with scoring and residual refinement, regularized by a variational information bottleneck objective. During inference, we perform routing over task attribute manifolds via optimal transport for more concise prediction. Experiments show that AREA consistently outperforms SOTA methods. Code is available at https://github.com/LAMDA-CL/ICML2026-AREA.
Lay Summary
This paper tackles catastrophic forgetting in CLIP-based CIL by focusing on the underlying mechanism of attribute utilization. We stabilize the learning process by explicitly anchoring attribute extraction via a PGA-based mechanism and refining aggregation through a lightweight, learnable scoring module. By introducing interventional and compression terms to implicitly optimize the VIB loss, our approach effectively filters out redundancy, allowing the model to focus on the most discriminative attributes. Additionally, we consider an OT-based selection strategy to regulate attribute isolation and collaboration across tasks, further enhancing incremental stability. Extensive experiments across multiple benchmarks verify the superiority of AREA.