IemProg / CoFiMA

Official code for "Weighted Ensemble Models Are Strong Continual Learners"

Weighted Ensemble Models Are Strong Continual Learners

Imad Eddine MAROUF¹  Subhankar Roy²  Enzo Tartaglione¹  Stéphane Lathuilière ¹

¹Telecom-Paris, Institut Polytechnique de Paris 

²University of Aberdeen 

The code repository for "Weighted Ensemble Models Are Strong Continual Learners" in PyTorch.

Abstract

In this work, we study the problem of continual learning (CL) where the goal is to learn a model on a sequence of tasks, with the data from previous tasks becoming unavailable while learning on the current task data. CL is essentially a balancing act between learning new tasks (plasticity) and maintaining performance on previously learned concepts (stability). To address the stability-plasticity trade-off, we propose performing weight-ensembling of the model parameters of the previous and current task. This weight-ensembled model, which we call Continual Model Averaging (or CoMA), achieves high accuracy on the current task without deviating significantly from the previous weight configuration, ensuring stability. We also propose an improved variant of CoMA, named Continual Fisher-weighted Model Averaging (or CoFiMA), that selectively weighs each parameter in the weight ensemble by leveraging the Fisher information of the model's weights. Both variants are conceptually simple, easy to implement, and effective in achieving state-of-the-art performance on several standard CL benchmarks.

Requirement

install the conda environment using the environment.yml file

conda env create --file environement.yml

Pre-trained Models

Please download pre-trained ViT-Base models from MoCo v3 and ImageNet-21K and then put or link the pre-trained models to CoFiMA/pretrained

Log file

The default log file for CoFiMA evaluated on the main benchmarks in Tab.1 are in CoFiMA/log.txt

Training

to launch the training of CoMA or CoFiMA on CIFAR-100, run the following command:

python main.py --config configs/cofima/cofima_cifar.yaml

python main.py --config configs/cofima/cofima_imgnetr.yaml

Citation

If you find this work helpful, please cite our paper.

@misc{marouf2023cofima,
      title={Weighted Ensemble Models Are Strong Continual Learners}, 
      author={Imad Eddine Marouf and Subhankar Roy and Enzo Tartaglione and Stéphane Lathuilière},
      year={2023},
      eprint={2310.11482},
      archivePrefix={arXiv},
      primaryClass={cs.CV}
}

Acknowledgment

This repo is heavily based on PyCIL, SLCA, many thanks.