Home

Awesome

Adaptive Adapter Routing for Long-Tailed Class-Incremental Learning

<div align="center"> <div> <a href='http://www.lamda.nju.edu.cn/qizh' target='_blank'>Zhi-Hong Qi</a>&emsp; <a href='http://www.lamda.nju.edu.cn/zhoudw' target='_blank'>Da-Wei Zhou</a>&emsp; <a>Yiran Yao</a>&emsp; <a href='http://www.lamda.nju.edu.cn/yehj' target='_blank'>Han-Jia Ye</a>&emsp; <a href='http://www.lamda.nju.edu.cn/zhandc' target='_blank'>De-Chuan Zhan</a> </div> <div> School of Artificial Intelligence, State Key Laboratory for Novel Software Technology, Nanjing University&emsp; </div> </div> <div align="center"> </div>

The code repository for "Adaptive Adapter Routing for Long-Tailed Class-Incremental Learning" (MJL 2024) in PyTorch.

<!-- If you use any content of this repo for your work, please cite the following bib entry: @article{zhou2024revisiting, author = {Zhou, Da-Wei and Cai, Zi-Wen and Ye, Han-Jia and Zhan, De-Chuan and Liu, Ziwei}, title = {Revisiting Class-Incremental Learning with Pre-Trained Models: Generalizability and Adaptivity are All You Need}, journal = {International Journal of Computer Vision}, year = {2024} } -->

🔧 Requirements

Environment

  1. torch 1.11.0
  2. torchvision 0.12.0
  3. timm 0.6.12

Dataset

We provide the processed datasets as follows:

These subsets are sampled from the original datasets. Please note that I do not have the right to distribute these datasets. If the distribution violates the license, I shall provide the filenames instead.

You need to modify the path of the datasets in ./utils/data.py according to your own path.

💡 Running scripts

To prepare your JSON files, refer to the settings in the exps folder and run the following command. The results can be found in the logs folder.

python main.py --config ./exps/[configname].json

🎈 Acknowledgement

This repo is based on LAMDA-PILOT, CIL_Survey and PyCIL.

💭 Correspondence

If you have any questions, please contact me via email or open an issue.