vix.ing · top · new · best · stats

Online Continual Learning with Contrastive Vision Transformer

2022/07/24 by Zhen Wang, Wang, Zhen, Liu Liu +7 · 4 citations
Computer Science · Engineering · Medicine · Psychology · #Artificial Intelligence (cs.AI) #Artificial intelligence #COVID-19 diagnosis using AI #Classifier (UML) #Cognitive psychology #Computer Vision and Pattern Recognition (cs.CV) #Computer science #Control engineering #Decoupling (probability) #Domain Adaptation and Few-Shot Learning #Engineering #FOS: Computer and information sciences #Forgetting #Machine Learning (cs.LG) #Machine learning #Psychology #Transformer #Voltage #cs.AI #cs.CV #cs.LG

paper · pdf · doi:10.48550/arxiv.2207.13516

published in arXiv (Cornell University) (Cornell University) · ECCV2022

arxiv created 2022/07/24 · openalex publication_date 2022/07/24 · arxiv updated 2022/07/28 · openalex created_date 2025/10/10 · openalex updated_date 2026/07/28

Abstract

Online continual learning (online CL) studies the problem of learning sequential tasks from an online data stream without task boundaries, aiming to adapt to new data while alleviating catastrophic forgetting on the past tasks. This paper proposes a framework Contrastive Vision Transformer (CVT), which designs a focal contrastive learning strategy based on a transformer architecture, to achieve a better stability-plasticity trade-off for online CL. Specifically, we design a new external attention mechanism for online CL that implicitly captures previous tasks' information. Besides, CVT contains learnable focuses for each class, which could accumulate the knowledge of previous classes to alleviate forgetting. Based on the learnable focuses, we design a focal contrastive loss to rebalance contrastive learning between new and past classes and consolidate previously learned representations. Moreover, CVT contains a dual-classifier structure for decoupling learning current classes and balancing all observed classes. The extensive experimental results show that our approach achieves state-of-the-art performance with even fewer parameters on online CL benchmarks and effectively alleviates the catastrophic forgetting.

Cited by

Related